Skip to content
Snippets Groups Projects
Commit 106b54be authored by Bing Gong's avatar Bing Gong
Browse files

Fix training issue

parent d73795fa
No related branches found
No related tags found
No related merge requests found
...@@ -6,9 +6,9 @@ ...@@ -6,9 +6,9 @@
#SBATCH --cpus-per-task=1 #SBATCH --cpus-per-task=1
#SBATCH --output=train_era5-out.%j #SBATCH --output=train_era5-out.%j
#SBATCH --error=train_era5-err.%j #SBATCH --error=train_era5-err.%j
#SBATCH --time=23:20:00 #SBATCH --time=00:20:00
#SBATCH --gres=gpu:1 #SBATCH --gres=gpu:1
#SBATCH --partition=gpus #SBATCH --partition=develgpus
#SBATCH --mail-type=ALL #SBATCH --mail-type=ALL
#SBATCH --mail-user=b.gong@fz-juelich.de #SBATCH --mail-user=b.gong@fz-juelich.de
##jutil env activate -p cjjsc42 ##jutil env activate -p cjjsc42
...@@ -24,5 +24,5 @@ module load cuDNN/7.5.1.10-CUDA-10.1.105 ...@@ -24,5 +24,5 @@ module load cuDNN/7.5.1.10-CUDA-10.1.105
#srun python scripts/train.py --input_dir data/kth --dataset kth --model savp --model_hparams_dict hparams/kth/ours_savp/model_hparams.json --output_dir logs/kth/ours_savp #srun python scripts/train.py --input_dir data/kth --dataset kth --model savp --model_hparams_dict hparams/kth/ours_savp/model_hparams.json --output_dir logs/kth/ours_savp
srun python ../scripts/train_v2.py --input_dir ../data/era5_size_64_64_3_3t_norm --dataset era5 --model savp --model_hparams_dict hparams/kth/ours_savp/model_hparams.json --output_dir ../logs/era5_size_64_64_3_3t_norm/ours_savp srun python ../scripts/train_v2.py --input_dir ../data/era5_size_64_64_3_3t_norm --dataset era5 --model savp --model_hparams_dict ../hparams/kth/ours_savp/model_hparams.json --output_dir ../logs/era5_size_64_64_3_3t_norm/ours_savp
#srun 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 #srun 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
...@@ -2,4 +2,3 @@ opencv-python ...@@ -2,4 +2,3 @@ opencv-python
scipy scipy
scikit-image scikit-image
pandas pandas
netCDF4
...@@ -4,17 +4,30 @@ from __future__ import print_function ...@@ -4,17 +4,30 @@ from __future__ import print_function
import argparse import argparse
import errno import errno
import json
import os import os
import math import math
import random
import cv2
import numpy as np
import tensorflow as tf import tensorflow as tf
import pickle
from random import seed
import random import random
import json import json
import numpy as np import numpy as np
#from six.moves import cPickle
import matplotlib import matplotlib
matplotlib.use('Agg') matplotlib.use('Agg')
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec import matplotlib.gridspec as gridspec
import matplotlib.animation as animation
import pandas as pd
import re
from video_prediction import datasets, models from video_prediction import datasets, models
from matplotlib.colors import LinearSegmentedColormap
#from matplotlib.ticker import MaxNLocator
#from video_prediction.utils.ffmpeg_gif import save_gif
from skimage.metrics import structural_similarity as ssim from skimage.metrics import structural_similarity as ssim
import pickle import pickle
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment