show model summary

haoyu-bu · Apr 27, 2018 · f9243ad · f9243ad
1 parent 9ffee15
commit f9243ad
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/wide_resnet.py b/wide_resnet.py
@@ -138,8 +138,15 @@ def __call__(self):
         predictions_a = Dense(units=101, kernel_initializer=self._weight_init, use_bias=self._use_bias,
                               kernel_regularizer=l2(self._weight_decay), activation="softmax",
                               name="pred_age")(flatten)
-
         model = Model(inputs=inputs, outputs=[predictions_g, predictions_a])
 
         return model
 
+
+def main():
+    model = WideResNet(64)()
+    model.summary()
+
+
+if __name__ == '__main__':
+    main()