I fixed the oversaturation by changing how i did the inv_norm. The fix me and @ZimoNitrome did can be seen in here, mixed precision still produces noise and fail to converge.
I fixed the oversaturation by changing how i did the inv_norm. The fix me and @ZimoNitrome did can be seen in here, mixed precision still produces noise and fail to converge.