Where "last_checkpoint" should be modified to reflect the absolute path of "model_final.pth"?

Wangt-CN / VC-R-CNN

[CVPR 2020] The official pytorch implementation of ``Visual Commonsense R-CNN''

MIT License

352 stars 61 forks source link

Where "last_checkpoint" should be modified to reflect the absolute path of "model_final.pth"? #6

Closed amanchadha closed 4 years ago

amanchadha commented 4 years ago

"2. Using our pretrained VC model on COCO

Here we also provide our pretrained VC model. You can put it into the model dictionary and set the last_checkpoint with the absolute path of model_final.pth. Then run the command:

python -m torch.distributed.launch --nproc_per_node=$NGPUS tools/test_net.py --config-file "path/to/config/file.yaml" TEST.IMS_PER_BA"

Can you please describe where "last_checkpoint" should be modified to reflect the absolute path of "model_final.pth"?

Wangt-CN commented 4 years ago

Sorry for the late reply. Actually if you train the model, the network would save the model in the set dictionary and create the 'last_checkpoint' file. For user who want directly use test file, I would upload this file later. Or you can just create one and the content is just the absolute path of model_final.pth.