diff --git a/config.py b/config.py index 98068e8..c9bed7d 100644 --- a/config.py +++ b/config.py @@ -291,7 +291,7 @@ def make_dataset(self, path=None, **kwargs): image_size=self.img_size, **kwargs) elif self.data_name == 'bedroom256': - return Horse_lmdb(path=path or self.data_path, + return Bedroom_lmdb(path=path or self.data_path, image_size=self.img_size, **kwargs) elif self.data_name == 'celebalmdb':