From af95fe85f20a2807d0b8aa372a5d8a4bb3f744ac Mon Sep 17 00:00:00 2001 From: Francois Vieille Date: Mon, 11 Mar 2019 22:33:09 +0100 Subject: [PATCH] fix some code --- iss/models/DataLoader.py | 2 +- iss/models/__init__.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) 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