Using astats while traning model is consuming memory.  * In the above picture i was traning a resnet model using this callback as the model sees more mini-batches it is comsuming more ram