diff options
Diffstat (limited to 'data/__init__.py')
| -rw-r--r-- | data/__init__.py | 3 |
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 |
