| {"device": "cuda", "dist": false, "res_dir": "work_dirs", "ex_name": "era5/windDir/ITS/w1_0.01_0_lr1e-3_m0.9", "fp16": false, "torchscript": false, "seed": 42, "fps": false, "test": true, "deterministic": false, "batch_size": 4, "val_batch_size": 4, "num_workers": 4, "data_root": "/home/gc/projects/openstl_wind/data", "dataname": "era5wind", "pre_seq_length": 12, "aft_seq_length": 12, "total_length": 24, "use_augment": false, "use_prefetcher": false, "drop_last": false, "method": "its", "config_file": "configs/weather/era5wind/its.py", "model_type": "TAU", "drop": 0.0, "drop_path": 0.1, "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": 32, "hid_T": 256, "N_T": 8, "N_S": 4, "momentum_ema": 0.9, "in_shape": [12, 4, 128, 128], "data_name": "era5wind", "metrics": ["mse", "mae", "rmse", "ssim", "psnr"]} |