diff --git a/iss/models/DataLoader.py b/iss/models/DataLoader.py index 3c1bd47..8ba8e4f 100644 --- a/iss/models/DataLoader.py +++ b/iss/models/DataLoader.py @@ -18,7 +18,7 @@ class ImageDataGeneratorWrapper: def image_data_generator(self, config): self.datagen = ImageDataGenerator( - rescale = 1./255 + rescale = 1./255 ) return self diff --git a/iss/models/__init__.py b/iss/models/__init__.py index e69de29..cbc4cfa 100644 --- a/iss/models/__init__.py +++ b/iss/models/__init__.py @@ -0,0 +1,3 @@ + +from .SimpleConvAutoEncoder import SimpleConvAutoEncoder +from .SimpleAutoEncoder import SimpleAutoEncoder \ No newline at end of file