Make ViT work for older torch versions

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification

https://classyvision.ai

MIT License

1.59k stars 277 forks source link

Closed mannatsingh closed 3 years ago

mannatsingh commented 3 years ago

Summary: Use a copy-pasted implementation of trunc_normal from PT 1.7 for older versions. Fixes https://github.com/facebookresearch/ClassyVision/issues/683

Differential Revision: D25982616

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25982616

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25982616