deblan-mirror/smart-iss-posts
1
0
Fork 0
mirror of https://github.com/prise6/smart-iss-posts synced 2024-05-03 14:13:10 +02:00
Code Issues Releases Wiki Activity

fix some code

Browse source
This commit is contained in:
Francois Vieille 2019-03-11 22:33:09 +01:00
parent fe33b892bd
commit af95fe85f2
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
iss/models/DataLoader.py
View file

@ -18,7 +18,7 @@ class ImageDataGeneratorWrapper:
def image_data_generator(self, config): def image_data_generator(self, config):
self.datagen = ImageDataGenerator( self.datagen = ImageDataGenerator(
rescale = 1./255 rescale = 1./255
) )
return self return self

3
iss/models/__init__.py
View file

@ -0,0 +1,3 @@
from .SimpleConvAutoEncoder import SimpleConvAutoEncoder
from .SimpleAutoEncoder import SimpleAutoEncoder