diff options
| -rw-r--r-- | data/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/data/__init__.py b/data/__init__.py index 420c1f8..381ec59 100644 --- a/data/__init__.py +++ b/data/__init__.py @@ -68,6 +68,7 @@ class RecursiveDatasetDataLoader(BaseDataLoader): def initialize(self, opt): #BaseDataLoader.initialize(self, opt) + self.opt = opt self.dataset = CreateDataset(opt) self.dataloader = self.dataset |