jkpark0825 / EANet

29 stars 3 forks source link

pytorch dependencies #9

Open mkloftdyn opened 3 weeks ago

mkloftdyn commented 3 weeks ago

On which version of cuda/pytorch etc. did you successfully run the code?

dqj5182 commented 3 weeks ago

Thank you for your interest!

The versions that I have tested was Python 3.9.12, PyTorch 1.7.1, CUDA 10.1.