args

1 files changed, 2 insertions, 5 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index 14ce906..87f331c 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -11,16 +11,13 @@ class RecursiveDataset(BaseDataset):
         self.root = opt.dataroot    
 
         ### input A (label maps)
-        self.dir_A = opt.dataroot
-        self.A_paths = sorted(make_dataset(self.dir_A))
-
-        self.dataset_size = len(self.A_paths) 
+        self.dataset_size = 1000000
       
     def __getitem__(self, index):
         ### input A (label maps)
         A_path = os.path.join(self.opt.dataroot, "frame_{:05d}.png".format(index))
         if not os.path.exists(A_path):
-            # print()
+            print("{} doesn't exist, waiting for it".format(A_path))
             while not os.path.exists(A_path):
                 # print('sleeping for {}'.format(self.opt.poll_delay))
                 time.sleep(self.opt.poll_delay)