easier to call

This commit is contained in:
Francois Vieille 2019-03-24 17:58:27 +01:00
parent 3e2147f146
commit 37b7a1685d
1 changed files with 2 additions and 0 deletions

View File

@ -5,3 +5,5 @@ from .SimpleConvAutoEncoder import SimpleConvAutoEncoder
from .SimpleAutoEncoder import SimpleAutoEncoder
from .VariationalAutoEncoder import VarAutoEncoder
from .VariationalConvAutoEncoder import VarConvAutoEncoder
from .ModelTrainer import ModelTrainer
from .DataLoader import ImageDataGeneratorWrapper