rescale stats doesn't fit in the GPU memory with the model at the same time

[mustafaghali]

for a given batch 32 the stats(means, mins, max) tensors with the gan model give's Cuda out of memory error, it works fine when I reduce the batch size to 26...