Specify torch and torchvision requirements

jiangtaoxie / fast-MPN-COV

@CVPR2018: Efficient unrolling iterative matrix square-root normalized ConvNets, implemented by PyTorch (and code of B-CNN,Compact bilinear pooling etc.) for training from scratch & finetuning.

http://peihuali.org/iSQRT-COV/index.html

MIT License

270 stars 56 forks source link

Specify torch and torchvision requirements #26

Closed akindofyoga closed 3 years ago

akindofyoga commented 3 years ago

The code fails on newer versions of torch due to this error: https://github.com/pytorch/examples/issues/836

Specifying a maximum version number for torch and torchvision will prevent people from running into this issue.