thanks I’m aware of this. My real concern is, is it correct to normalize by dividing on the .data or no_grad max of the variable ?
Knowing that the max is taking across the fake_features and real_features and giving the fact that real_features has no gradient.