I don’t know why when I set the affine=True of BatchNorm2d, it will occur error : RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
But when I set the affine=False of BatchNorm2d , the error will not appear.
My code is here, anybody can help me ?