Correct save location for trained model

[parismic]

It seems that the save path is incorrectly documented.

In the args the default dir is model_checkpoints and the README.md says models/bert/saves/Reuters/best_model.pt. I test it on SST-2 and the trained model was saved at model_checkpoints.

Or are these two different things?

See:

hedwig/models/bert/args.py

Line 11 in c602b82

parser.add_argument('--save-path', type=str, default=os.path.join('model_checkpoints', 'bert'))

[Ashutosh-Adhikari]

@parismic , yes that's an issue to be resolved. Thanks for pointing it out.