Fixed model suffix .torch.torch in convert_vissl_to_torchvision.py

facebookresearch / vissl

VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.

https://vissl.ai

MIT License

3.26k stars 334 forks source link

Closed rgeirhos closed 3 years ago

rgeirhos commented 3 years ago

When following the instructions from https://github.com/facebookresearch/vissl/blob/main/MODEL_ZOO.md#converting-vissl-to-torchvision, the resulting converted model will have a ".torch.torch" suffix. This is avoided by checking whether the user has provided a .torch suffix for the model name.

facebook-github-bot commented 3 years ago

@iseessel has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.