From 0bb81d6fea7d05cf233106296bb2257bf053bea8 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 13:22:06 +0200 Subject: k --- data/__init__.py | 1 + 1 file changed, 1 insertion(+) (limited to 'data/__init__.py') 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 -- cgit v1.2.3-70-g09d2