The scaling with p of w at test time is moved to a scaling of 1/p at training time.
Both activations (train and test) will therefore still have the same expected values, but the “workload” was moved from test to training.
I think that’s probably the effect you are seeing.