diff --git a/video_prediction_savp/video_prediction/datasets/era5_dataset.py b/video_prediction_savp/video_prediction/datasets/era5_dataset.py index 4d80aacf701f74aa03a948765086905b386c6bb5..8cc9b757bd86479a710dd67dc5652ffb5d3e94fd 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 +