mirror of
https://github.com/prise6/smart-iss-posts
synced 2024-06-08 00:32:12 +02:00
fix some code
This commit is contained in:
parent
fe33b892bd
commit
af95fe85f2
|
@ -18,7 +18,7 @@ class ImageDataGeneratorWrapper:
|
|||
|
||||
def image_data_generator(self, config):
|
||||
self.datagen = ImageDataGenerator(
|
||||
rescale = 1./255
|
||||
rescale = 1./255
|
||||
)
|
||||
return self
|
||||
|
||||
|
|
|
@ -0,0 +1,3 @@
|
|||
|
||||
from .SimpleConvAutoEncoder import SimpleConvAutoEncoder
|
||||
from .SimpleAutoEncoder import SimpleAutoEncoder
|
Loading…
Reference in a new issue