Thank you so much , I fixed that , but I still getting same error , do you think the conv.weight size might be wrong ? I just wrote this function  gradient loss function for medical imaging in order to get less blurry image ( I am generating CT from MRI)
I appreciate if you help me