Track model training progress with step and loss details
Track and display model training progress in real time