| {"device": "cuda", "dist": false, "res_dir": "work_dirs", "ex_name": "mv_mmnist/med3_ciatt_lr1e-3_m0_newdata", "fp16": false, "torchscript": false, "seed": 42, "fps": false, "test": true, "deterministic": false, "batch_size": 16, "val_batch_size": 16, "num_workers": 4, "data_root": "/root/data/lsh/openstl_weather/openstl/data", "dataname": "mv_mmnist", "pre_seq_length": 10, "aft_seq_length": 10, "total_length": 20, "use_augment": false, "use_prefetcher": false, "drop_last": false, "method": "its", "config_file": "configs/mv_mmnist/its.py", "model_type": "TAU", "drop": 0.0, "drop_path": 0.0, "overwrite": false, "epoch": 200, "log_step": 1, "opt": "adam", "opt_eps": null, "opt_betas": null, "momentum": 0.9, "weight_decay": 0.0, "clip_grad": null, "clip_mode": "norm", "no_display_method_info": false, "sched": "cosine", "lr": 0.001, "lr_k_decay": 1.0, "warmup_lr": 1e-05, "min_lr": 1e-06, "final_div_factor": 10000.0, "warmup_epoch": 0, "decay_epoch": 100, "decay_rate": 0.1, "filter_bias_and_bn": false, "gpus": [0], "metric_for_bestckpt": "val_loss", "ckpt_path": null, "spatio_kernel_enc": 3, "spatio_kernel_dec": 3, "hid_S": 64, "hid_T": 512, "N_T": 8, "N_S": 4, "momentum_ema": 0, "in_shape": [10, 3, 64, 64], "data_name": "mv_mmnist", "metrics": ["mse", "mae", "ssim", "psnr"]} |