the derivative for 'cudnn_batch_norm_backward save_mean' is not implemented #4422
Description
loss_gp = ..
loss_gp.backward() # error
Having this error when BatchNorm + wgan_gp (second order) is used.
It's installed from master fec3d4.
{{ message }}