Find matching configuration file for a model
Engin1980 opened this issue · 1 comments
Dear authors, thank you for your work and for your repo.
Is there a simple way how to find matching configuration .py file (from /configs folder) for a particular .pth file?
E.g., I am looking for the configuration files for models:
vfnet_r50_dcn_ms_2x_47.8.pth
vfnet_r2_101_dcn_ms_2x_51.1.pth
For ResNet 50 I have tried vrefnet_r50_fpn_1x_coco.py
and vrefnet_r50_fpn_mstrain_2x_coco.py
giving me mAP 0.91. For R2 I have tried vfnet_r2_101_fpn_mstrain_2x_coco.py
, got error 'pth file is not valid checkpoint file'.
Thank you for your answer in advance.
Best regards
MV
Hi @Engin1980, thanks for interesting in our work.
vfnet_r50_dcn_ms_2x_47.8.pth
-> vfnet_r50_fpn_mdconv_c3-c5_mstrain_2x_coco.py
vfnet_r2_101_dcn_ms_2x_51.1.pth
-> vfnet_r2_101_fpn_mdconv_c3-c5_mstrain_2x_coco.py
The rule is : dcn -> mdconv, ms -> mstrain.
By the way, if you use vfnet_r2_101_fpn_mdconv_c3-c5_mstrain_2x_coco.py and corresponding model file in inference, please comment delattr(self, 'conv2') in res2net.py.