Closed rookiecm closed 2 years ago
Thanks for sharing the code. Would you please specify the pytorch version?
We use python 3.8.12, pytorch 1.7.1, and cuda 10.1
Thanks for sharing the code. Would you please specify the pytorch version?