all_results.json

{
    "epoch": 200.8,
    "test_loss": 10.003050804138184,
    "test_runtime": 1356.7841,
    "test_samples_per_second": 5.178,
    "test_steps_per_second": 0.027,
    "test_tse_ndup": 0.19662171045338356,
    "test_tse_nnof": 0.0,
    "test_tse_nnon": 0.0,
    "test_tse_time": 0.0,
    "test_tse_type": 0.07749211227763136,
    "train_loss": 3.3319753212356567,
    "train_runtime": 62203.6746,
    "train_samples_per_second": 205.776,
    "train_steps_per_second": 1.608,
    "tse_ndup": 0.004091020854021363,
    "tse_nnof": 0.0,
    "tse_nnon": 0.0,
    "tse_time": 0.0,
    "tse_type": 0.0003831938631525549
}