fix: not compatible with the new versions of torch and torchvision
Merge pull request #215 from breezedeus/dev fix: not compatible with the new versions of torch and torchvision
Merge pull request #215 from breezedeus/dev fix: not compatible with the new versions of torch and torchvision