YngMgC

Seem that the version of pytorch dismatch with the gpu

3

I run the code on an nvidia gpu A6000, I met the following error: Traceback (most recent call last): File "demo.py", line 127, in input_ = F.pad(input_, (0,padw,0,padh), 'reflect') File...