From 919ba03dae6ca25efc2f0bafc5a027efc0fe18e1 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 14:25:28 +0200 Subject: k --- data/__init__.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'data') 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 -- cgit v1.2.3-70-g09d2