From 87ae8bc78b116e02cadd80d14bb11a09bd949532 Mon Sep 17 00:00:00 2001 From: Bing Gong <b.gong@fz-juelich.de> Date: Wed, 23 Sep 2020 07:49:30 +0000 Subject: [PATCH] Update era5_dataset.py --- video_prediction_savp/video_prediction/datasets/era5_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/video_prediction_savp/video_prediction/datasets/era5_dataset.py b/video_prediction_savp/video_prediction/datasets/era5_dataset.py index 4d80aacf..8cc9b757 100644 --- a/video_prediction_savp/video_prediction/datasets/era5_dataset.py +++ b/video_prediction_savp/video_prediction/datasets/era5_dataset.py @@ -121,7 +121,7 @@ class ERA5Dataset(VarLenFeatureVideoDataset): # _parser, batch_size, drop_remainder=True, num_parallel_calls=num_parallel_calls)) # Bing: Parallel data mapping, num_parallel_calls normally depends on the hardware, however, normally should be equal to be the usalbe number of CPUs dataset = dataset.prefetch(batch_size) # Bing: Take the data to buffer inorder to save the waiting time for GPU return dataset -/V2 + -- GitLab