search

lindawangg / COVID-Net

COVID-Net Open Source Initiative
Other
1.15k stars 480 forks source link

How to train network from scratch? #60

Closed shahinsaleh closed 4 years ago

shahinsaleh commented 4 years ago

Hello,

I want to train you network from scratch and not from the pre-trained "Small" and "Large" models.

Could you please describe how to do that since I want to compare the effects of data augmentation vs dropout vs no augmentation & no dropout?

Thank you!

haydengunraj commented 4 years ago

You can use the metagraph files for the pre-trained models and comment out the line(s) where the pre-trained weights are loaded. For example, in train_tf.py you would comment out line 83: saver.restore(sess, os.path.join(args.weightspath, args.ckptname)).