From b5666fa2309b65bce9d251fd25bfebf9a972c4ab Mon Sep 17 00:00:00 2001
From: Mehdi Cherti <mehdicherti@gmail.com>
Date: Mon, 22 Mar 2021 18:17:11 +0100
Subject: [PATCH] remove Flatten layer in resnet
---
tutorials/day4/tutorial/ResNet.py | 2 --
1 file changed, 2 deletions(-)
diff --git a/tutorials/day4/tutorial/ResNet.py b/tutorials/day4/tutorial/ResNet.py
index b92ed5e..9426af7 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
--
GitLab