Train.py code

[Geonhee-LEE]

Thanks for your contributions.
I want to try transfer learning with new datasets. Would you upload the train.py on this repo or send me the file?