shiming-chen/FREE

HELP: I run the code, but got the following error,can you help me?

ChocalateWZLH opened this issue · 0 comments

Traceback (most recent call last):
File "C:/Users/Desktop/FREE-main/train_free.py", line 237, in
fake = netG(z, c=input_attv)
File "D:\PyCharm\Anaconda\envs\Python 3_8\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "C:\Users\14367\Desktop\FREE-main\model.py", line 59, in forward
x1 = self.lrelu(self.fc1(z))
File "D:\PyCharm\Anaconda\envs\Python 3_8\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "D:\PyCharm\Anaconda\envs\Python 3_8\lib\site-packages\torch\nn\modules\linear.py", line 114, in forward
return F.linear(input, self.weight, self.bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (64x1336 and 2048x4096)