fine-tune

Author: andrewferlitsch

zoo/resnet/resnet34.py

@@ -95,7 +95,7 @@ def classifier(x, n_classes):
     x = GlobalAveragePooling2D()(x)
 
     # Final Dense Outputting Layer for the outputs
-    outputs = Dense(n_classes, activation='softmax')(x)
+    outputs = Dense(n_classes, activation='softmax', kernel_initializer='he_normal')(x)
     return outputs
 
 # The input tensor

zoo/resnet/resnet_cifar10.py

@@ -140,7 +140,7 @@ def classifier(x, n_classes):
     x = Flatten()(x)
 
     # Final Dense Outputting Layer 
-    outputs = Dense(n_classes, activation='softmax')(x)
+    outputs = Dense(n_classes, activation='softmax', kernel_initializer='he_normal')(x)
     return outputs
 
 #-------------------

zoo/resnet/resnet_cifar10_v2.py

@@ -140,7 +140,7 @@ def classifier(x, n_classes):
     x = Flatten()(x)
 
     # Final Dense Outputting Layer 
-    outputs = Dense(n_classes, activation='softmax')(x)
+    outputs = Dense(n_classes, activation='softmax', kernel_initializer='he_normal')(x)
     return outputs
 
 #-------------------