part-affinity icon indicating copy to clipboard operation
part-affinity copied to clipboard

Published 20 hours ago • verified publisher icon NiteshBharadwaj

pretrained model

Open zkloveai opened this issue 6 years ago • 3 comments

First of all, thank you for your excellent work. can you provide pre-training model？ thanks a lot!

zkloveai avatar Jan 09 '19 09:01 zkloveai

Thank you very much! Here is a link for pre-trained model: https://drive.google.com/drive/folders/1PAh00CPORDpwKgBNYgVYppDC21knu46n

NiteshBharadwaj avatar Jan 17 '19 15:01 NiteshBharadwaj

We're training a model of our own, I can upload it when done. How many epochs is good here? 20?

maxkazmsft avatar Jan 23 '19 23:01 maxkazmsft

Thank you for your interest. The above model is trained for 80 epochs. With learn rate drops at 40th and 60th epoch. So this may be close to the best model from this codebase. To improve precision, you might want to look at heatmap creation once as I am not sure if my approach is best, and probably try a more aggressive data augmentation.

NiteshBharadwaj avatar Jan 24 '19 08:01 NiteshBharadwaj