From 385635fe206624a9538f8bd6dd96cc448455fafd Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 21 Aug 2023 16:34:26 -0500
Subject: [PATCH] snapshot...
---
modules/GSOC/E2_ESRGAN/lib/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py
index d9959205..a9822846 100644
--- a/modules/GSOC/E2_ESRGAN/lib/dataset.py
+++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py
@@ -155,7 +155,7 @@ class OpdNpyDataset:
# dataset = dataset.cache()
# dataset = dataset.prefetch(buffer_size=1)
- file_idxs = np.arange(len(self.num_files))
+ file_idxs = np.arange(self.num_files)
dataset = tf.data.Dataset.from_tensor_slices(list(file_idxs))
dataset = dataset.shuffle(2000, reshuffle_each_iteration=True)
dataset = dataset.batch(batch_size)
--
GitLab