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

Update era5_dataset.py

parent 76539e52
No related branches found
No related tags found
No related merge requests found
Pipeline #46235 failed
...@@ -121,7 +121,7 @@ class ERA5Dataset(VarLenFeatureVideoDataset): ...@@ -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 # _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 dataset = dataset.prefetch(batch_size) # Bing: Take the data to buffer inorder to save the waiting time for GPU
return dataset return dataset
/V2
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment