Closed Yukun-Huang closed 5 years ago
In train.py, line 110 : varIn, varTar = varTar.float(), varIn.float()
varIn, varTar = varTar.float(), varIn.float()
I think the location of varIn and varTar should be exchanged.
Fixed
The same error occurs in line 134 remained unchanged.
In train.py, line 110 :
varIn, varTar = varTar.float(), varIn.float()
I think the location of varIn and varTar should be exchanged.