Updating code of the article for TorchVision's Multi-Weight Support API by datumbox · Pull Request #915 · pytorch/pytorch.github.io