diff --git a/README.md b/README.md index 405e62f85838fbba7fb9f050afcffc6c4d09396d..7dc50941db6efd931180987630df625973dc9178 100644 --- a/README.md +++ b/README.md @@ -33,12 +33,12 @@ pip install -r requirements.txt bash data/download_and_preprocess_dataset_era5.sh --data era5 --input_dir /splits --output_dir data/era5 ``` -### Model Training +### Model Training for ERA5 data ```python -python scripts/train.py --input_dir data/era5 --dataset era5 --model savp --model_hparams_dict hparams/kth/ours_savp/model_hparams.json --output_dir logs/era5/ours_savp +python scripts/train_v2.py --input_dir data/era5 --dataset era5 --model savp --model_hparams_dict hparams/kth/ours_savp/model_hparams.json --output_dir logs/era5/ours_savp ``` + ### Model Evaluation - -\ \ No newline at end of file + diff --git a/results_test_samples/era5_size_64_64_3_norm_dup/ours_savp/Sample_Batch_id_0_Sample_1.mp4 b/results_test_samples/era5_size_64_64_3_norm_dup/ours_savp/Sample_Batch_id_0_Sample_1.mp4 new file mode 100644 index 0000000000000000000000000000000000000000..33402e208e069d2d17d968a933167f6b445efec1 Binary files /dev/null and b/results_test_samples/era5_size_64_64_3_norm_dup/ours_savp/Sample_Batch_id_0_Sample_1.mp4 differ