wider support for pytorch version

Open brando90 opened this issue 3 years ago • 1 comments

pip install datasets torch==1.8.0 tqdm torchvision==0.9.0

this seems VERY restrictive :(

Feb 09 '23 00:02 brando90

We found that sometimes the results differ when using different PyTorch versions, so we implicitly set the version for others to reproduce the numbers in the paper. The code should run fine with other nearby versions too.

Apr 18 '23 17:04 ylsung