In the get_crit_loss function , if I define the critic loss as

crit_loss = ( crit_real_pred.mean() - crit_fake_pred.mean() ) + (c_lambda * gp)
which is according to the min(g) max(c) equation, I get assertion error

whereas if I interchange the first 2 terms as
crit_fake_pred.mean() - crit_real_pred.mean()
I get success.