Commit 2cf9d0e9 authored by Gabriele Franch's avatar Gabriele Franch
Browse files

Update testing in taasss

parent 0812e9fd
# import os
import argparse
# import math
# import shutil
import numpy as np
from tqdm import tqdm
import core.trainer as trainer
from cikm_inter_dst_predrnn_run_taasss_utils import get_batcher, padding_taasss
from core.models.model_factory import Model
from core.utils import preprocess
# from data_provider.CIKM.data_iterator import clean_fold, sample, imsave
# from core.utils.util import nor, de_nor
from tqdm import tqdm
from cikm_inter_dst_predrnn_run_taasss_utils import (
padding_taasss,
get_batcher,
)
parser = argparse.ArgumentParser(
description="PyTorch video prediction model - DST PredRNN"
......
This diff is collapsed.
This diff is collapsed.
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment