diff --git a/tutorials/day4/tutorial/ResNet.py b/tutorials/day4/tutorial/ResNet.py index b92ed5ea6ebff2824e2bc52719dc125f520d7f76..9426af7baaf4fe15f1bcdcc81d5c879b09ace7f4 100644 --- a/tutorials/day4/tutorial/ResNet.py +++ b/tutorials/day4/tutorial/ResNet.py @@ -2,7 +2,6 @@ from tensorflow.keras import layers from tensorflow.keras.layers import Dense from tensorflow.keras.layers import Conv2D from tensorflow.keras.layers import AveragePooling2D -from tensorflow.keras.layers import Flatten from tensorflow.keras.layers import MaxPool2D from tensorflow.keras.layers import Input from tensorflow.keras.layers import BatchNormalization @@ -72,7 +71,6 @@ def ResNet(classes, name, input_shape, block_layers_num, weight_decay): for i in range(block_layers_num - 1): x = ResidualBlock(x, filters=64, kernel_size=(3, 3), weight_decay=weight_decay, downsample=False) x = GlobalAveragePooling2D()(x) - x = Flatten()(x) x = Dense(classes, activation='softmax')(x) model = Model(input, x, name=name) return model