def _train_update(device, step, loss, tracker, epoch, writer):
    test_utils.print_training_update(device, step, loss.item(), tracker.rate(), tracker.global_rate(), epoch, summary_writer=writer)
def _train_update(device, step, loss, tracker, epoch, writer):
    test_utils.print_training_update(device, step, loss.item(), tracker.rate(), tracker.global_rate(), epoch, summary_writer=writer)