Skip to content

Latest commit

 

History

History
10 lines (8 loc) · 592 Bytes

File metadata and controls

10 lines (8 loc) · 592 Bytes

transfer-learning-with-tensorflow

retrain_v5.py is a customised version of retrain.py for training on image data

Customizations Done

  • Added 2 dense(Fully connected) layers followed by a SoftMax layer.
  • Added dropout to both the dense layers.
  • Added batch_normalization for dense layers
  • Replaced Gradient-descent with Gradient-descent with Momentum
  • Added live streamed confusion Matrix