Heng Wang 8a2433eacb Add model saving and loading to resnet50_trainer.py ... Summary: Script caffe2/caffe2/python/examples/resnet50_trainer.py can be used to train a ResNet-50 model with Imagenet data (or similar). However, currently the script does not actually save the model, so it is kind of useless. Task 1: After each Epoch, save the model in a file "<filename>_X.mdl' where X is the epoch number and <filename> is given as a command line parameter. By default, use "resnet50_model" as filename. Task 2: Add a functionality to restore the model from a previous file: - add a command line parameter "load_model", which user can use to specify a filename. - if this parameter is set, load the model parameters from the previous file Reviewed By: prigoyal Differential Revision: D4984340 fbshipit-source-id: 333e92679ba52a7effe9917fdfc2d55d652b868f		2017-05-05 10:08:37 -07:00
..
char_rnn.py	Fix char_rnn LSTM import	2017-04-20 22:46:19 -07:00
lmdb_create_example.py	doxygen python block added	2017-03-29 06:46:16 -07:00
resnet50_trainer.py	Add model saving and loading to resnet50_trainer.py	2017-05-05 10:08:37 -07:00