Epoch 1, Step 0:   Train Loss 3.0852, Val Loss 2.4835
Epoch 1, Step 50:  Train Loss 0.5300, Val Loss 0.4591
...
Epoch 5, Step 550: Train Loss 0.0246, Val Loss 0.0727
Epoch 5, Step 600: Train Loss 0.1635, Val Loss 0.1263

Checkpoint train accuracy: 95.0%
Checkpoint val accuracy:   97.5%
Training time: 641.97 minutes

        __
        __