Fix keras model
@@ -2,10 +2,19 @@ from __future__ import print_function
import keras
-from keras.layers import Dense, Conv2D, BatchNormalization, Activation
-from keras.layers import AveragePooling2D, Input, Flatten
+from keras.layers import (
+ Activation,
+ AveragePooling2D,
+ BatchNormalization,
+ Dense,
+ Conv2D,
+ Input,
+ Flatten
+)
from keras.models import Model
from keras.optimizers import Adam
+from keras.regularizers import l2
+
from utils import lr_schedule
