Fix torchvision 0.17 compatibility issue#54

Atry wants to merge 2 commits intoyangxy:mainyangxy/PASD:mainfrom

Atry:torchvision017Atry/PASD:torchvision017Copy head branch name to clipboard