TypeError: 'int' object is not callable

Po-Hsun-Su / pytorch-ssim

pytorch structural similarity (SSIM) loss

Other

1.89k stars 367 forks source link

Open NoviceCrom opened 4 years ago

NoviceCrom commented 4 years ago

when executing (, channel, , _) = img1.size()

JingyunLiang commented 4 years ago

Likely you calculate ssim value and replace it with the loss definition SSIM() somewhere. Just change the name to ssim_loss = SSIM().