Hi,
It seems that one of variables with size [64, 2] is changed after loss computation.
Is this a FC layer weights?
Also, did you try to replace every inplace operation?
a += 1 # inplace addition
a = a+1 # normal addition
Hi,
It seems that one of variables with size [64, 2] is changed after loss computation.
Is this a FC layer weights?
Also, did you try to replace every inplace operation?
a += 1 # inplace addition
a = a+1 # normal addition