all_results.json

{
    "epoch": 168.92,
    "test_loss": 10.012152671813965,
    "test_runtime": 1628.4473,
    "test_samples_per_second": 5.199,
    "test_steps_per_second": 0.028,
    "test_tse_ndup": 0.17967297977995939,
    "test_tse_nnof": 0.0,
    "test_tse_nnon": 0.0,
    "test_tse_time": 0.20978844964272034,
    "test_tse_type": 0.04320047391956446,
    "train_loss": 3.096821060371399,
    "train_runtime": 63835.4946,
    "train_samples_per_second": 200.515,
    "train_steps_per_second": 1.567,
    "tse_ndup": 0.003601619428289543,
    "tse_nnof": 0.0,
    "tse_nnon": 0.0,
    "tse_time": 0.052002719382478045,
    "tse_type": 0.0006408855316656025
}