summaryrefslogtreecommitdiff
path: root/data
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-04-18 13:22:06 +0200
committerJules Laplace <julescarbon@gmail.com>2018-04-18 13:22:06 +0200
commit0bb81d6fea7d05cf233106296bb2257bf053bea8 (patch)
tree5edb5ac16489d2c97bf2f3e8d3c36bef6be95c75 /data
parentb601d5cf5b6198da1d777dc218c2bf12627b6064 (diff)
k
Diffstat (limited to 'data')
-rw-r--r--data/__init__.py1
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