fix some code

2019-03-11 22:33:09 +01:00 · 2019-03-11 22:33:09 +01:00 · af95fe85f2
parent fe33b892bd
commit af95fe85f2
2 changed files with 4 additions and 1 deletions
--- 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

--- a/iss/models/init.py
+++ b/iss/models/init.py
@ -0,0 +1,3 @@
+
+from .SimpleConvAutoEncoder import SimpleConvAutoEncoder
+from .SimpleAutoEncoder import SimpleAutoEncoder