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