saahiluppal / catr

Image Captioning Using Transformer
Apache License 2.0
254 stars 53 forks source link

No simple way to test custom trained model #14

Closed Aa20475 closed 3 years ago

Aa20475 commented 3 years ago

It'll be good if we can provide model checkpoint file for prediction in predict.py

iamshant commented 2 years ago

After putting model checkpoint in configuration.py, remove the default argument for --v parser. before: parser.add_argument('--v', type=str, help='version', default='v3') after: parser.add_argument('--v', type=str, help='version')