search

pudae / tensorflow-densenet

Tensorflow-DenseNet with ImageNet Pretrained Models
Apache License 2.0
169 stars 59 forks source link

training issues #14

Closed laukun closed 5 years ago

laukun commented 6 years ago

Hi, I want to ask how to determine the maximum number of training steps if fine-tune this pretrained densenet using my own dataset ?

insikk commented 6 years ago

Using Tensorboard, see how train and validation error drops. Set enough max iter num to see trends. When validation error starts to increase, you want to stop it there. It indicates over-fitting