summaryrefslogtreecommitdiff
path: root/data/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'data/__init__.py')
-rw-r--r--data/__init__.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/data/__init__.py b/data/__init__.py
index 0c7a752..9d34e6f 100644
--- a/data/__init__.py
+++ b/data/__init__.py
@@ -76,6 +76,9 @@ class RecursiveDatasetDataLoader(BaseDataLoader):
shuffle=not opt.serial_batches,
num_workers=int(opt.nThreads))
+ def append(self, arg):
+ self.dataset.append(arg)
+
def load_data(self):
return self