From d67b2a46d9de205d87bef396ec1d917c8923525f Mon Sep 17 00:00:00 2001 From: John Schreck Date: Fri, 20 Dec 2024 13:54:10 -0700 Subject: [PATCH] Updated batches_per_epoch method --- credit/datasets/era5_multistep_batcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/credit/datasets/era5_multistep_batcher.py b/credit/datasets/era5_multistep_batcher.py index f13ca28..bbf7a3c 100644 --- a/credit/datasets/era5_multistep_batcher.py +++ b/credit/datasets/era5_multistep_batcher.py @@ -337,7 +337,7 @@ def set_epoch(self, epoch): self.initialize_batch() def batches_per_epoch(self): - return math.ceil((self.batch_indices) / self.batch_size) + return math.ceil(len(self.batch_indices) / self.batch_size) def __getitem__(self, _): """