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