rrm workspace: TRM/HRM/SRM code, Maze dataset, dynamical-analysis pipelineHEADmain

Curated export for clone-and-run Maze training (2x A6000) + diagnostics.
trm/hrm pretrain.py carry trajectory-augmentation code (backward-compatible).
Heavy artifacts (checkpoints/wandb/npz) gitignored; see PROVENANCE.md.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

1 files changed, 60099 insertions, 0 deletions

diff --git a/research/flossing/step9_A_hrm_single_perturb_sigma1e-3_26040_10k.json b/research/flossing/step9_A_hrm_single_perturb_sigma1e-3_26040_10k.json
new file mode 100644
index 0000000..eeb87be
--- /dev/null
+++ b/research/flossing/step9_A_hrm_single_perturb_sigma1e-3_26040_10k.json
@@ -0,0 +1,60099 @@
+{
+  "args": {
+    "model": "hrm",
+    "ckpt_root": "/home/yurenh2/rrm/hrm/checkpoints/Sudoku-extreme-1k-aug-1000 ACT-torch/HierarchicalReasoningModel_ACTV1 righteous-python",
+    "ckpt_name": "step_26040",
+    "mode": "single_perturbed_ce",
+    "train_steps": 10000,
+    "batch_size": 8,
+    "lr": 1e-05,
+    "noise_std": 0.001,
+    "n_trajectories": 4,
+    "perturb": "both",
+    "seed": 42,
+    "eval_every": 1000,
+    "eval_n": 512,
+    "eval_batch_size": 32,
+    "out": "step9_A_hrm_single_perturb_sigma1e-3_26040_10k.json"
+  },
+  "initial_acc": 0.517578125,
+  "initial_tok_acc": 0.8276668595679012,
+  "steps": [
+    {
+      "train_step": 1,
+      "loss": 0.5722061223623729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5722061223623729
+    },
+    {
+      "train_step": 2,
+      "loss": 0.45929683693608303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45929683693608303
+    },
+    {
+      "train_step": 3,
+      "loss": 0.7136228880623771,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7136228880623771
+    },
+    {
+      "train_step": 4,
+      "loss": 0.5846900052808484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5846900052808484
+    },
+    {
+      "train_step": 5,
+      "loss": 0.6159796798271049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6159796798271049
+    },
+    {
+      "train_step": 6,
+      "loss": 0.8053403153020356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8053403153020356
+    },
+    {
+      "train_step": 7,
+      "loss": 0.5961508115316996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5961508115316996
+    },
+    {
+      "train_step": 8,
+      "loss": 0.6196548952784331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6196548952784331
+    },
+    {
+      "train_step": 9,
+      "loss": 0.762387158184892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.762387158184892
+    },
+    {
+      "train_step": 10,
+      "loss": 0.8358805799485152,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8358805799485152
+    },
+    {
+      "train_step": 11,
+      "loss": 0.6539895413955588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6539895413955588
+    },
+    {
+      "train_step": 12,
+      "loss": 0.5792601904760699,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5792601904760699
+    },
+    {
+      "train_step": 13,
+      "loss": 0.5065058153906429,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5065058153906429
+    },
+    {
+      "train_step": 14,
+      "loss": 0.6890501395277147,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6890501395277147
+    },
+    {
+      "train_step": 15,
+      "loss": 0.5644083294917321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5644083294917321
+    },
+    {
+      "train_step": 16,
+      "loss": 0.6949949238567923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6949949238567923
+    },
+    {
+      "train_step": 17,
+      "loss": 0.576666080384253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576666080384253
+    },
+    {
+      "train_step": 18,
+      "loss": 0.7292916730856105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7292916730856105
+    },
+    {
+      "train_step": 19,
+      "loss": 0.5703042803776484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5703042803776484
+    },
+    {
+      "train_step": 20,
+      "loss": 0.4783099817866219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4783099817866219
+    },
+    {
+      "train_step": 21,
+      "loss": 0.6146882760349935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6146882760349935
+    },
+    {
+      "train_step": 22,
+      "loss": 0.6072644678197452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6072644678197452
+    },
+    {
+      "train_step": 23,
+      "loss": 0.540833786776496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.540833786776496
+    },
+    {
+      "train_step": 24,
+      "loss": 0.5375143991421933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5375143991421933
+    },
+    {
+      "train_step": 25,
+      "loss": 0.8039783067309711,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8039783067309711
+    },
+    {
+      "train_step": 26,
+      "loss": 0.8112540154979432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8112540154979432
+    },
+    {
+      "train_step": 27,
+      "loss": 0.7451552765219742,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7451552765219742
+    },
+    {
+      "train_step": 28,
+      "loss": 0.42200510131503044,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42200510131503044
+    },
+    {
+      "train_step": 29,
+      "loss": 0.8204330355794491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8204330355794491
+    },
+    {
+      "train_step": 30,
+      "loss": 0.7789162934586004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7789162934586004
+    },
+    {
+      "train_step": 31,
+      "loss": 0.6650838683726834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6650838683726834
+    },
+    {
+      "train_step": 32,
+      "loss": 0.6835378991721058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6835378991721058
+    },
+    {
+      "train_step": 33,
+      "loss": 0.6351421614369134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6351421614369134
+    },
+    {
+      "train_step": 34,
+      "loss": 0.5439016161712796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5439016161712796
+    },
+    {
+      "train_step": 35,
+      "loss": 0.6973728548940465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6973728548940465
+    },
+    {
+      "train_step": 36,
+      "loss": 0.4158837589658385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4158837589658385
+    },
+    {
+      "train_step": 37,
+      "loss": 0.6376344143818163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6376344143818163
+    },
+    {
+      "train_step": 38,
+      "loss": 0.8071378941171534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8071378941171534
+    },
+    {
+      "train_step": 39,
+      "loss": 0.5133664982865878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5133664982865878
+    },
+    {
+      "train_step": 40,
+      "loss": 0.7466921863338265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7466921863338265
+    },
+    {
+      "train_step": 41,
+      "loss": 0.7271783367264417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7271783367264417
+    },
+    {
+      "train_step": 42,
+      "loss": 0.47152319017322614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47152319017322614
+    },
+    {
+      "train_step": 43,
+      "loss": 0.7143275830304626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7143275830304626
+    },
+    {
+      "train_step": 44,
+      "loss": 0.7406699357731598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7406699357731598
+    },
+    {
+      "train_step": 45,
+      "loss": 0.8906864903275457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8906864903275457
+    },
+    {
+      "train_step": 46,
+      "loss": 0.6564614441873039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6564614441873039
+    },
+    {
+      "train_step": 47,
+      "loss": 0.6460024496726704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6460024496726704
+    },
+    {
+      "train_step": 48,
+      "loss": 0.6273640262810699,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6273640262810699
+    },
+    {
+      "train_step": 49,
+      "loss": 0.6011363955263954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6011363955263954
+    },
+    {
+      "train_step": 50,
+      "loss": 0.5777706586710252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5777706586710252
+    },
+    {
+      "train_step": 51,
+      "loss": 0.6029018471136175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6029018471136175
+    },
+    {
+      "train_step": 52,
+      "loss": 0.811207544945964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.811207544945964
+    },
+    {
+      "train_step": 53,
+      "loss": 0.5445926930425956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5445926930425956
+    },
+    {
+      "train_step": 54,
+      "loss": 0.6125066285625319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6125066285625319
+    },
+    {
+      "train_step": 55,
+      "loss": 0.6058789457178287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6058789457178287
+    },
+    {
+      "train_step": 56,
+      "loss": 0.609897268747426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.609897268747426
+    },
+    {
+      "train_step": 57,
+      "loss": 0.5687035180783825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5687035180783825
+    },
+    {
+      "train_step": 58,
+      "loss": 0.7260088988838804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7260088988838804
+    },
+    {
+      "train_step": 59,
+      "loss": 0.6485419943767733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6485419943767733
+    },
+    {
+      "train_step": 60,
+      "loss": 0.45919196508545534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45919196508545534
+    },
+    {
+      "train_step": 61,
+      "loss": 0.6681042694042862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6681042694042862
+    },
+    {
+      "train_step": 62,
+      "loss": 0.5665617091337505,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5665617091337505
+    },
+    {
+      "train_step": 63,
+      "loss": 0.47752025241241597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47752025241241597
+    },
+    {
+      "train_step": 64,
+      "loss": 0.7238736317528336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7238736317528336
+    },
+    {
+      "train_step": 65,
+      "loss": 0.6171409915672765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6171409915672765
+    },
+    {
+      "train_step": 66,
+      "loss": 0.8298195224367432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8298195224367432
+    },
+    {
+      "train_step": 67,
+      "loss": 0.6166602377651421,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6166602377651421
+    },
+    {
+      "train_step": 68,
+      "loss": 0.4812461652811265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4812461652811265
+    },
+    {
+      "train_step": 69,
+      "loss": 0.7146309416593143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7146309416593143
+    },
+    {
+      "train_step": 70,
+      "loss": 0.7971038143885922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7971038143885922
+    },
+    {
+      "train_step": 71,
+      "loss": 0.6909741092693128,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6909741092693128
+    },
+    {
+      "train_step": 72,
+      "loss": 0.7589538494261404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7589538494261404
+    },
+    {
+      "train_step": 73,
+      "loss": 0.4620726634723172,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4620726634723172
+    },
+    {
+      "train_step": 74,
+      "loss": 0.5549923002259853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5549923002259853
+    },
+    {
+      "train_step": 75,
+      "loss": 0.6912912329103322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6912912329103322
+    },
+    {
+      "train_step": 76,
+      "loss": 0.6305897553738488,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6305897553738488
+    },
+    {
+      "train_step": 77,
+      "loss": 0.8017616418579723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8017616418579723
+    },
+    {
+      "train_step": 78,
+      "loss": 0.6287799485776839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6287799485776839
+    },
+    {
+      "train_step": 79,
+      "loss": 0.8585216266608756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8585216266608756
+    },
+    {
+      "train_step": 80,
+      "loss": 0.7472014873513797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7472014873513797
+    },
+    {
+      "train_step": 81,
+      "loss": 0.633687343550376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.633687343550376
+    },
+    {
+      "train_step": 82,
+      "loss": 0.48201960235907726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48201960235907726
+    },
+    {
+      "train_step": 83,
+      "loss": 0.6035286264276332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6035286264276332
+    },
+    {
+      "train_step": 84,
+      "loss": 0.6576632411959986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6576632411959986
+    },
+    {
+      "train_step": 85,
+      "loss": 0.7134882902522685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7134882902522685
+    },
+    {
+      "train_step": 86,
+      "loss": 0.7065349283723992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7065349283723992
+    },
+    {
+      "train_step": 87,
+      "loss": 0.7438146790668424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7438146790668424
+    },
+    {
+      "train_step": 88,
+      "loss": 0.6218000448217265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6218000448217265
+    },
+    {
+      "train_step": 89,
+      "loss": 0.8700120794860837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8700120794860837
+    },
+    {
+      "train_step": 90,
+      "loss": 0.7371303924423656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7371303924423656
+    },
+    {
+      "train_step": 91,
+      "loss": 0.48921372567298865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48921372567298865
+    },
+    {
+      "train_step": 92,
+      "loss": 0.6237789333112076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6237789333112076
+    },
+    {
+      "train_step": 93,
+      "loss": 0.7269920606571815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7269920606571815
+    },
+    {
+      "train_step": 94,
+      "loss": 0.5547152452282261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5547152452282261
+    },
+    {
+      "train_step": 95,
+      "loss": 0.7217088317274487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7217088317274487
+    },
+    {
+      "train_step": 96,
+      "loss": 0.6328883667811298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6328883667811298
+    },
+    {
+      "train_step": 97,
+      "loss": 0.7055319343952777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7055319343952777
+    },
+    {
+      "train_step": 98,
+      "loss": 0.4313197340766451,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4313197340766451
+    },
+    {
+      "train_step": 99,
+      "loss": 0.7304578536251888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7304578536251888
+    },
+    {
+      "train_step": 100,
+      "loss": 0.6138641362041768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6138641362041768
+    },
+    {
+      "train_step": 101,
+      "loss": 0.5376574328140846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5376574328140846
+    },
+    {
+      "train_step": 102,
+      "loss": 0.561892075834022,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.561892075834022
+    },
+    {
+      "train_step": 103,
+      "loss": 0.6846742630473168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6846742630473168
+    },
+    {
+      "train_step": 104,
+      "loss": 0.6579880229182533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6579880229182533
+    },
+    {
+      "train_step": 105,
+      "loss": 0.7741267091200076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7741267091200076
+    },
+    {
+      "train_step": 106,
+      "loss": 0.6451696290434137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6451696290434137
+    },
+    {
+      "train_step": 107,
+      "loss": 0.6687125051619672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6687125051619672
+    },
+    {
+      "train_step": 108,
+      "loss": 0.7018668863775546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7018668863775546
+    },
+    {
+      "train_step": 109,
+      "loss": 0.7266023096352303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7266023096352303
+    },
+    {
+      "train_step": 110,
+      "loss": 0.7880995456375552,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7880995456375552
+    },
+    {
+      "train_step": 111,
+      "loss": 0.5885450816840674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5885450816840674
+    },
+    {
+      "train_step": 112,
+      "loss": 0.7242709911089705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7242709911089705
+    },
+    {
+      "train_step": 113,
+      "loss": 0.7276598558299501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7276598558299501
+    },
+    {
+      "train_step": 114,
+      "loss": 0.6702200374125016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6702200374125016
+    },
+    {
+      "train_step": 115,
+      "loss": 0.5714846981551598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5714846981551598
+    },
+    {
+      "train_step": 116,
+      "loss": 0.5657578388945544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5657578388945544
+    },
+    {
+      "train_step": 117,
+      "loss": 0.6794507392210997,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6794507392210997
+    },
+    {
+      "train_step": 118,
+      "loss": 0.7580783655818277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7580783655818277
+    },
+    {
+      "train_step": 119,
+      "loss": 0.5005178006213803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5005178006213803
+    },
+    {
+      "train_step": 120,
+      "loss": 0.6131527045669319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6131527045669319
+    },
+    {
+      "train_step": 121,
+      "loss": 0.6324643782598521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6324643782598521
+    },
+    {
+      "train_step": 122,
+      "loss": 0.7369920231227449,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7369920231227449
+    },
+    {
+      "train_step": 123,
+      "loss": 0.7550563734895801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7550563734895801
+    },
+    {
+      "train_step": 124,
+      "loss": 0.8355422972692994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8355422972692994
+    },
+    {
+      "train_step": 125,
+      "loss": 0.9641014244073305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9641014244073305
+    },
+    {
+      "train_step": 126,
+      "loss": 0.6570065495698784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6570065495698784
+    },
+    {
+      "train_step": 127,
+      "loss": 0.519611333982759,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.519611333982759
+    },
+    {
+      "train_step": 128,
+      "loss": 0.5298224156322244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5298224156322244
+    },
+    {
+      "train_step": 129,
+      "loss": 0.4764743872084383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4764743872084383
+    },
+    {
+      "train_step": 130,
+      "loss": 0.39000958369883904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39000958369883904
+    },
+    {
+      "train_step": 131,
+      "loss": 0.6638946318854171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6638946318854171
+    },
+    {
+      "train_step": 132,
+      "loss": 0.46203611574062353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46203611574062353
+    },
+    {
+      "train_step": 133,
+      "loss": 0.550199076763911,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.550199076763911
+    },
+    {
+      "train_step": 134,
+      "loss": 0.6798986045399134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6798986045399134
+    },
+    {
+      "train_step": 135,
+      "loss": 0.5736990149302912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5736990149302912
+    },
+    {
+      "train_step": 136,
+      "loss": 0.6471452333677663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6471452333677663
+    },
+    {
+      "train_step": 137,
+      "loss": 0.7298323882834536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7298323882834536
+    },
+    {
+      "train_step": 138,
+      "loss": 0.7599284843382429,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7599284843382429
+    },
+    {
+      "train_step": 139,
+      "loss": 0.6136511745852353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6136511745852353
+    },
+    {
+      "train_step": 140,
+      "loss": 0.6836320579904285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6836320579904285
+    },
+    {
+      "train_step": 141,
+      "loss": 0.8281159656218879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8281159656218879
+    },
+    {
+      "train_step": 142,
+      "loss": 0.8054997584217882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8054997584217882
+    },
+    {
+      "train_step": 143,
+      "loss": 0.4652359743879645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4652359743879645
+    },
+    {
+      "train_step": 144,
+      "loss": 0.5233011893710943,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5233011893710943
+    },
+    {
+      "train_step": 145,
+      "loss": 0.8200478110886663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8200478110886663
+    },
+    {
+      "train_step": 146,
+      "loss": 0.7987874050820242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7987874050820242
+    },
+    {
+      "train_step": 147,
+      "loss": 0.5678808581595848,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5678808581595848
+    },
+    {
+      "train_step": 148,
+      "loss": 0.6470012710252948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6470012710252948
+    },
+    {
+      "train_step": 149,
+      "loss": 0.6786582403965381,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6786582403965381
+    },
+    {
+      "train_step": 150,
+      "loss": 0.4981910048632104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4981910048632104
+    },
+    {
+      "train_step": 151,
+      "loss": 0.5454484390457278,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5454484390457278
+    },
+    {
+      "train_step": 152,
+      "loss": 0.5279421045401805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5279421045401805
+    },
+    {
+      "train_step": 153,
+      "loss": 0.6138891108224882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6138891108224882
+    },
+    {
+      "train_step": 154,
+      "loss": 0.6592429801239478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6592429801239478
+    },
+    {
+      "train_step": 155,
+      "loss": 0.6839466805734284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6839466805734284
+    },
+    {
+      "train_step": 156,
+      "loss": 0.5622090383184137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5622090383184137
+    },
+    {
+      "train_step": 157,
+      "loss": 0.7289048772148774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7289048772148774
+    },
+    {
+      "train_step": 158,
+      "loss": 0.7311794169164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7311794169164
+    },
+    {
+      "train_step": 159,
+      "loss": 0.8326504138253177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8326504138253177
+    },
+    {
+      "train_step": 160,
+      "loss": 0.6850441089512525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6850441089512525
+    },
+    {
+      "train_step": 161,
+      "loss": 0.8563625128021255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8563625128021255
+    },
+    {
+      "train_step": 162,
+      "loss": 0.48410951443780637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48410951443780637
+    },
+    {
+      "train_step": 163,
+      "loss": 0.5533328166766376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5533328166766376
+    },
+    {
+      "train_step": 164,
+      "loss": 0.6736093690422332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6736093690422332
+    },
+    {
+      "train_step": 165,
+      "loss": 0.4934474486703281,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4934474486703281
+    },
+    {
+      "train_step": 166,
+      "loss": 0.4964812775969277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4964812775969277
+    },
+    {
+      "train_step": 167,
+      "loss": 0.6819101084382071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6819101084382071
+    },
+    {
+      "train_step": 168,
+      "loss": 0.8064796131142828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8064796131142828
+    },
+    {
+      "train_step": 169,
+      "loss": 0.6855391892952214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6855391892952214
+    },
+    {
+      "train_step": 170,
+      "loss": 0.5872733079960657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5872733079960657
+    },
+    {
+      "train_step": 171,
+      "loss": 0.48323557785969884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48323557785969884
+    },
+    {
+      "train_step": 172,
+      "loss": 0.5911798828389497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5911798828389497
+    },
+    {
+      "train_step": 173,
+      "loss": 0.5741901670735802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5741901670735802
+    },
+    {
+      "train_step": 174,
+      "loss": 0.5282280343216441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5282280343216441
+    },
+    {
+      "train_step": 175,
+      "loss": 0.7658926916168521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7658926916168521
+    },
+    {
+      "train_step": 176,
+      "loss": 0.5954038790300076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5954038790300076
+    },
+    {
+      "train_step": 177,
+      "loss": 0.7268269970619976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7268269970619976
+    },
+    {
+      "train_step": 178,
+      "loss": 0.6502824716893829,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6502824716893829
+    },
+    {
+      "train_step": 179,
+      "loss": 0.6222345523619243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6222345523619243
+    },
+    {
+      "train_step": 180,
+      "loss": 0.5826332115539427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5826332115539427
+    },
+    {
+      "train_step": 181,
+      "loss": 0.5930676766622165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5930676766622165
+    },
+    {
+      "train_step": 182,
+      "loss": 0.4364551750945208,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4364551750945208
+    },
+    {
+      "train_step": 183,
+      "loss": 0.6961772772674201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6961772772674201
+    },
+    {
+      "train_step": 184,
+      "loss": 0.6458194510345703,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6458194510345703
+    },
+    {
+      "train_step": 185,
+      "loss": 0.3974733501200519,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3974733501200519
+    },
+    {
+      "train_step": 186,
+      "loss": 0.41261678741234814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41261678741234814
+    },
+    {
+      "train_step": 187,
+      "loss": 0.5279023970800076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5279023970800076
+    },
+    {
+      "train_step": 188,
+      "loss": 0.6311967899893863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6311967899893863
+    },
+    {
+      "train_step": 189,
+      "loss": 0.7861903542508532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7861903542508532
+    },
+    {
+      "train_step": 190,
+      "loss": 0.8187933767036965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8187933767036965
+    },
+    {
+      "train_step": 191,
+      "loss": 0.5496315432135794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5496315432135794
+    },
+    {
+      "train_step": 192,
+      "loss": 0.6923951126054774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6923951126054774
+    },
+    {
+      "train_step": 193,
+      "loss": 0.6587673354183083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6587673354183083
+    },
+    {
+      "train_step": 194,
+      "loss": 0.4177396170830188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4177396170830188
+    },
+    {
+      "train_step": 195,
+      "loss": 0.8849726849260189,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8849726849260189
+    },
+    {
+      "train_step": 196,
+      "loss": 0.4383989785495815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4383989785495815
+    },
+    {
+      "train_step": 197,
+      "loss": 0.808661913667715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.808661913667715
+    },
+    {
+      "train_step": 198,
+      "loss": 0.5150975646523904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5150975646523904
+    },
+    {
+      "train_step": 199,
+      "loss": 0.6303474563094252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6303474563094252
+    },
+    {
+      "train_step": 200,
+      "loss": 0.554920368375037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.554920368375037
+    },
+    {
+      "train_step": 201,
+      "loss": 0.742109085761314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.742109085761314
+    },
+    {
+      "train_step": 202,
+      "loss": 0.655906014616983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.655906014616983
+    },
+    {
+      "train_step": 203,
+      "loss": 0.609240115970195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.609240115970195
+    },
+    {
+      "train_step": 204,
+      "loss": 0.5679821740070302,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5679821740070302
+    },
+    {
+      "train_step": 205,
+      "loss": 0.7617062502191817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7617062502191817
+    },
+    {
+      "train_step": 206,
+      "loss": 0.6377928328411574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6377928328411574
+    },
+    {
+      "train_step": 207,
+      "loss": 0.623676787199582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.623676787199582
+    },
+    {
+      "train_step": 208,
+      "loss": 0.8909967986421643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8909967986421643
+    },
+    {
+      "train_step": 209,
+      "loss": 0.6720728646852714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6720728646852714
+    },
+    {
+      "train_step": 210,
+      "loss": 0.6935942179854833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6935942179854833
+    },
+    {
+      "train_step": 211,
+      "loss": 0.7116833400240049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7116833400240049
+    },
+    {
+      "train_step": 212,
+      "loss": 0.47388548280902276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47388548280902276
+    },
+    {
+      "train_step": 213,
+      "loss": 0.716631197240073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.716631197240073
+    },
+    {
+      "train_step": 214,
+      "loss": 0.6979170548149222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6979170548149222
+    },
+    {
+      "train_step": 215,
+      "loss": 0.7802992205644224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7802992205644224
+    },
+    {
+      "train_step": 216,
+      "loss": 0.604812926400658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.604812926400658
+    },
+    {
+      "train_step": 217,
+      "loss": 0.4906953291986613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4906953291986613
+    },
+    {
+      "train_step": 218,
+      "loss": 0.6530658761961141,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6530658761961141
+    },
+    {
+      "train_step": 219,
+      "loss": 0.3433352445524865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3433352445524865
+    },
+    {
+      "train_step": 220,
+      "loss": 0.6061104450579117,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6061104450579117
+    },
+    {
+      "train_step": 221,
+      "loss": 0.6158323820935785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6158323820935785
+    },
+    {
+      "train_step": 222,
+      "loss": 0.6165154799025941,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6165154799025941
+    },
+    {
+      "train_step": 223,
+      "loss": 0.7283675088723623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7283675088723623
+    },
+    {
+      "train_step": 224,
+      "loss": 0.709938162333141,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.709938162333141
+    },
+    {
+      "train_step": 225,
+      "loss": 0.6124354067826088,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6124354067826088
+    },
+    {
+      "train_step": 226,
+      "loss": 0.547328048792503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.547328048792503
+    },
+    {
+      "train_step": 227,
+      "loss": 0.6693894944755145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6693894944755145
+    },
+    {
+      "train_step": 228,
+      "loss": 0.7014979567450881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7014979567450881
+    },
+    {
+      "train_step": 229,
+      "loss": 0.694530028049857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.694530028049857
+    },
+    {
+      "train_step": 230,
+      "loss": 0.5441954329819993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441954329819993
+    },
+    {
+      "train_step": 231,
+      "loss": 0.5652765245538096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5652765245538096
+    },
+    {
+      "train_step": 232,
+      "loss": 0.581759619913265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.581759619913265
+    },
+    {
+      "train_step": 233,
+      "loss": 0.3972182445687764,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3972182445687764
+    },
+    {
+      "train_step": 234,
+      "loss": 0.6273337100675531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6273337100675531
+    },
+    {
+      "train_step": 235,
+      "loss": 0.7071919602498077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7071919602498077
+    },
+    {
+      "train_step": 236,
+      "loss": 0.5494249164740211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5494249164740211
+    },
+    {
+      "train_step": 237,
+      "loss": 0.7565048311289115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7565048311289115
+    },
+    {
+      "train_step": 238,
+      "loss": 0.7586020741688894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7586020741688894
+    },
+    {
+      "train_step": 239,
+      "loss": 0.7178494484874834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7178494484874834
+    },
+    {
+      "train_step": 240,
+      "loss": 0.5767165609746663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5767165609746663
+    },
+    {
+      "train_step": 241,
+      "loss": 0.6027481590304866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6027481590304866
+    },
+    {
+      "train_step": 242,
+      "loss": 0.8043244996668243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8043244996668243
+    },
+    {
+      "train_step": 243,
+      "loss": 0.6150830421287071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6150830421287071
+    },
+    {
+      "train_step": 244,
+      "loss": 0.6005896312371318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6005896312371318
+    },
+    {
+      "train_step": 245,
+      "loss": 0.7266476122717075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7266476122717075
+    },
+    {
+      "train_step": 246,
+      "loss": 0.701385056554313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.701385056554313
+    },
+    {
+      "train_step": 247,
+      "loss": 0.6634100136199306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6634100136199306
+    },
+    {
+      "train_step": 248,
+      "loss": 0.47243641559585536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47243641559585536
+    },
+    {
+      "train_step": 249,
+      "loss": 0.6194287801228964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6194287801228964
+    },
+    {
+      "train_step": 250,
+      "loss": 0.4720035084762124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4720035084762124
+    },
+    {
+      "train_step": 251,
+      "loss": 0.6224032315472777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6224032315472777
+    },
+    {
+      "train_step": 252,
+      "loss": 0.5387947814503524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5387947814503524
+    },
+    {
+      "train_step": 253,
+      "loss": 0.6814270525445917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6814270525445917
+    },
+    {
+      "train_step": 254,
+      "loss": 0.6636651888257079,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6636651888257079
+    },
+    {
+      "train_step": 255,
+      "loss": 0.5482507302721015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5482507302721015
+    },
+    {
+      "train_step": 256,
+      "loss": 0.8349175416756258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8349175416756258
+    },
+    {
+      "train_step": 257,
+      "loss": 0.3994146482006806,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3994146482006806
+    },
+    {
+      "train_step": 258,
+      "loss": 0.521921366160246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.521921366160246
+    },
+    {
+      "train_step": 259,
+      "loss": 0.6037291657898428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6037291657898428
+    },
+    {
+      "train_step": 260,
+      "loss": 0.6703795442791107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6703795442791107
+    },
+    {
+      "train_step": 261,
+      "loss": 0.7735127960427839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7735127960427839
+    },
+    {
+      "train_step": 262,
+      "loss": 0.6297473716282873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6297473716282873
+    },
+    {
+      "train_step": 263,
+      "loss": 0.5758666099385816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5758666099385816
+    },
+    {
+      "train_step": 264,
+      "loss": 0.526145518654981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.526145518654981
+    },
+    {
+      "train_step": 265,
+      "loss": 0.5803848129787706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5803848129787706
+    },
+    {
+      "train_step": 266,
+      "loss": 0.6487909189642035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6487909189642035
+    },
+    {
+      "train_step": 267,
+      "loss": 0.5459567827802386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5459567827802386
+    },
+    {
+      "train_step": 268,
+      "loss": 0.596283735029685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.596283735029685
+    },
+    {
+      "train_step": 269,
+      "loss": 0.708104848913171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.708104848913171
+    },
+    {
+      "train_step": 270,
+      "loss": 0.6348728902247751,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6348728902247751
+    },
+    {
+      "train_step": 271,
+      "loss": 0.7033414744261468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7033414744261468
+    },
+    {
+      "train_step": 272,
+      "loss": 0.5024540809033157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5024540809033157
+    },
+    {
+      "train_step": 273,
+      "loss": 0.7717956674001103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7717956674001103
+    },
+    {
+      "train_step": 274,
+      "loss": 0.576467820697903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576467820697903
+    },
+    {
+      "train_step": 275,
+      "loss": 0.6472069050086158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6472069050086158
+    },
+    {
+      "train_step": 276,
+      "loss": 0.7176403041394278,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7176403041394278
+    },
+    {
+      "train_step": 277,
+      "loss": 0.5366988735034054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5366988735034054
+    },
+    {
+      "train_step": 278,
+      "loss": 0.5172419336629482,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5172419336629482
+    },
+    {
+      "train_step": 279,
+      "loss": 0.582189446425206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.582189446425206
+    },
+    {
+      "train_step": 280,
+      "loss": 0.7398959961674926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7398959961674926
+    },
+    {
+      "train_step": 281,
+      "loss": 0.816445448400068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.816445448400068
+    },
+    {
+      "train_step": 282,
+      "loss": 0.6079494976498812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6079494976498812
+    },
+    {
+      "train_step": 283,
+      "loss": 0.4879590886989632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4879590886989632
+    },
+    {
+      "train_step": 284,
+      "loss": 0.5909829137168624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5909829137168624
+    },
+    {
+      "train_step": 285,
+      "loss": 0.8577292352924335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8577292352924335
+    },
+    {
+      "train_step": 286,
+      "loss": 0.6122498892960357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6122498892960357
+    },
+    {
+      "train_step": 287,
+      "loss": 0.4787235582767049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4787235582767049
+    },
+    {
+      "train_step": 288,
+      "loss": 0.7526500343243154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7526500343243154
+    },
+    {
+      "train_step": 289,
+      "loss": 0.7335186504630609,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7335186504630609
+    },
+    {
+      "train_step": 290,
+      "loss": 0.6927047367378534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6927047367378534
+    },
+    {
+      "train_step": 291,
+      "loss": 0.6399572850888214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6399572850888214
+    },
+    {
+      "train_step": 292,
+      "loss": 0.6139883927833537,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6139883927833537
+    },
+    {
+      "train_step": 293,
+      "loss": 0.6723689652199532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6723689652199532
+    },
+    {
+      "train_step": 294,
+      "loss": 0.4440038139912361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4440038139912361
+    },
+    {
+      "train_step": 295,
+      "loss": 0.5983558514427549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5983558514427549
+    },
+    {
+      "train_step": 296,
+      "loss": 0.3031271882172629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3031271882172629
+    },
+    {
+      "train_step": 297,
+      "loss": 0.7994344634532232,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7994344634532232
+    },
+    {
+      "train_step": 298,
+      "loss": 0.47860378519609836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47860378519609836
+    },
+    {
+      "train_step": 299,
+      "loss": 0.535875066001356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.535875066001356
+    },
+    {
+      "train_step": 300,
+      "loss": 0.4857031075327226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4857031075327226
+    },
+    {
+      "train_step": 301,
+      "loss": 0.5400318853886875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5400318853886875
+    },
+    {
+      "train_step": 302,
+      "loss": 0.6237830766117478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6237830766117478
+    },
+    {
+      "train_step": 303,
+      "loss": 0.6627863643593457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6627863643593457
+    },
+    {
+      "train_step": 304,
+      "loss": 0.7255288739740268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7255288739740268
+    },
+    {
+      "train_step": 305,
+      "loss": 0.5931973145060226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5931973145060226
+    },
+    {
+      "train_step": 306,
+      "loss": 0.5747967406557544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5747967406557544
+    },
+    {
+      "train_step": 307,
+      "loss": 0.6714451792402361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6714451792402361
+    },
+    {
+      "train_step": 308,
+      "loss": 0.5902128475450966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5902128475450966
+    },
+    {
+      "train_step": 309,
+      "loss": 0.4380403672214913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4380403672214913
+    },
+    {
+      "train_step": 310,
+      "loss": 0.6650752324424947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6650752324424947
+    },
+    {
+      "train_step": 311,
+      "loss": 0.4425652742381559,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4425652742381559
+    },
+    {
+      "train_step": 312,
+      "loss": 0.6700280527312561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6700280527312561
+    },
+    {
+      "train_step": 313,
+      "loss": 0.48272114067328525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48272114067328525
+    },
+    {
+      "train_step": 314,
+      "loss": 0.3984228401738462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3984228401738462
+    },
+    {
+      "train_step": 315,
+      "loss": 0.8956596401553872,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8956596401553872
+    },
+    {
+      "train_step": 316,
+      "loss": 0.7090390610874786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7090390610874786
+    },
+    {
+      "train_step": 317,
+      "loss": 0.5827476821358073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5827476821358073
+    },
+    {
+      "train_step": 318,
+      "loss": 0.6518022686818359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6518022686818359
+    },
+    {
+      "train_step": 319,
+      "loss": 0.5792041207545868,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5792041207545868
+    },
+    {
+      "train_step": 320,
+      "loss": 0.5187574099324788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5187574099324788
+    },
+    {
+      "train_step": 321,
+      "loss": 0.6178311694162558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6178311694162558
+    },
+    {
+      "train_step": 322,
+      "loss": 0.5086946425566101,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5086946425566101
+    },
+    {
+      "train_step": 323,
+      "loss": 0.5590294044704799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5590294044704799
+    },
+    {
+      "train_step": 324,
+      "loss": 0.8840375736609792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8840375736609792
+    },
+    {
+      "train_step": 325,
+      "loss": 0.6168666038052381,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6168666038052381
+    },
+    {
+      "train_step": 326,
+      "loss": 0.5277437363212499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5277437363212499
+    },
+    {
+      "train_step": 327,
+      "loss": 0.8252739862673681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8252739862673681
+    },
+    {
+      "train_step": 328,
+      "loss": 0.6357851100525138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6357851100525138
+    },
+    {
+      "train_step": 329,
+      "loss": 0.5064615874760784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5064615874760784
+    },
+    {
+      "train_step": 330,
+      "loss": 0.5138048450515481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5138048450515481
+    },
+    {
+      "train_step": 331,
+      "loss": 0.6685340440086038,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6685340440086038
+    },
+    {
+      "train_step": 332,
+      "loss": 0.7113792297479424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7113792297479424
+    },
+    {
+      "train_step": 333,
+      "loss": 0.6720024251978981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6720024251978981
+    },
+    {
+      "train_step": 334,
+      "loss": 0.5676420034749761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5676420034749761
+    },
+    {
+      "train_step": 335,
+      "loss": 0.40729610413032136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40729610413032136
+    },
+    {
+      "train_step": 336,
+      "loss": 0.6352503829504589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6352503829504589
+    },
+    {
+      "train_step": 337,
+      "loss": 0.45420461614324187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45420461614324187
+    },
+    {
+      "train_step": 338,
+      "loss": 0.6487329665398726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6487329665398726
+    },
+    {
+      "train_step": 339,
+      "loss": 0.3864679229953612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3864679229953612
+    },
+    {
+      "train_step": 340,
+      "loss": 0.58563776798085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.58563776798085
+    },
+    {
+      "train_step": 341,
+      "loss": 0.6042515796818315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6042515796818315
+    },
+    {
+      "train_step": 342,
+      "loss": 0.6916191064597615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6916191064597615
+    },
+    {
+      "train_step": 343,
+      "loss": 0.4724032159665161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4724032159665161
+    },
+    {
+      "train_step": 344,
+      "loss": 0.45107135069823295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45107135069823295
+    },
+    {
+      "train_step": 345,
+      "loss": 0.46179175874920825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46179175874920825
+    },
+    {
+      "train_step": 346,
+      "loss": 0.5672300110702178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5672300110702178
+    },
+    {
+      "train_step": 347,
+      "loss": 0.5463478483248939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5463478483248939
+    },
+    {
+      "train_step": 348,
+      "loss": 0.7501376093803955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7501376093803955
+    },
+    {
+      "train_step": 349,
+      "loss": 0.5982620495451068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5982620495451068
+    },
+    {
+      "train_step": 350,
+      "loss": 0.5405679057487328,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5405679057487328
+    },
+    {
+      "train_step": 351,
+      "loss": 0.7867778533331199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7867778533331199
+    },
+    {
+      "train_step": 352,
+      "loss": 0.604014620208009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.604014620208009
+    },
+    {
+      "train_step": 353,
+      "loss": 0.6390460405574271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6390460405574271
+    },
+    {
+      "train_step": 354,
+      "loss": 0.7362600127361557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7362600127361557
+    },
+    {
+      "train_step": 355,
+      "loss": 0.7811485887562025,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7811485887562025
+    },
+    {
+      "train_step": 356,
+      "loss": 0.5965833782592905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5965833782592905
+    },
+    {
+      "train_step": 357,
+      "loss": 0.7192729117296466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7192729117296466
+    },
+    {
+      "train_step": 358,
+      "loss": 0.5346347302023914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5346347302023914
+    },
+    {
+      "train_step": 359,
+      "loss": 0.5748850509782588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5748850509782588
+    },
+    {
+      "train_step": 360,
+      "loss": 0.582017630091755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.582017630091755
+    },
+    {
+      "train_step": 361,
+      "loss": 0.5801802919378065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5801802919378065
+    },
+    {
+      "train_step": 362,
+      "loss": 0.6595363848500826,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6595363848500826
+    },
+    {
+      "train_step": 363,
+      "loss": 0.6559371548997145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6559371548997145
+    },
+    {
+      "train_step": 364,
+      "loss": 0.7592511348767645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7592511348767645
+    },
+    {
+      "train_step": 365,
+      "loss": 0.6397308132633213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6397308132633213
+    },
+    {
+      "train_step": 366,
+      "loss": 0.38934332701415086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38934332701415086
+    },
+    {
+      "train_step": 367,
+      "loss": 0.56324120674481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.56324120674481
+    },
+    {
+      "train_step": 368,
+      "loss": 0.6825242533197668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6825242533197668
+    },
+    {
+      "train_step": 369,
+      "loss": 0.7305560862823229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7305560862823229
+    },
+    {
+      "train_step": 370,
+      "loss": 0.5921260433045582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5921260433045582
+    },
+    {
+      "train_step": 371,
+      "loss": 0.6317085025243152,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6317085025243152
+    },
+    {
+      "train_step": 372,
+      "loss": 0.6006964750036378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6006964750036378
+    },
+    {
+      "train_step": 373,
+      "loss": 0.798315730994674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.798315730994674
+    },
+    {
+      "train_step": 374,
+      "loss": 0.5556166278913173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5556166278913173
+    },
+    {
+      "train_step": 375,
+      "loss": 0.6521709041100006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6521709041100006
+    },
+    {
+      "train_step": 376,
+      "loss": 0.5749309861919994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5749309861919994
+    },
+    {
+      "train_step": 377,
+      "loss": 0.6456293819406118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6456293819406118
+    },
+    {
+      "train_step": 378,
+      "loss": 0.5644891354077246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5644891354077246
+    },
+    {
+      "train_step": 379,
+      "loss": 0.4494520577879822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4494520577879822
+    },
+    {
+      "train_step": 380,
+      "loss": 0.4956462195366453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4956462195366453
+    },
+    {
+      "train_step": 381,
+      "loss": 0.5532791526550768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5532791526550768
+    },
+    {
+      "train_step": 382,
+      "loss": 0.42848691790660315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42848691790660315
+    },
+    {
+      "train_step": 383,
+      "loss": 0.6235126633457879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6235126633457879
+    },
+    {
+      "train_step": 384,
+      "loss": 0.42629520568309964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42629520568309964
+    },
+    {
+      "train_step": 385,
+      "loss": 0.4188192745584753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4188192745584753
+    },
+    {
+      "train_step": 386,
+      "loss": 0.4603885908429759,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4603885908429759
+    },
+    {
+      "train_step": 387,
+      "loss": 0.4704592270213736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4704592270213736
+    },
+    {
+      "train_step": 388,
+      "loss": 0.6698733412933213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6698733412933213
+    },
+    {
+      "train_step": 389,
+      "loss": 0.6669547450677192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6669547450677192
+    },
+    {
+      "train_step": 390,
+      "loss": 0.566411506356849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.566411506356849
+    },
+    {
+      "train_step": 391,
+      "loss": 0.5838518658535967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5838518658535967
+    },
+    {
+      "train_step": 392,
+      "loss": 0.6158271329940727,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6158271329940727
+    },
+    {
+      "train_step": 393,
+      "loss": 0.6390333892110409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6390333892110409
+    },
+    {
+      "train_step": 394,
+      "loss": 0.7058838823217399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7058838823217399
+    },
+    {
+      "train_step": 395,
+      "loss": 0.7128358272952632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7128358272952632
+    },
+    {
+      "train_step": 396,
+      "loss": 0.5365973846591547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5365973846591547
+    },
+    {
+      "train_step": 397,
+      "loss": 0.5159454246485291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5159454246485291
+    },
+    {
+      "train_step": 398,
+      "loss": 0.4357534974369514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4357534974369514
+    },
+    {
+      "train_step": 399,
+      "loss": 0.7284433154594001,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7284433154594001
+    },
+    {
+      "train_step": 400,
+      "loss": 0.4936620024269893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4936620024269893
+    },
+    {
+      "train_step": 401,
+      "loss": 0.5182754787329236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5182754787329236
+    },
+    {
+      "train_step": 402,
+      "loss": 0.6085836749219503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6085836749219503
+    },
+    {
+      "train_step": 403,
+      "loss": 0.5837500308015724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5837500308015724
+    },
+    {
+      "train_step": 404,
+      "loss": 0.5839387222580388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5839387222580388
+    },
+    {
+      "train_step": 405,
+      "loss": 0.4310326106225454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4310326106225454
+    },
+    {
+      "train_step": 406,
+      "loss": 0.7530008772045512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7530008772045512
+    },
+    {
+      "train_step": 407,
+      "loss": 0.7621715083146721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7621715083146721
+    },
+    {
+      "train_step": 408,
+      "loss": 0.5323991423408473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5323991423408473
+    },
+    {
+      "train_step": 409,
+      "loss": 0.5306368922656282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5306368922656282
+    },
+    {
+      "train_step": 410,
+      "loss": 0.925532169004482,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.925532169004482
+    },
+    {
+      "train_step": 411,
+      "loss": 0.6279993880406941,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6279993880406941
+    },
+    {
+      "train_step": 412,
+      "loss": 0.5916720942643822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5916720942643822
+    },
+    {
+      "train_step": 413,
+      "loss": 0.8514798112434587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8514798112434587
+    },
+    {
+      "train_step": 414,
+      "loss": 0.6228290921107948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6228290921107948
+    },
+    {
+      "train_step": 415,
+      "loss": 0.820172414936786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.820172414936786
+    },
+    {
+      "train_step": 416,
+      "loss": 0.5977440797674166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5977440797674166
+    },
+    {
+      "train_step": 417,
+      "loss": 0.620161519917033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.620161519917033
+    },
+    {
+      "train_step": 418,
+      "loss": 0.7365695659151087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7365695659151087
+    },
+    {
+      "train_step": 419,
+      "loss": 0.6142672913588327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6142672913588327
+    },
+    {
+      "train_step": 420,
+      "loss": 0.6083191241944809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6083191241944809
+    },
+    {
+      "train_step": 421,
+      "loss": 0.5435226678446274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5435226678446274
+    },
+    {
+      "train_step": 422,
+      "loss": 0.5787162142112275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5787162142112275
+    },
+    {
+      "train_step": 423,
+      "loss": 0.3769667143656818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3769667143656818
+    },
+    {
+      "train_step": 424,
+      "loss": 0.5207527249194535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5207527249194535
+    },
+    {
+      "train_step": 425,
+      "loss": 0.5168840601510157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5168840601510157
+    },
+    {
+      "train_step": 426,
+      "loss": 0.44007990746592296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44007990746592296
+    },
+    {
+      "train_step": 427,
+      "loss": 0.45946514471765054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45946514471765054
+    },
+    {
+      "train_step": 428,
+      "loss": 0.6053324992819149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6053324992819149
+    },
+    {
+      "train_step": 429,
+      "loss": 0.5041604351120974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5041604351120974
+    },
+    {
+      "train_step": 430,
+      "loss": 0.8087425305364052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8087425305364052
+    },
+    {
+      "train_step": 431,
+      "loss": 0.6997226812489862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6997226812489862
+    },
+    {
+      "train_step": 432,
+      "loss": 0.5642608198894019,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5642608198894019
+    },
+    {
+      "train_step": 433,
+      "loss": 0.641535603803648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.641535603803648
+    },
+    {
+      "train_step": 434,
+      "loss": 0.6668534941844091,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6668534941844091
+    },
+    {
+      "train_step": 435,
+      "loss": 0.6249176521998069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6249176521998069
+    },
+    {
+      "train_step": 436,
+      "loss": 0.543479834519558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.543479834519558
+    },
+    {
+      "train_step": 437,
+      "loss": 0.49309139213371755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49309139213371755
+    },
+    {
+      "train_step": 438,
+      "loss": 0.698480153483082,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.698480153483082
+    },
+    {
+      "train_step": 439,
+      "loss": 0.5213923836500254,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5213923836500254
+    },
+    {
+      "train_step": 440,
+      "loss": 0.5052213370492351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5052213370492351
+    },
+    {
+      "train_step": 441,
+      "loss": 0.8421310358173777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8421310358173777
+    },
+    {
+      "train_step": 442,
+      "loss": 0.5739465587975672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5739465587975672
+    },
+    {
+      "train_step": 443,
+      "loss": 0.568187928218137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.568187928218137
+    },
+    {
+      "train_step": 444,
+      "loss": 0.7839861227890267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7839861227890267
+    },
+    {
+      "train_step": 445,
+      "loss": 0.7568093552683173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7568093552683173
+    },
+    {
+      "train_step": 446,
+      "loss": 0.5049460376176159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5049460376176159
+    },
+    {
+      "train_step": 447,
+      "loss": 0.6472000851638964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6472000851638964
+    },
+    {
+      "train_step": 448,
+      "loss": 0.5397992990952842,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5397992990952842
+    },
+    {
+      "train_step": 449,
+      "loss": 0.7914265184339258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7914265184339258
+    },
+    {
+      "train_step": 450,
+      "loss": 0.7757523685737556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7757523685737556
+    },
+    {
+      "train_step": 451,
+      "loss": 0.5798955249557407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5798955249557407
+    },
+    {
+      "train_step": 452,
+      "loss": 0.4982078876758577,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4982078876758577
+    },
+    {
+      "train_step": 453,
+      "loss": 0.5928685670318973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5928685670318973
+    },
+    {
+      "train_step": 454,
+      "loss": 0.7298880685902144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7298880685902144
+    },
+    {
+      "train_step": 455,
+      "loss": 0.6612322000204677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6612322000204677
+    },
+    {
+      "train_step": 456,
+      "loss": 0.6670487312654714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6670487312654714
+    },
+    {
+      "train_step": 457,
+      "loss": 0.44654206066467755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44654206066467755
+    },
+    {
+      "train_step": 458,
+      "loss": 0.479198162656528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.479198162656528
+    },
+    {
+      "train_step": 459,
+      "loss": 0.5706965186178572,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706965186178572
+    },
+    {
+      "train_step": 460,
+      "loss": 0.5701042670638706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5701042670638706
+    },
+    {
+      "train_step": 461,
+      "loss": 0.5808250314347451,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5808250314347451
+    },
+    {
+      "train_step": 462,
+      "loss": 0.5649108862797688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5649108862797688
+    },
+    {
+      "train_step": 463,
+      "loss": 0.49702636023235186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49702636023235186
+    },
+    {
+      "train_step": 464,
+      "loss": 0.7932678383168953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7932678383168953
+    },
+    {
+      "train_step": 465,
+      "loss": 0.6020453970581782,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6020453970581782
+    },
+    {
+      "train_step": 466,
+      "loss": 0.6107025892716078,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6107025892716078
+    },
+    {
+      "train_step": 467,
+      "loss": 0.5556030630689134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5556030630689134
+    },
+    {
+      "train_step": 468,
+      "loss": 0.5281275778655642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5281275778655642
+    },
+    {
+      "train_step": 469,
+      "loss": 0.8261619166388543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8261619166388543
+    },
+    {
+      "train_step": 470,
+      "loss": 0.6604067337431077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6604067337431077
+    },
+    {
+      "train_step": 471,
+      "loss": 0.5473029733594391,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5473029733594391
+    },
+    {
+      "train_step": 472,
+      "loss": 0.7690487381561861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7690487381561861
+    },
+    {
+      "train_step": 473,
+      "loss": 0.6519397065341663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6519397065341663
+    },
+    {
+      "train_step": 474,
+      "loss": 0.5476297243524582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5476297243524582
+    },
+    {
+      "train_step": 475,
+      "loss": 0.3859776392311201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3859776392311201
+    },
+    {
+      "train_step": 476,
+      "loss": 0.8722479728406244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8722479728406244
+    },
+    {
+      "train_step": 477,
+      "loss": 0.6201971830385444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6201971830385444
+    },
+    {
+      "train_step": 478,
+      "loss": 0.41268298640602324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41268298640602324
+    },
+    {
+      "train_step": 479,
+      "loss": 0.5715338049440292,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5715338049440292
+    },
+    {
+      "train_step": 480,
+      "loss": 0.643723067716552,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.643723067716552
+    },
+    {
+      "train_step": 481,
+      "loss": 0.6618012816312548,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6618012816312548
+    },
+    {
+      "train_step": 482,
+      "loss": 0.6119853172885225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6119853172885225
+    },
+    {
+      "train_step": 483,
+      "loss": 0.8243369582797337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8243369582797337
+    },
+    {
+      "train_step": 484,
+      "loss": 0.5722963263550299,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5722963263550299
+    },
+    {
+      "train_step": 485,
+      "loss": 0.5702995739576763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5702995739576763
+    },
+    {
+      "train_step": 486,
+      "loss": 0.7184246163391897,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7184246163391897
+    },
+    {
+      "train_step": 487,
+      "loss": 0.6113190808703154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6113190808703154
+    },
+    {
+      "train_step": 488,
+      "loss": 0.4614091696017695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4614091696017695
+    },
+    {
+      "train_step": 489,
+      "loss": 0.6441186222330367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6441186222330367
+    },
+    {
+      "train_step": 490,
+      "loss": 0.49947515682079247,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49947515682079247
+    },
+    {
+      "train_step": 491,
+      "loss": 0.597802980398619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.597802980398619
+    },
+    {
+      "train_step": 492,
+      "loss": 0.675121922052875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.675121922052875
+    },
+    {
+      "train_step": 493,
+      "loss": 0.7601016484603295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7601016484603295
+    },
+    {
+      "train_step": 494,
+      "loss": 0.6134084855536622,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6134084855536622
+    },
+    {
+      "train_step": 495,
+      "loss": 0.6943629270178765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6943629270178765
+    },
+    {
+      "train_step": 496,
+      "loss": 0.7409042716866305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7409042716866305
+    },
+    {
+      "train_step": 497,
+      "loss": 0.49510888052305446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49510888052305446
+    },
+    {
+      "train_step": 498,
+      "loss": 0.6402575316404633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6402575316404633
+    },
+    {
+      "train_step": 499,
+      "loss": 0.7478884791219668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7478884791219668
+    },
+    {
+      "train_step": 500,
+      "loss": 0.4741565454301042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4741565454301042
+    },
+    {
+      "train_step": 501,
+      "loss": 0.5814928874339488,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5814928874339488
+    },
+    {
+      "train_step": 502,
+      "loss": 0.6102499859603239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6102499859603239
+    },
+    {
+      "train_step": 503,
+      "loss": 0.6532197878647917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6532197878647917
+    },
+    {
+      "train_step": 504,
+      "loss": 0.7015171693264453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7015171693264453
+    },
+    {
+      "train_step": 505,
+      "loss": 0.5834256249119952,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5834256249119952
+    },
+    {
+      "train_step": 506,
+      "loss": 0.5650215156791858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5650215156791858
+    },
+    {
+      "train_step": 507,
+      "loss": 0.5795426557524224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5795426557524224
+    },
+    {
+      "train_step": 508,
+      "loss": 0.6052460971318407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6052460971318407
+    },
+    {
+      "train_step": 509,
+      "loss": 0.5732208791512428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5732208791512428
+    },
+    {
+      "train_step": 510,
+      "loss": 0.4459822206912242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4459822206912242
+    },
+    {
+      "train_step": 511,
+      "loss": 0.46046585663979145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46046585663979145
+    },
+    {
+      "train_step": 512,
+      "loss": 0.588405809401251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.588405809401251
+    },
+    {
+      "train_step": 513,
+      "loss": 0.7615934708741305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7615934708741305
+    },
+    {
+      "train_step": 514,
+      "loss": 0.5564638849602298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5564638849602298
+    },
+    {
+      "train_step": 515,
+      "loss": 0.6908528182141069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6908528182141069
+    },
+    {
+      "train_step": 516,
+      "loss": 0.5947087650218866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5947087650218866
+    },
+    {
+      "train_step": 517,
+      "loss": 0.6529873340896538,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6529873340896538
+    },
+    {
+      "train_step": 518,
+      "loss": 0.4116847618460302,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4116847618460302
+    },
+    {
+      "train_step": 519,
+      "loss": 0.7289094349725336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7289094349725336
+    },
+    {
+      "train_step": 520,
+      "loss": 0.5145766113325575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5145766113325575
+    },
+    {
+      "train_step": 521,
+      "loss": 0.4833301096815309,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4833301096815309
+    },
+    {
+      "train_step": 522,
+      "loss": 0.6916291751652434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6916291751652434
+    },
+    {
+      "train_step": 523,
+      "loss": 0.5129503676809044,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5129503676809044
+    },
+    {
+      "train_step": 524,
+      "loss": 0.5385653004002524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5385653004002524
+    },
+    {
+      "train_step": 525,
+      "loss": 0.7628526290475103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7628526290475103
+    },
+    {
+      "train_step": 526,
+      "loss": 0.675031289765221,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.675031289765221
+    },
+    {
+      "train_step": 527,
+      "loss": 0.6596140808309354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6596140808309354
+    },
+    {
+      "train_step": 528,
+      "loss": 0.39614811315710347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39614811315710347
+    },
+    {
+      "train_step": 529,
+      "loss": 0.7053849481418866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7053849481418866
+    },
+    {
+      "train_step": 530,
+      "loss": 0.26106060599284925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.26106060599284925
+    },
+    {
+      "train_step": 531,
+      "loss": 0.6050892837392168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6050892837392168
+    },
+    {
+      "train_step": 532,
+      "loss": 0.7568990607266419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7568990607266419
+    },
+    {
+      "train_step": 533,
+      "loss": 0.7210140315942192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7210140315942192
+    },
+    {
+      "train_step": 534,
+      "loss": 0.5571358125915945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5571358125915945
+    },
+    {
+      "train_step": 535,
+      "loss": 0.5707160944609018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5707160944609018
+    },
+    {
+      "train_step": 536,
+      "loss": 0.6791824294805493,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6791824294805493
+    },
+    {
+      "train_step": 537,
+      "loss": 0.5380261175245321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5380261175245321
+    },
+    {
+      "train_step": 538,
+      "loss": 0.597412184186807,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.597412184186807
+    },
+    {
+      "train_step": 539,
+      "loss": 0.5643018713210205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5643018713210205
+    },
+    {
+      "train_step": 540,
+      "loss": 0.5950741556693678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5950741556693678
+    },
+    {
+      "train_step": 541,
+      "loss": 0.524439335673568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.524439335673568
+    },
+    {
+      "train_step": 542,
+      "loss": 0.5418508402716582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5418508402716582
+    },
+    {
+      "train_step": 543,
+      "loss": 0.5494175845077406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5494175845077406
+    },
+    {
+      "train_step": 544,
+      "loss": 0.6241390745734775,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6241390745734775
+    },
+    {
+      "train_step": 545,
+      "loss": 0.5077880944772251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5077880944772251
+    },
+    {
+      "train_step": 546,
+      "loss": 0.49479387667073094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49479387667073094
+    },
+    {
+      "train_step": 547,
+      "loss": 0.7167235238125824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7167235238125824
+    },
+    {
+      "train_step": 548,
+      "loss": 0.5308906860102646,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5308906860102646
+    },
+    {
+      "train_step": 549,
+      "loss": 0.4525577747087966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4525577747087966
+    },
+    {
+      "train_step": 550,
+      "loss": 0.7781584644594867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7781584644594867
+    },
+    {
+      "train_step": 551,
+      "loss": 0.639998101533287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.639998101533287
+    },
+    {
+      "train_step": 552,
+      "loss": 0.5773846985057167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5773846985057167
+    },
+    {
+      "train_step": 553,
+      "loss": 0.6466006630801014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6466006630801014
+    },
+    {
+      "train_step": 554,
+      "loss": 0.6902593093778108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6902593093778108
+    },
+    {
+      "train_step": 555,
+      "loss": 0.7512775918398263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7512775918398263
+    },
+    {
+      "train_step": 556,
+      "loss": 0.4902729135192856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4902729135192856
+    },
+    {
+      "train_step": 557,
+      "loss": 0.48429333884533887,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48429333884533887
+    },
+    {
+      "train_step": 558,
+      "loss": 0.7986893654499666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7986893654499666
+    },
+    {
+      "train_step": 559,
+      "loss": 0.5504385571875361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5504385571875361
+    },
+    {
+      "train_step": 560,
+      "loss": 0.7317071590755674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7317071590755674
+    },
+    {
+      "train_step": 561,
+      "loss": 0.6187898667812776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6187898667812776
+    },
+    {
+      "train_step": 562,
+      "loss": 0.6578745492134265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6578745492134265
+    },
+    {
+      "train_step": 563,
+      "loss": 0.44341655100405086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44341655100405086
+    },
+    {
+      "train_step": 564,
+      "loss": 0.629017215487373,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.629017215487373
+    },
+    {
+      "train_step": 565,
+      "loss": 0.4647548634898732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4647548634898732
+    },
+    {
+      "train_step": 566,
+      "loss": 0.6662516755251249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6662516755251249
+    },
+    {
+      "train_step": 567,
+      "loss": 0.4583157799024435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4583157799024435
+    },
+    {
+      "train_step": 568,
+      "loss": 0.7178575686754634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7178575686754634
+    },
+    {
+      "train_step": 569,
+      "loss": 0.5170138493446285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5170138493446285
+    },
+    {
+      "train_step": 570,
+      "loss": 0.9210218437340503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9210218437340503
+    },
+    {
+      "train_step": 571,
+      "loss": 0.603839468101051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.603839468101051
+    },
+    {
+      "train_step": 572,
+      "loss": 0.5522599559385648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5522599559385648
+    },
+    {
+      "train_step": 573,
+      "loss": 0.4830593399462103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4830593399462103
+    },
+    {
+      "train_step": 574,
+      "loss": 0.6811705218835293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6811705218835293
+    },
+    {
+      "train_step": 575,
+      "loss": 0.473667075455893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.473667075455893
+    },
+    {
+      "train_step": 576,
+      "loss": 0.6299978065564291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6299978065564291
+    },
+    {
+      "train_step": 577,
+      "loss": 0.7051611980943036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7051611980943036
+    },
+    {
+      "train_step": 578,
+      "loss": 0.4978622905453151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4978622905453151
+    },
+    {
+      "train_step": 579,
+      "loss": 0.59267570886637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.59267570886637
+    },
+    {
+      "train_step": 580,
+      "loss": 0.6570821547631245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6570821547631245
+    },
+    {
+      "train_step": 581,
+      "loss": 0.6357026283178171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6357026283178171
+    },
+    {
+      "train_step": 582,
+      "loss": 0.5035661072709876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035661072709876
+    },
+    {
+      "train_step": 583,
+      "loss": 0.6971827809622291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6971827809622291
+    },
+    {
+      "train_step": 584,
+      "loss": 0.7598615383934717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7598615383934717
+    },
+    {
+      "train_step": 585,
+      "loss": 0.6295018600327559,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6295018600327559
+    },
+    {
+      "train_step": 586,
+      "loss": 0.48389417208000807,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48389417208000807
+    },
+    {
+      "train_step": 587,
+      "loss": 0.8194523048144947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8194523048144947
+    },
+    {
+      "train_step": 588,
+      "loss": 0.6346961856461989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6346961856461989
+    },
+    {
+      "train_step": 589,
+      "loss": 0.5633329812700483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5633329812700483
+    },
+    {
+      "train_step": 590,
+      "loss": 0.6030491031705593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6030491031705593
+    },
+    {
+      "train_step": 591,
+      "loss": 0.6112164482554567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6112164482554567
+    },
+    {
+      "train_step": 592,
+      "loss": 0.691998235574388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.691998235574388
+    },
+    {
+      "train_step": 593,
+      "loss": 0.6945815452263104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6945815452263104
+    },
+    {
+      "train_step": 594,
+      "loss": 0.5279763596672903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5279763596672903
+    },
+    {
+      "train_step": 595,
+      "loss": 0.7231954806324281,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7231954806324281
+    },
+    {
+      "train_step": 596,
+      "loss": 0.63020346190361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.63020346190361
+    },
+    {
+      "train_step": 597,
+      "loss": 0.5842135197431614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5842135197431614
+    },
+    {
+      "train_step": 598,
+      "loss": 0.6114745076218919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6114745076218919
+    },
+    {
+      "train_step": 599,
+      "loss": 0.7687706204405772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7687706204405772
+    },
+    {
+      "train_step": 600,
+      "loss": 0.8220216685934567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8220216685934567
+    },
+    {
+      "train_step": 601,
+      "loss": 0.40638052861583523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40638052861583523
+    },
+    {
+      "train_step": 602,
+      "loss": 0.7130166555770293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7130166555770293
+    },
+    {
+      "train_step": 603,
+      "loss": 0.5528292522257037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5528292522257037
+    },
+    {
+      "train_step": 604,
+      "loss": 0.7330637601108538,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7330637601108538
+    },
+    {
+      "train_step": 605,
+      "loss": 0.5607867011147605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5607867011147605
+    },
+    {
+      "train_step": 606,
+      "loss": 0.6614447753319836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6614447753319836
+    },
+    {
+      "train_step": 607,
+      "loss": 0.6649032967689351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6649032967689351
+    },
+    {
+      "train_step": 608,
+      "loss": 0.6185406796428595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6185406796428595
+    },
+    {
+      "train_step": 609,
+      "loss": 0.6171938228872595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6171938228872595
+    },
+    {
+      "train_step": 610,
+      "loss": 0.5751214352066307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5751214352066307
+    },
+    {
+      "train_step": 611,
+      "loss": 0.5581626721443929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5581626721443929
+    },
+    {
+      "train_step": 612,
+      "loss": 0.5672919217260602,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5672919217260602
+    },
+    {
+      "train_step": 613,
+      "loss": 0.6559661954411321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6559661954411321
+    },
+    {
+      "train_step": 614,
+      "loss": 0.5435864539140842,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5435864539140842
+    },
+    {
+      "train_step": 615,
+      "loss": 0.5066059713092215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5066059713092215
+    },
+    {
+      "train_step": 616,
+      "loss": 0.6826143330896385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6826143330896385
+    },
+    {
+      "train_step": 617,
+      "loss": 0.509504267749155,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.509504267749155
+    },
+    {
+      "train_step": 618,
+      "loss": 0.6713114903123576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6713114903123576
+    },
+    {
+      "train_step": 619,
+      "loss": 0.5951776966229342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5951776966229342
+    },
+    {
+      "train_step": 620,
+      "loss": 0.5388449192504465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5388449192504465
+    },
+    {
+      "train_step": 621,
+      "loss": 0.5754286677449139,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5754286677449139
+    },
+    {
+      "train_step": 622,
+      "loss": 0.4112004845842573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4112004845842573
+    },
+    {
+      "train_step": 623,
+      "loss": 0.3538876543926621,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3538876543926621
+    },
+    {
+      "train_step": 624,
+      "loss": 0.6504453201026116,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6504453201026116
+    },
+    {
+      "train_step": 625,
+      "loss": 0.34004167797256496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34004167797256496
+    },
+    {
+      "train_step": 626,
+      "loss": 0.5493664381204458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5493664381204458
+    },
+    {
+      "train_step": 627,
+      "loss": 0.41323803504105516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41323803504105516
+    },
+    {
+      "train_step": 628,
+      "loss": 0.5102046036242586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5102046036242586
+    },
+    {
+      "train_step": 629,
+      "loss": 0.6305061859054073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6305061859054073
+    },
+    {
+      "train_step": 630,
+      "loss": 0.6044799462271011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6044799462271011
+    },
+    {
+      "train_step": 631,
+      "loss": 0.5181059926462612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5181059926462612
+    },
+    {
+      "train_step": 632,
+      "loss": 0.3632910770910304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3632910770910304
+    },
+    {
+      "train_step": 633,
+      "loss": 0.8931400972092356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8931400972092356
+    },
+    {
+      "train_step": 634,
+      "loss": 0.7237746961305018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7237746961305018
+    },
+    {
+      "train_step": 635,
+      "loss": 0.5144025476417514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5144025476417514
+    },
+    {
+      "train_step": 636,
+      "loss": 0.6125937320386313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6125937320386313
+    },
+    {
+      "train_step": 637,
+      "loss": 0.545689810415769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.545689810415769
+    },
+    {
+      "train_step": 638,
+      "loss": 0.6955075890408903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6955075890408903
+    },
+    {
+      "train_step": 639,
+      "loss": 0.5629546383840918,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5629546383840918
+    },
+    {
+      "train_step": 640,
+      "loss": 0.5229841381712728,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5229841381712728
+    },
+    {
+      "train_step": 641,
+      "loss": 0.5610817386828193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5610817386828193
+    },
+    {
+      "train_step": 642,
+      "loss": 0.8212989706720303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8212989706720303
+    },
+    {
+      "train_step": 643,
+      "loss": 0.6847075686512039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6847075686512039
+    },
+    {
+      "train_step": 644,
+      "loss": 0.5961445508356853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5961445508356853
+    },
+    {
+      "train_step": 645,
+      "loss": 0.5008530392037153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5008530392037153
+    },
+    {
+      "train_step": 646,
+      "loss": 0.8020802749741371,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8020802749741371
+    },
+    {
+      "train_step": 647,
+      "loss": 0.5674181528046324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5674181528046324
+    },
+    {
+      "train_step": 648,
+      "loss": 0.6745495421402334,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6745495421402334
+    },
+    {
+      "train_step": 649,
+      "loss": 0.7336602432559788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7336602432559788
+    },
+    {
+      "train_step": 650,
+      "loss": 0.7402911515246167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7402911515246167
+    },
+    {
+      "train_step": 651,
+      "loss": 0.6338261796197431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6338261796197431
+    },
+    {
+      "train_step": 652,
+      "loss": 0.6088736617438656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6088736617438656
+    },
+    {
+      "train_step": 653,
+      "loss": 0.5952982304558776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5952982304558776
+    },
+    {
+      "train_step": 654,
+      "loss": 0.7054811067028626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7054811067028626
+    },
+    {
+      "train_step": 655,
+      "loss": 0.6187014855997516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6187014855997516
+    },
+    {
+      "train_step": 656,
+      "loss": 0.5622835052114901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5622835052114901
+    },
+    {
+      "train_step": 657,
+      "loss": 0.589021963785529,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.589021963785529
+    },
+    {
+      "train_step": 658,
+      "loss": 0.4369653058989347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4369653058989347
+    },
+    {
+      "train_step": 659,
+      "loss": 0.4642430561854192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4642430561854192
+    },
+    {
+      "train_step": 660,
+      "loss": 0.6040046957106957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6040046957106957
+    },
+    {
+      "train_step": 661,
+      "loss": 0.5495125658644686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5495125658644686
+    },
+    {
+      "train_step": 662,
+      "loss": 0.5859853905247775,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5859853905247775
+    },
+    {
+      "train_step": 663,
+      "loss": 0.5857864380005122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5857864380005122
+    },
+    {
+      "train_step": 664,
+      "loss": 0.7315409503877545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7315409503877545
+    },
+    {
+      "train_step": 665,
+      "loss": 0.5810912877735124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5810912877735124
+    },
+    {
+      "train_step": 666,
+      "loss": 0.7243815411440337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7243815411440337
+    },
+    {
+      "train_step": 667,
+      "loss": 0.5319706087168731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5319706087168731
+    },
+    {
+      "train_step": 668,
+      "loss": 0.3585780538435849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3585780538435849
+    },
+    {
+      "train_step": 669,
+      "loss": 0.5182628152225426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5182628152225426
+    },
+    {
+      "train_step": 670,
+      "loss": 0.51041552421321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.51041552421321
+    },
+    {
+      "train_step": 671,
+      "loss": 0.4792161070740684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4792161070740684
+    },
+    {
+      "train_step": 672,
+      "loss": 0.6059494627373084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6059494627373084
+    },
+    {
+      "train_step": 673,
+      "loss": 0.7267684025144351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7267684025144351
+    },
+    {
+      "train_step": 674,
+      "loss": 0.48804039264586513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48804039264586513
+    },
+    {
+      "train_step": 675,
+      "loss": 0.53945146525725,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.53945146525725
+    },
+    {
+      "train_step": 676,
+      "loss": 0.6771191145418495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6771191145418495
+    },
+    {
+      "train_step": 677,
+      "loss": 0.48142003412729983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48142003412729983
+    },
+    {
+      "train_step": 678,
+      "loss": 0.5533797043964015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5533797043964015
+    },
+    {
+      "train_step": 679,
+      "loss": 0.620137607912069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.620137607912069
+    },
+    {
+      "train_step": 680,
+      "loss": 0.49167497074489985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49167497074489985
+    },
+    {
+      "train_step": 681,
+      "loss": 0.750557412467942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.750557412467942
+    },
+    {
+      "train_step": 682,
+      "loss": 0.4999811974166491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4999811974166491
+    },
+    {
+      "train_step": 683,
+      "loss": 0.3590485447415843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3590485447415843
+    },
+    {
+      "train_step": 684,
+      "loss": 0.4649327665984958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4649327665984958
+    },
+    {
+      "train_step": 685,
+      "loss": 0.49672075838028734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49672075838028734
+    },
+    {
+      "train_step": 686,
+      "loss": 0.8006439611877165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8006439611877165
+    },
+    {
+      "train_step": 687,
+      "loss": 0.5436859981056533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5436859981056533
+    },
+    {
+      "train_step": 688,
+      "loss": 0.6042885576728747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6042885576728747
+    },
+    {
+      "train_step": 689,
+      "loss": 0.56352446585834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.56352446585834
+    },
+    {
+      "train_step": 690,
+      "loss": 0.6062962760929127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6062962760929127
+    },
+    {
+      "train_step": 691,
+      "loss": 0.5727349257749528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5727349257749528
+    },
+    {
+      "train_step": 692,
+      "loss": 0.5028045327837299,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5028045327837299
+    },
+    {
+      "train_step": 693,
+      "loss": 0.33529037256767,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33529037256767
+    },
+    {
+      "train_step": 694,
+      "loss": 0.5105005599423945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5105005599423945
+    },
+    {
+      "train_step": 695,
+      "loss": 0.46303342996680164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46303342996680164
+    },
+    {
+      "train_step": 696,
+      "loss": 0.7133993751207882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7133993751207882
+    },
+    {
+      "train_step": 697,
+      "loss": 0.5164713889984514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5164713889984514
+    },
+    {
+      "train_step": 698,
+      "loss": 0.529375466994935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.529375466994935
+    },
+    {
+      "train_step": 699,
+      "loss": 0.7248608728237194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7248608728237194
+    },
+    {
+      "train_step": 700,
+      "loss": 0.5468740941559601,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5468740941559601
+    },
+    {
+      "train_step": 701,
+      "loss": 0.6778802509771948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6778802509771948
+    },
+    {
+      "train_step": 702,
+      "loss": 0.7113175199950544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7113175199950544
+    },
+    {
+      "train_step": 703,
+      "loss": 0.5198334359603097,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5198334359603097
+    },
+    {
+      "train_step": 704,
+      "loss": 0.6342826694800034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6342826694800034
+    },
+    {
+      "train_step": 705,
+      "loss": 0.828720482482538,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.828720482482538
+    },
+    {
+      "train_step": 706,
+      "loss": 0.7143159487018089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7143159487018089
+    },
+    {
+      "train_step": 707,
+      "loss": 0.6541624684776912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6541624684776912
+    },
+    {
+      "train_step": 708,
+      "loss": 0.6057100299924085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6057100299924085
+    },
+    {
+      "train_step": 709,
+      "loss": 0.7079873130074632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7079873130074632
+    },
+    {
+      "train_step": 710,
+      "loss": 0.3932832766680607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3932832766680607
+    },
+    {
+      "train_step": 711,
+      "loss": 0.5838225093559527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5838225093559527
+    },
+    {
+      "train_step": 712,
+      "loss": 0.8712019611689062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8712019611689062
+    },
+    {
+      "train_step": 713,
+      "loss": 0.6425981869795913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6425981869795913
+    },
+    {
+      "train_step": 714,
+      "loss": 0.4015640014653081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4015640014653081
+    },
+    {
+      "train_step": 715,
+      "loss": 0.6521833488572916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6521833488572916
+    },
+    {
+      "train_step": 716,
+      "loss": 0.7168224512915816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7168224512915816
+    },
+    {
+      "train_step": 717,
+      "loss": 0.6051247942363462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6051247942363462
+    },
+    {
+      "train_step": 718,
+      "loss": 0.4264967481024925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4264967481024925
+    },
+    {
+      "train_step": 719,
+      "loss": 0.6527374191867488,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6527374191867488
+    },
+    {
+      "train_step": 720,
+      "loss": 0.46452865277952526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46452865277952526
+    },
+    {
+      "train_step": 721,
+      "loss": 0.37226178308271834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37226178308271834
+    },
+    {
+      "train_step": 722,
+      "loss": 0.42019356204504504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42019356204504504
+    },
+    {
+      "train_step": 723,
+      "loss": 0.6487529336793574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6487529336793574
+    },
+    {
+      "train_step": 724,
+      "loss": 0.587310668051922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.587310668051922
+    },
+    {
+      "train_step": 725,
+      "loss": 0.5379898299014249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5379898299014249
+    },
+    {
+      "train_step": 726,
+      "loss": 0.5966888457069356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5966888457069356
+    },
+    {
+      "train_step": 727,
+      "loss": 0.8331352055955393,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8331352055955393
+    },
+    {
+      "train_step": 728,
+      "loss": 0.4953075509322125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4953075509322125
+    },
+    {
+      "train_step": 729,
+      "loss": 0.6829710416619555,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6829710416619555
+    },
+    {
+      "train_step": 730,
+      "loss": 0.5864212767577186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5864212767577186
+    },
+    {
+      "train_step": 731,
+      "loss": 0.4444959721930736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4444959721930736
+    },
+    {
+      "train_step": 732,
+      "loss": 0.4885391831429136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4885391831429136
+    },
+    {
+      "train_step": 733,
+      "loss": 0.5449450693066927,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5449450693066927
+    },
+    {
+      "train_step": 734,
+      "loss": 0.6812560319749912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6812560319749912
+    },
+    {
+      "train_step": 735,
+      "loss": 0.5249856975475417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5249856975475417
+    },
+    {
+      "train_step": 736,
+      "loss": 0.6755103873590089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6755103873590089
+    },
+    {
+      "train_step": 737,
+      "loss": 0.576537513066862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576537513066862
+    },
+    {
+      "train_step": 738,
+      "loss": 0.637760913509341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.637760913509341
+    },
+    {
+      "train_step": 739,
+      "loss": 0.6130667869834248,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6130667869834248
+    },
+    {
+      "train_step": 740,
+      "loss": 0.7395480159443812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7395480159443812
+    },
+    {
+      "train_step": 741,
+      "loss": 0.8317531170571545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8317531170571545
+    },
+    {
+      "train_step": 742,
+      "loss": 0.516202660371597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.516202660371597
+    },
+    {
+      "train_step": 743,
+      "loss": 0.7887756412490713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7887756412490713
+    },
+    {
+      "train_step": 744,
+      "loss": 0.5031197224648137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5031197224648137
+    },
+    {
+      "train_step": 745,
+      "loss": 0.4861547737141712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4861547737141712
+    },
+    {
+      "train_step": 746,
+      "loss": 0.6062773476046918,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6062773476046918
+    },
+    {
+      "train_step": 747,
+      "loss": 0.5223022877607469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5223022877607469
+    },
+    {
+      "train_step": 748,
+      "loss": 0.5964224079727517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5964224079727517
+    },
+    {
+      "train_step": 749,
+      "loss": 0.4836469873198953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4836469873198953
+    },
+    {
+      "train_step": 750,
+      "loss": 0.4292684898085408,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4292684898085408
+    },
+    {
+      "train_step": 751,
+      "loss": 0.6831914197787158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6831914197787158
+    },
+    {
+      "train_step": 752,
+      "loss": 0.6100824544518999,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6100824544518999
+    },
+    {
+      "train_step": 753,
+      "loss": 0.4819427116387227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4819427116387227
+    },
+    {
+      "train_step": 754,
+      "loss": 0.6510506386214023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6510506386214023
+    },
+    {
+      "train_step": 755,
+      "loss": 0.6942653040583036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6942653040583036
+    },
+    {
+      "train_step": 756,
+      "loss": 0.45228365667195025,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45228365667195025
+    },
+    {
+      "train_step": 757,
+      "loss": 0.7941494175073095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7941494175073095
+    },
+    {
+      "train_step": 758,
+      "loss": 0.5843238989857849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5843238989857849
+    },
+    {
+      "train_step": 759,
+      "loss": 0.44309976509838556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44309976509838556
+    },
+    {
+      "train_step": 760,
+      "loss": 0.6510946430435374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6510946430435374
+    },
+    {
+      "train_step": 761,
+      "loss": 0.5403627748823139,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5403627748823139
+    },
+    {
+      "train_step": 762,
+      "loss": 0.826984876723182,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.826984876723182
+    },
+    {
+      "train_step": 763,
+      "loss": 0.7417506649788608,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7417506649788608
+    },
+    {
+      "train_step": 764,
+      "loss": 0.6292271574662281,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6292271574662281
+    },
+    {
+      "train_step": 765,
+      "loss": 0.5804672054759761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5804672054759761
+    },
+    {
+      "train_step": 766,
+      "loss": 0.6763943436786604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6763943436786604
+    },
+    {
+      "train_step": 767,
+      "loss": 0.8321900094797018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8321900094797018
+    },
+    {
+      "train_step": 768,
+      "loss": 0.4206233379313322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4206233379313322
+    },
+    {
+      "train_step": 769,
+      "loss": 0.6145769720060649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6145769720060649
+    },
+    {
+      "train_step": 770,
+      "loss": 0.7294285525041776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7294285525041776
+    },
+    {
+      "train_step": 771,
+      "loss": 0.7090015104249451,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7090015104249451
+    },
+    {
+      "train_step": 772,
+      "loss": 0.5301492707655947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5301492707655947
+    },
+    {
+      "train_step": 773,
+      "loss": 0.6881574127837647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6881574127837647
+    },
+    {
+      "train_step": 774,
+      "loss": 0.43999406247767026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43999406247767026
+    },
+    {
+      "train_step": 775,
+      "loss": 0.7970392675252227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7970392675252227
+    },
+    {
+      "train_step": 776,
+      "loss": 0.41999419765855667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41999419765855667
+    },
+    {
+      "train_step": 777,
+      "loss": 0.9487606423378697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9487606423378697
+    },
+    {
+      "train_step": 778,
+      "loss": 0.6417765099893813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6417765099893813
+    },
+    {
+      "train_step": 779,
+      "loss": 0.6713988857360813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6713988857360813
+    },
+    {
+      "train_step": 780,
+      "loss": 0.6184704316474346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6184704316474346
+    },
+    {
+      "train_step": 781,
+      "loss": 0.5910475444080324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5910475444080324
+    },
+    {
+      "train_step": 782,
+      "loss": 0.38614293224318913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38614293224318913
+    },
+    {
+      "train_step": 783,
+      "loss": 0.6380804845156902,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6380804845156902
+    },
+    {
+      "train_step": 784,
+      "loss": 0.5611809846228235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5611809846228235
+    },
+    {
+      "train_step": 785,
+      "loss": 0.7116794705826024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7116794705826024
+    },
+    {
+      "train_step": 786,
+      "loss": 0.5569985620404726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5569985620404726
+    },
+    {
+      "train_step": 787,
+      "loss": 0.5482342472140502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5482342472140502
+    },
+    {
+      "train_step": 788,
+      "loss": 0.4145670689278167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4145670689278167
+    },
+    {
+      "train_step": 789,
+      "loss": 0.6249759736806655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6249759736806655
+    },
+    {
+      "train_step": 790,
+      "loss": 0.7124444527522481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7124444527522481
+    },
+    {
+      "train_step": 791,
+      "loss": 0.6334958539583314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6334958539583314
+    },
+    {
+      "train_step": 792,
+      "loss": 0.5531674787375752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5531674787375752
+    },
+    {
+      "train_step": 793,
+      "loss": 0.6308959402096255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6308959402096255
+    },
+    {
+      "train_step": 794,
+      "loss": 0.6331418333879684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6331418333879684
+    },
+    {
+      "train_step": 795,
+      "loss": 0.8474867773557895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8474867773557895
+    },
+    {
+      "train_step": 796,
+      "loss": 0.5971001312539425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5971001312539425
+    },
+    {
+      "train_step": 797,
+      "loss": 0.6795440217218696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6795440217218696
+    },
+    {
+      "train_step": 798,
+      "loss": 0.6656630438334769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6656630438334769
+    },
+    {
+      "train_step": 799,
+      "loss": 0.5377417168925612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5377417168925612
+    },
+    {
+      "train_step": 800,
+      "loss": 0.6725170836819476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6725170836819476
+    },
+    {
+      "train_step": 801,
+      "loss": 0.4553879925748762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4553879925748762
+    },
+    {
+      "train_step": 802,
+      "loss": 0.556356064698161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.556356064698161
+    },
+    {
+      "train_step": 803,
+      "loss": 0.6300186121600942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6300186121600942
+    },
+    {
+      "train_step": 804,
+      "loss": 0.6514343975935516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6514343975935516
+    },
+    {
+      "train_step": 805,
+      "loss": 0.6902655489066373,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6902655489066373
+    },
+    {
+      "train_step": 806,
+      "loss": 0.6494618932517185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6494618932517185
+    },
+    {
+      "train_step": 807,
+      "loss": 0.5586793602656562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5586793602656562
+    },
+    {
+      "train_step": 808,
+      "loss": 0.6240110700244907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6240110700244907
+    },
+    {
+      "train_step": 809,
+      "loss": 0.6232082198542214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6232082198542214
+    },
+    {
+      "train_step": 810,
+      "loss": 0.6286583971546554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6286583971546554
+    },
+    {
+      "train_step": 811,
+      "loss": 0.4828121379997241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4828121379997241
+    },
+    {
+      "train_step": 812,
+      "loss": 0.63499160934877,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.63499160934877
+    },
+    {
+      "train_step": 813,
+      "loss": 0.46499498316188037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46499498316188037
+    },
+    {
+      "train_step": 814,
+      "loss": 0.7165748041532489,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7165748041532489
+    },
+    {
+      "train_step": 815,
+      "loss": 0.5885222206138735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5885222206138735
+    },
+    {
+      "train_step": 816,
+      "loss": 0.44759776332085244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44759776332085244
+    },
+    {
+      "train_step": 817,
+      "loss": 0.4432005460706524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4432005460706524
+    },
+    {
+      "train_step": 818,
+      "loss": 0.5677001434260828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5677001434260828
+    },
+    {
+      "train_step": 819,
+      "loss": 0.7171260232959067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7171260232959067
+    },
+    {
+      "train_step": 820,
+      "loss": 0.6653126778574261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6653126778574261
+    },
+    {
+      "train_step": 821,
+      "loss": 0.8141894452773566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8141894452773566
+    },
+    {
+      "train_step": 822,
+      "loss": 0.6777001361014389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6777001361014389
+    },
+    {
+      "train_step": 823,
+      "loss": 0.6525439793533551,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6525439793533551
+    },
+    {
+      "train_step": 824,
+      "loss": 0.4371347332920895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4371347332920895
+    },
+    {
+      "train_step": 825,
+      "loss": 0.7408360023067411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7408360023067411
+    },
+    {
+      "train_step": 826,
+      "loss": 0.6620162162555089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6620162162555089
+    },
+    {
+      "train_step": 827,
+      "loss": 0.4828272791036091,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4828272791036091
+    },
+    {
+      "train_step": 828,
+      "loss": 0.6112125108164038,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6112125108164038
+    },
+    {
+      "train_step": 829,
+      "loss": 0.5484915054114565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5484915054114565
+    },
+    {
+      "train_step": 830,
+      "loss": 0.7536236689004334,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7536236689004334
+    },
+    {
+      "train_step": 831,
+      "loss": 0.79612776455693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.79612776455693
+    },
+    {
+      "train_step": 832,
+      "loss": 0.8141994906657495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8141994906657495
+    },
+    {
+      "train_step": 833,
+      "loss": 0.5134048479961921,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5134048479961921
+    },
+    {
+      "train_step": 834,
+      "loss": 0.5125336081387644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5125336081387644
+    },
+    {
+      "train_step": 835,
+      "loss": 0.708334150160697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.708334150160697
+    },
+    {
+      "train_step": 836,
+      "loss": 0.6397032058886539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6397032058886539
+    },
+    {
+      "train_step": 837,
+      "loss": 0.3930643458132378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3930643458132378
+    },
+    {
+      "train_step": 838,
+      "loss": 0.6408114681983446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6408114681983446
+    },
+    {
+      "train_step": 839,
+      "loss": 0.8139103027570471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8139103027570471
+    },
+    {
+      "train_step": 840,
+      "loss": 0.5400085701900772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5400085701900772
+    },
+    {
+      "train_step": 841,
+      "loss": 0.49203155975666596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49203155975666596
+    },
+    {
+      "train_step": 842,
+      "loss": 0.5365436353574653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5365436353574653
+    },
+    {
+      "train_step": 843,
+      "loss": 0.7110630490795703,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7110630490795703
+    },
+    {
+      "train_step": 844,
+      "loss": 0.602611912423065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.602611912423065
+    },
+    {
+      "train_step": 845,
+      "loss": 0.6013524823837033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6013524823837033
+    },
+    {
+      "train_step": 846,
+      "loss": 0.6887946019608744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6887946019608744
+    },
+    {
+      "train_step": 847,
+      "loss": 0.5297516291386086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5297516291386086
+    },
+    {
+      "train_step": 848,
+      "loss": 0.33805818686315203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33805818686315203
+    },
+    {
+      "train_step": 849,
+      "loss": 0.589838534003435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.589838534003435
+    },
+    {
+      "train_step": 850,
+      "loss": 0.6327830226398788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6327830226398788
+    },
+    {
+      "train_step": 851,
+      "loss": 0.7047810594058392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7047810594058392
+    },
+    {
+      "train_step": 852,
+      "loss": 0.6112442883577311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6112442883577311
+    },
+    {
+      "train_step": 853,
+      "loss": 0.573387031724469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.573387031724469
+    },
+    {
+      "train_step": 854,
+      "loss": 0.42024024702314167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42024024702314167
+    },
+    {
+      "train_step": 855,
+      "loss": 0.4993143012873326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4993143012873326
+    },
+    {
+      "train_step": 856,
+      "loss": 0.6783494526136526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6783494526136526
+    },
+    {
+      "train_step": 857,
+      "loss": 0.517651796185952,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.517651796185952
+    },
+    {
+      "train_step": 858,
+      "loss": 0.7243068946951576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7243068946951576
+    },
+    {
+      "train_step": 859,
+      "loss": 0.5006506790929768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5006506790929768
+    },
+    {
+      "train_step": 860,
+      "loss": 0.6865212269786002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6865212269786002
+    },
+    {
+      "train_step": 861,
+      "loss": 0.45528487496691333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45528487496691333
+    },
+    {
+      "train_step": 862,
+      "loss": 0.6296904937329653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6296904937329653
+    },
+    {
+      "train_step": 863,
+      "loss": 0.7331554426145436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7331554426145436
+    },
+    {
+      "train_step": 864,
+      "loss": 0.38839759070012686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38839759070012686
+    },
+    {
+      "train_step": 865,
+      "loss": 0.7245250152224372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7245250152224372
+    },
+    {
+      "train_step": 866,
+      "loss": 0.5009739077024815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5009739077024815
+    },
+    {
+      "train_step": 867,
+      "loss": 0.6085038828887404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6085038828887404
+    },
+    {
+      "train_step": 868,
+      "loss": 0.6166511602766933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6166511602766933
+    },
+    {
+      "train_step": 869,
+      "loss": 0.4593065053100382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4593065053100382
+    },
+    {
+      "train_step": 870,
+      "loss": 0.7881825715956611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7881825715956611
+    },
+    {
+      "train_step": 871,
+      "loss": 0.7287977021916603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7287977021916603
+    },
+    {
+      "train_step": 872,
+      "loss": 0.5335558342859484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5335558342859484
+    },
+    {
+      "train_step": 873,
+      "loss": 0.5652937608705715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5652937608705715
+    },
+    {
+      "train_step": 874,
+      "loss": 0.6482902243342201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6482902243342201
+    },
+    {
+      "train_step": 875,
+      "loss": 0.6382206824635265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6382206824635265
+    },
+    {
+      "train_step": 876,
+      "loss": 0.4824701981284659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4824701981284659
+    },
+    {
+      "train_step": 877,
+      "loss": 0.5770684618451684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5770684618451684
+    },
+    {
+      "train_step": 878,
+      "loss": 0.5717452346353095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5717452346353095
+    },
+    {
+      "train_step": 879,
+      "loss": 0.5039496155144333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5039496155144333
+    },
+    {
+      "train_step": 880,
+      "loss": 0.5486304118375865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5486304118375865
+    },
+    {
+      "train_step": 881,
+      "loss": 0.6632556103675055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6632556103675055
+    },
+    {
+      "train_step": 882,
+      "loss": 0.5393311977004361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5393311977004361
+    },
+    {
+      "train_step": 883,
+      "loss": 0.315309571940639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.315309571940639
+    },
+    {
+      "train_step": 884,
+      "loss": 0.5543209749865423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5543209749865423
+    },
+    {
+      "train_step": 885,
+      "loss": 0.566469349617158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.566469349617158
+    },
+    {
+      "train_step": 886,
+      "loss": 0.7118819185088189,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7118819185088189
+    },
+    {
+      "train_step": 887,
+      "loss": 0.49823670580135515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49823670580135515
+    },
+    {
+      "train_step": 888,
+      "loss": 0.6020658132289196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6020658132289196
+    },
+    {
+      "train_step": 889,
+      "loss": 0.5872816812850473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5872816812850473
+    },
+    {
+      "train_step": 890,
+      "loss": 0.5677270301782633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5677270301782633
+    },
+    {
+      "train_step": 891,
+      "loss": 0.7570815148664497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7570815148664497
+    },
+    {
+      "train_step": 892,
+      "loss": 0.6773698123552944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6773698123552944
+    },
+    {
+      "train_step": 893,
+      "loss": 0.6482476054736793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6482476054736793
+    },
+    {
+      "train_step": 894,
+      "loss": 0.6834294917788499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6834294917788499
+    },
+    {
+      "train_step": 895,
+      "loss": 0.6087473462643084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6087473462643084
+    },
+    {
+      "train_step": 896,
+      "loss": 0.5953252595620765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5953252595620765
+    },
+    {
+      "train_step": 897,
+      "loss": 0.5067941336584711,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5067941336584711
+    },
+    {
+      "train_step": 898,
+      "loss": 0.6794473109199813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6794473109199813
+    },
+    {
+      "train_step": 899,
+      "loss": 0.8521124579540649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8521124579540649
+    },
+    {
+      "train_step": 900,
+      "loss": 0.5252656343376381,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5252656343376381
+    },
+    {
+      "train_step": 901,
+      "loss": 0.6674673139239379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6674673139239379
+    },
+    {
+      "train_step": 902,
+      "loss": 0.6923597820459334,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6923597820459334
+    },
+    {
+      "train_step": 903,
+      "loss": 0.6104542024972943,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6104542024972943
+    },
+    {
+      "train_step": 904,
+      "loss": 0.3777958344419668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3777958344419668
+    },
+    {
+      "train_step": 905,
+      "loss": 0.8905123679903524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8905123679903524
+    },
+    {
+      "train_step": 906,
+      "loss": 0.922743173242675,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.922743173242675
+    },
+    {
+      "train_step": 907,
+      "loss": 0.6730964475567931,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6730964475567931
+    },
+    {
+      "train_step": 908,
+      "loss": 0.7294597150390298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7294597150390298
+    },
+    {
+      "train_step": 909,
+      "loss": 0.6457843788640008,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6457843788640008
+    },
+    {
+      "train_step": 910,
+      "loss": 0.712403493648226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.712403493648226
+    },
+    {
+      "train_step": 911,
+      "loss": 0.5673788244524276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5673788244524276
+    },
+    {
+      "train_step": 912,
+      "loss": 0.4331708663020697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4331708663020697
+    },
+    {
+      "train_step": 913,
+      "loss": 0.4705710253445029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4705710253445029
+    },
+    {
+      "train_step": 914,
+      "loss": 0.7422499876422695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7422499876422695
+    },
+    {
+      "train_step": 915,
+      "loss": 0.7294930156478526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7294930156478526
+    },
+    {
+      "train_step": 916,
+      "loss": 0.47134453612047084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47134453612047084
+    },
+    {
+      "train_step": 917,
+      "loss": 0.7113715891736234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7113715891736234
+    },
+    {
+      "train_step": 918,
+      "loss": 0.5453186575015724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5453186575015724
+    },
+    {
+      "train_step": 919,
+      "loss": 0.5623178566992533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5623178566992533
+    },
+    {
+      "train_step": 920,
+      "loss": 0.5108289179662325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5108289179662325
+    },
+    {
+      "train_step": 921,
+      "loss": 0.6190510249111394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6190510249111394
+    },
+    {
+      "train_step": 922,
+      "loss": 0.5852041843601996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5852041843601996
+    },
+    {
+      "train_step": 923,
+      "loss": 0.44609673525762966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44609673525762966
+    },
+    {
+      "train_step": 924,
+      "loss": 0.6080316391372095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6080316391372095
+    },
+    {
+      "train_step": 925,
+      "loss": 0.5253840454151381,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5253840454151381
+    },
+    {
+      "train_step": 926,
+      "loss": 0.4938736768365098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4938736768365098
+    },
+    {
+      "train_step": 927,
+      "loss": 0.5567046666423245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5567046666423245
+    },
+    {
+      "train_step": 928,
+      "loss": 0.45576596327790103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45576596327790103
+    },
+    {
+      "train_step": 929,
+      "loss": 0.6128257612931093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6128257612931093
+    },
+    {
+      "train_step": 930,
+      "loss": 0.5142051632520023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5142051632520023
+    },
+    {
+      "train_step": 931,
+      "loss": 0.7250353769505213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7250353769505213
+    },
+    {
+      "train_step": 932,
+      "loss": 0.7012329003033684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7012329003033684
+    },
+    {
+      "train_step": 933,
+      "loss": 0.526495850383518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.526495850383518
+    },
+    {
+      "train_step": 934,
+      "loss": 0.5044488105489789,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5044488105489789
+    },
+    {
+      "train_step": 935,
+      "loss": 0.5622060570049251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5622060570049251
+    },
+    {
+      "train_step": 936,
+      "loss": 0.45526652484637364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45526652484637364
+    },
+    {
+      "train_step": 937,
+      "loss": 0.5159243628718293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5159243628718293
+    },
+    {
+      "train_step": 938,
+      "loss": 0.6494626505849256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6494626505849256
+    },
+    {
+      "train_step": 939,
+      "loss": 0.2718848913899297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2718848913899297
+    },
+    {
+      "train_step": 940,
+      "loss": 0.4845817566552508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4845817566552508
+    },
+    {
+      "train_step": 941,
+      "loss": 0.40894699929556116,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40894699929556116
+    },
+    {
+      "train_step": 942,
+      "loss": 0.7373931845295445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7373931845295445
+    },
+    {
+      "train_step": 943,
+      "loss": 0.5265431598170579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5265431598170579
+    },
+    {
+      "train_step": 944,
+      "loss": 0.5626253514137319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5626253514137319
+    },
+    {
+      "train_step": 945,
+      "loss": 0.5612831942720494,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5612831942720494
+    },
+    {
+      "train_step": 946,
+      "loss": 0.5774689770441305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5774689770441305
+    },
+    {
+      "train_step": 947,
+      "loss": 0.4465387299284067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4465387299284067
+    },
+    {
+      "train_step": 948,
+      "loss": 0.633515100905678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.633515100905678
+    },
+    {
+      "train_step": 949,
+      "loss": 0.5054008662825776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5054008662825776
+    },
+    {
+      "train_step": 950,
+      "loss": 0.7928067173021989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7928067173021989
+    },
+    {
+      "train_step": 951,
+      "loss": 0.7121877607135522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7121877607135522
+    },
+    {
+      "train_step": 952,
+      "loss": 0.6981107619519415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6981107619519415
+    },
+    {
+      "train_step": 953,
+      "loss": 0.7055007327818456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7055007327818456
+    },
+    {
+      "train_step": 954,
+      "loss": 0.6137986270213971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6137986270213971
+    },
+    {
+      "train_step": 955,
+      "loss": 0.5766255964603761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5766255964603761
+    },
+    {
+      "train_step": 956,
+      "loss": 0.5035629108365197,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035629108365197
+    },
+    {
+      "train_step": 957,
+      "loss": 0.5661892102698394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5661892102698394
+    },
+    {
+      "train_step": 958,
+      "loss": 0.43565663981357916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43565663981357916
+    },
+    {
+      "train_step": 959,
+      "loss": 0.8064421335511578,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8064421335511578
+    },
+    {
+      "train_step": 960,
+      "loss": 0.6374707253480432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6374707253480432
+    },
+    {
+      "train_step": 961,
+      "loss": 0.8203768780582458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8203768780582458
+    },
+    {
+      "train_step": 962,
+      "loss": 0.48521958009705457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48521958009705457
+    },
+    {
+      "train_step": 963,
+      "loss": 0.8281928240911737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8281928240911737
+    },
+    {
+      "train_step": 964,
+      "loss": 0.5772343638554993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5772343638554993
+    },
+    {
+      "train_step": 965,
+      "loss": 0.5979298260269412,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5979298260269412
+    },
+    {
+      "train_step": 966,
+      "loss": 0.6441742200197874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6441742200197874
+    },
+    {
+      "train_step": 967,
+      "loss": 0.6375381257718291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6375381257718291
+    },
+    {
+      "train_step": 968,
+      "loss": 0.4795526742706291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4795526742706291
+    },
+    {
+      "train_step": 969,
+      "loss": 0.5375192788915341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5375192788915341
+    },
+    {
+      "train_step": 970,
+      "loss": 0.5486316402829068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5486316402829068
+    },
+    {
+      "train_step": 971,
+      "loss": 0.6470110711360243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6470110711360243
+    },
+    {
+      "train_step": 972,
+      "loss": 0.5461999131353803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5461999131353803
+    },
+    {
+      "train_step": 973,
+      "loss": 0.6113093413425509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6113093413425509
+    },
+    {
+      "train_step": 974,
+      "loss": 0.6821959078754289,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6821959078754289
+    },
+    {
+      "train_step": 975,
+      "loss": 0.7902617304168378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7902617304168378
+    },
+    {
+      "train_step": 976,
+      "loss": 0.6566970156078166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6566970156078166
+    },
+    {
+      "train_step": 977,
+      "loss": 0.6282632631367626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6282632631367626
+    },
+    {
+      "train_step": 978,
+      "loss": 0.596352334786958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.596352334786958
+    },
+    {
+      "train_step": 979,
+      "loss": 0.7152047957992705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7152047957992705
+    },
+    {
+      "train_step": 980,
+      "loss": 0.5145122173619353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5145122173619353
+    },
+    {
+      "train_step": 981,
+      "loss": 0.5147937110399371,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5147937110399371
+    },
+    {
+      "train_step": 982,
+      "loss": 0.5802696852443963,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5802696852443963
+    },
+    {
+      "train_step": 983,
+      "loss": 0.4459977115857478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4459977115857478
+    },
+    {
+      "train_step": 984,
+      "loss": 0.6228112201402785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6228112201402785
+    },
+    {
+      "train_step": 985,
+      "loss": 0.7632068217897576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7632068217897576
+    },
+    {
+      "train_step": 986,
+      "loss": 0.4609905660434219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4609905660434219
+    },
+    {
+      "train_step": 987,
+      "loss": 0.5900026847719829,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5900026847719829
+    },
+    {
+      "train_step": 988,
+      "loss": 0.823402621668426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.823402621668426
+    },
+    {
+      "train_step": 989,
+      "loss": 0.6381615138224359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6381615138224359
+    },
+    {
+      "train_step": 990,
+      "loss": 0.4655677020870473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4655677020870473
+    },
+    {
+      "train_step": 991,
+      "loss": 0.720331218665702,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.720331218665702
+    },
+    {
+      "train_step": 992,
+      "loss": 0.7977212518463558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7977212518463558
+    },
+    {
+      "train_step": 993,
+      "loss": 0.5679537445706544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5679537445706544
+    },
+    {
+      "train_step": 994,
+      "loss": 0.42422921638954636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42422921638954636
+    },
+    {
+      "train_step": 995,
+      "loss": 0.406938491118441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.406938491118441
+    },
+    {
+      "train_step": 996,
+      "loss": 0.6575719023395531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6575719023395531
+    },
+    {
+      "train_step": 997,
+      "loss": 0.44969496758908645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44969496758908645
+    },
+    {
+      "train_step": 998,
+      "loss": 0.5076852929918216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5076852929918216
+    },
+    {
+      "train_step": 999,
+      "loss": 0.6410522282989027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6410522282989027
+    },
+    {
+      "train_step": 1000,
+      "loss": 0.5241950125337081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5241950125337081
+    },
+    {
+      "train_step": 1001,
+      "loss": 0.46967277694975235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46967277694975235
+    },
+    {
+      "train_step": 1002,
+      "loss": 0.49483436846594697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49483436846594697
+    },
+    {
+      "train_step": 1003,
+      "loss": 0.722723326990294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.722723326990294
+    },
+    {
+      "train_step": 1004,
+      "loss": 0.5045324581705561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5045324581705561
+    },
+    {
+      "train_step": 1005,
+      "loss": 0.5985029338899334,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5985029338899334
+    },
+    {
+      "train_step": 1006,
+      "loss": 0.5886019024853982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5886019024853982
+    },
+    {
+      "train_step": 1007,
+      "loss": 0.7115394798904353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7115394798904353
+    },
+    {
+      "train_step": 1008,
+      "loss": 0.5948716783216319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5948716783216319
+    },
+    {
+      "train_step": 1009,
+      "loss": 0.6462917994989599,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6462917994989599
+    },
+    {
+      "train_step": 1010,
+      "loss": 0.690305581332118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.690305581332118
+    },
+    {
+      "train_step": 1011,
+      "loss": 0.7203630771774592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7203630771774592
+    },
+    {
+      "train_step": 1012,
+      "loss": 0.5678855297247072,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5678855297247072
+    },
+    {
+      "train_step": 1013,
+      "loss": 0.47132661066037096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47132661066037096
+    },
+    {
+      "train_step": 1014,
+      "loss": 0.6787307945785224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6787307945785224
+    },
+    {
+      "train_step": 1015,
+      "loss": 0.7384552159555279,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7384552159555279
+    },
+    {
+      "train_step": 1016,
+      "loss": 0.5360499069177562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5360499069177562
+    },
+    {
+      "train_step": 1017,
+      "loss": 0.7309176738536758,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7309176738536758
+    },
+    {
+      "train_step": 1018,
+      "loss": 0.5594918461324698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5594918461324698
+    },
+    {
+      "train_step": 1019,
+      "loss": 0.36244974510877426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36244974510877426
+    },
+    {
+      "train_step": 1020,
+      "loss": 0.6399667307085888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6399667307085888
+    },
+    {
+      "train_step": 1021,
+      "loss": 0.43967738631209213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43967738631209213
+    },
+    {
+      "train_step": 1022,
+      "loss": 0.9094124159011298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9094124159011298
+    },
+    {
+      "train_step": 1023,
+      "loss": 0.5871442615908096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5871442615908096
+    },
+    {
+      "train_step": 1024,
+      "loss": 0.5348323124255014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5348323124255014
+    },
+    {
+      "train_step": 1025,
+      "loss": 0.63997413129949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.63997413129949
+    },
+    {
+      "train_step": 1026,
+      "loss": 0.5361296422271816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5361296422271816
+    },
+    {
+      "train_step": 1027,
+      "loss": 0.5639159957720203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5639159957720203
+    },
+    {
+      "train_step": 1028,
+      "loss": 0.7732214902432452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7732214902432452
+    },
+    {
+      "train_step": 1029,
+      "loss": 0.6774311798552501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6774311798552501
+    },
+    {
+      "train_step": 1030,
+      "loss": 0.5421065932443047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5421065932443047
+    },
+    {
+      "train_step": 1031,
+      "loss": 0.6383882149315616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6383882149315616
+    },
+    {
+      "train_step": 1032,
+      "loss": 0.5589092587575979,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5589092587575979
+    },
+    {
+      "train_step": 1033,
+      "loss": 0.5129066538527044,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5129066538527044
+    },
+    {
+      "train_step": 1034,
+      "loss": 0.8157668036558567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8157668036558567
+    },
+    {
+      "train_step": 1035,
+      "loss": 0.4820286664044204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4820286664044204
+    },
+    {
+      "train_step": 1036,
+      "loss": 0.7286418700929074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7286418700929074
+    },
+    {
+      "train_step": 1037,
+      "loss": 0.5090935238468035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5090935238468035
+    },
+    {
+      "train_step": 1038,
+      "loss": 0.33678218920970365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33678218920970365
+    },
+    {
+      "train_step": 1039,
+      "loss": 0.6817711209647018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6817711209647018
+    },
+    {
+      "train_step": 1040,
+      "loss": 0.6716317441989764,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6716317441989764
+    },
+    {
+      "train_step": 1041,
+      "loss": 0.5013660921999564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5013660921999564
+    },
+    {
+      "train_step": 1042,
+      "loss": 0.5763611902884554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5763611902884554
+    },
+    {
+      "train_step": 1043,
+      "loss": 0.5838840733644239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5838840733644239
+    },
+    {
+      "train_step": 1044,
+      "loss": 0.5488998622348195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5488998622348195
+    },
+    {
+      "train_step": 1045,
+      "loss": 0.33852692423895103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33852692423895103
+    },
+    {
+      "train_step": 1046,
+      "loss": 0.6377174107277733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6377174107277733
+    },
+    {
+      "train_step": 1047,
+      "loss": 0.6815719863645308,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6815719863645308
+    },
+    {
+      "train_step": 1048,
+      "loss": 0.6950502237929045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6950502237929045
+    },
+    {
+      "train_step": 1049,
+      "loss": 0.5324343730694677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5324343730694677
+    },
+    {
+      "train_step": 1050,
+      "loss": 0.7176696836184827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7176696836184827
+    },
+    {
+      "train_step": 1051,
+      "loss": 0.5967916764964801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5967916764964801
+    },
+    {
+      "train_step": 1052,
+      "loss": 0.571502990358326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.571502990358326
+    },
+    {
+      "train_step": 1053,
+      "loss": 0.6631590630664933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6631590630664933
+    },
+    {
+      "train_step": 1054,
+      "loss": 0.4178339724041274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4178339724041274
+    },
+    {
+      "train_step": 1055,
+      "loss": 0.646712457930327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.646712457930327
+    },
+    {
+      "train_step": 1056,
+      "loss": 0.7261638492903912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7261638492903912
+    },
+    {
+      "train_step": 1057,
+      "loss": 0.601517637732616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.601517637732616
+    },
+    {
+      "train_step": 1058,
+      "loss": 0.388745538982796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.388745538982796
+    },
+    {
+      "train_step": 1059,
+      "loss": 0.6760616091004883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6760616091004883
+    },
+    {
+      "train_step": 1060,
+      "loss": 0.5706084967373275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706084967373275
+    },
+    {
+      "train_step": 1061,
+      "loss": 0.9184468648837297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9184468648837297
+    },
+    {
+      "train_step": 1062,
+      "loss": 0.3946516857433466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3946516857433466
+    },
+    {
+      "train_step": 1063,
+      "loss": 0.5291181436536679,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5291181436536679
+    },
+    {
+      "train_step": 1064,
+      "loss": 0.5062861031724093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5062861031724093
+    },
+    {
+      "train_step": 1065,
+      "loss": 0.5204020294043791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5204020294043791
+    },
+    {
+      "train_step": 1066,
+      "loss": 0.6610169140343244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6610169140343244
+    },
+    {
+      "train_step": 1067,
+      "loss": 0.6120022098947104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6120022098947104
+    },
+    {
+      "train_step": 1068,
+      "loss": 0.5769662930197177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5769662930197177
+    },
+    {
+      "train_step": 1069,
+      "loss": 0.6092741594974382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6092741594974382
+    },
+    {
+      "train_step": 1070,
+      "loss": 0.49922258749594894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49922258749594894
+    },
+    {
+      "train_step": 1071,
+      "loss": 0.6331663520744617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6331663520744617
+    },
+    {
+      "train_step": 1072,
+      "loss": 0.5789621936551564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5789621936551564
+    },
+    {
+      "train_step": 1073,
+      "loss": 0.4852069698160894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4852069698160894
+    },
+    {
+      "train_step": 1074,
+      "loss": 0.5929660354195592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5929660354195592
+    },
+    {
+      "train_step": 1075,
+      "loss": 0.5930460931741265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5930460931741265
+    },
+    {
+      "train_step": 1076,
+      "loss": 0.4888088011408592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4888088011408592
+    },
+    {
+      "train_step": 1077,
+      "loss": 0.7502581613257783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7502581613257783
+    },
+    {
+      "train_step": 1078,
+      "loss": 0.7303487577705892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7303487577705892
+    },
+    {
+      "train_step": 1079,
+      "loss": 0.6426103976635514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6426103976635514
+    },
+    {
+      "train_step": 1080,
+      "loss": 0.6376572273301914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6376572273301914
+    },
+    {
+      "train_step": 1081,
+      "loss": 0.8611897454047157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8611897454047157
+    },
+    {
+      "train_step": 1082,
+      "loss": 0.5142430813411452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5142430813411452
+    },
+    {
+      "train_step": 1083,
+      "loss": 0.6315800898761387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6315800898761387
+    },
+    {
+      "train_step": 1084,
+      "loss": 0.4415586745215316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4415586745215316
+    },
+    {
+      "train_step": 1085,
+      "loss": 0.6305069976453471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6305069976453471
+    },
+    {
+      "train_step": 1086,
+      "loss": 0.6503943874678284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6503943874678284
+    },
+    {
+      "train_step": 1087,
+      "loss": 0.5165503073925242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5165503073925242
+    },
+    {
+      "train_step": 1088,
+      "loss": 0.8034044448059198,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8034044448059198
+    },
+    {
+      "train_step": 1089,
+      "loss": 0.5965075908847118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5965075908847118
+    },
+    {
+      "train_step": 1090,
+      "loss": 0.44928945664474634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44928945664474634
+    },
+    {
+      "train_step": 1091,
+      "loss": 0.45081617305046084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45081617305046084
+    },
+    {
+      "train_step": 1092,
+      "loss": 0.5358164346535895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5358164346535895
+    },
+    {
+      "train_step": 1093,
+      "loss": 0.5398643522310878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5398643522310878
+    },
+    {
+      "train_step": 1094,
+      "loss": 0.7290698414053607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7290698414053607
+    },
+    {
+      "train_step": 1095,
+      "loss": 0.4458248053044836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4458248053044836
+    },
+    {
+      "train_step": 1096,
+      "loss": 0.6926462818789969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6926462818789969
+    },
+    {
+      "train_step": 1097,
+      "loss": 0.3810155528173554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3810155528173554
+    },
+    {
+      "train_step": 1098,
+      "loss": 0.8743392793031036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8743392793031036
+    },
+    {
+      "train_step": 1099,
+      "loss": 0.6167363150022495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6167363150022495
+    },
+    {
+      "train_step": 1100,
+      "loss": 0.6711536692181607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6711536692181607
+    },
+    {
+      "train_step": 1101,
+      "loss": 0.5710508262966961,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5710508262966961
+    },
+    {
+      "train_step": 1102,
+      "loss": 0.5873874076100161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5873874076100161
+    },
+    {
+      "train_step": 1103,
+      "loss": 0.6438588973094744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6438588973094744
+    },
+    {
+      "train_step": 1104,
+      "loss": 0.5663235230452544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5663235230452544
+    },
+    {
+      "train_step": 1105,
+      "loss": 0.49934746799336105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49934746799336105
+    },
+    {
+      "train_step": 1106,
+      "loss": 0.7100392058176915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7100392058176915
+    },
+    {
+      "train_step": 1107,
+      "loss": 0.4442572329757179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4442572329757179
+    },
+    {
+      "train_step": 1108,
+      "loss": 0.6983707917676666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6983707917676666
+    },
+    {
+      "train_step": 1109,
+      "loss": 0.7637614613897459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7637614613897459
+    },
+    {
+      "train_step": 1110,
+      "loss": 0.5159089294651976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5159089294651976
+    },
+    {
+      "train_step": 1111,
+      "loss": 0.6268297100457835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6268297100457835
+    },
+    {
+      "train_step": 1112,
+      "loss": 0.648248381203428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.648248381203428
+    },
+    {
+      "train_step": 1113,
+      "loss": 0.513238628541864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.513238628541864
+    },
+    {
+      "train_step": 1114,
+      "loss": 0.5859789940226261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5859789940226261
+    },
+    {
+      "train_step": 1115,
+      "loss": 0.5589877015843894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5589877015843894
+    },
+    {
+      "train_step": 1116,
+      "loss": 0.4910955648167391,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4910955648167391
+    },
+    {
+      "train_step": 1117,
+      "loss": 0.6029976869053464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6029976869053464
+    },
+    {
+      "train_step": 1118,
+      "loss": 0.6283528363207858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6283528363207858
+    },
+    {
+      "train_step": 1119,
+      "loss": 0.5747945769183755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5747945769183755
+    },
+    {
+      "train_step": 1120,
+      "loss": 0.6176213243279122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6176213243279122
+    },
+    {
+      "train_step": 1121,
+      "loss": 0.7842119867632649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7842119867632649
+    },
+    {
+      "train_step": 1122,
+      "loss": 0.5293018387587769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5293018387587769
+    },
+    {
+      "train_step": 1123,
+      "loss": 0.8104099353543849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8104099353543849
+    },
+    {
+      "train_step": 1124,
+      "loss": 0.4970432455618348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4970432455618348
+    },
+    {
+      "train_step": 1125,
+      "loss": 0.5006429527334986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5006429527334986
+    },
+    {
+      "train_step": 1126,
+      "loss": 0.6986692191016558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6986692191016558
+    },
+    {
+      "train_step": 1127,
+      "loss": 0.7332765580417819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7332765580417819
+    },
+    {
+      "train_step": 1128,
+      "loss": 0.5033505262898624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5033505262898624
+    },
+    {
+      "train_step": 1129,
+      "loss": 0.8112870414032726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8112870414032726
+    },
+    {
+      "train_step": 1130,
+      "loss": 0.603464374669498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.603464374669498
+    },
+    {
+      "train_step": 1131,
+      "loss": 0.7432711314753293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7432711314753293
+    },
+    {
+      "train_step": 1132,
+      "loss": 0.877144283020517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.877144283020517
+    },
+    {
+      "train_step": 1133,
+      "loss": 0.6108739277783616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6108739277783616
+    },
+    {
+      "train_step": 1134,
+      "loss": 0.6107306194784885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6107306194784885
+    },
+    {
+      "train_step": 1135,
+      "loss": 0.4370806891794164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4370806891794164
+    },
+    {
+      "train_step": 1136,
+      "loss": 0.5725131917902386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5725131917902386
+    },
+    {
+      "train_step": 1137,
+      "loss": 0.6188363135322137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6188363135322137
+    },
+    {
+      "train_step": 1138,
+      "loss": 0.887005314667841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.887005314667841
+    },
+    {
+      "train_step": 1139,
+      "loss": 0.5354264146623846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5354264146623846
+    },
+    {
+      "train_step": 1140,
+      "loss": 0.5645837653546592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5645837653546592
+    },
+    {
+      "train_step": 1141,
+      "loss": 0.6763555832000658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6763555832000658
+    },
+    {
+      "train_step": 1142,
+      "loss": 0.5267087154652409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5267087154652409
+    },
+    {
+      "train_step": 1143,
+      "loss": 0.6377352868250791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6377352868250791
+    },
+    {
+      "train_step": 1144,
+      "loss": 0.7595553353490861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7595553353490861
+    },
+    {
+      "train_step": 1145,
+      "loss": 0.6418006035604491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6418006035604491
+    },
+    {
+      "train_step": 1146,
+      "loss": 0.6696976828512595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6696976828512595
+    },
+    {
+      "train_step": 1147,
+      "loss": 0.5816458502941682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5816458502941682
+    },
+    {
+      "train_step": 1148,
+      "loss": 0.37922312425643084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37922312425643084
+    },
+    {
+      "train_step": 1149,
+      "loss": 0.3775187367641555,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3775187367641555
+    },
+    {
+      "train_step": 1150,
+      "loss": 0.4936924869598876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4936924869598876
+    },
+    {
+      "train_step": 1151,
+      "loss": 0.5950687919494129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5950687919494129
+    },
+    {
+      "train_step": 1152,
+      "loss": 0.4392459432398495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4392459432398495
+    },
+    {
+      "train_step": 1153,
+      "loss": 0.597071573690506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.597071573690506
+    },
+    {
+      "train_step": 1154,
+      "loss": 0.45924019175856673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45924019175856673
+    },
+    {
+      "train_step": 1155,
+      "loss": 0.6197638916329118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6197638916329118
+    },
+    {
+      "train_step": 1156,
+      "loss": 0.6670445533302636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6670445533302636
+    },
+    {
+      "train_step": 1157,
+      "loss": 0.7527179932356722,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7527179932356722
+    },
+    {
+      "train_step": 1158,
+      "loss": 0.7564505456679586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7564505456679586
+    },
+    {
+      "train_step": 1159,
+      "loss": 0.6349786706213119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6349786706213119
+    },
+    {
+      "train_step": 1160,
+      "loss": 0.7738596515187491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7738596515187491
+    },
+    {
+      "train_step": 1161,
+      "loss": 0.6514374941450026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6514374941450026
+    },
+    {
+      "train_step": 1162,
+      "loss": 0.7494702242087092,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7494702242087092
+    },
+    {
+      "train_step": 1163,
+      "loss": 0.5431570061791513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5431570061791513
+    },
+    {
+      "train_step": 1164,
+      "loss": 0.6444629804582153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6444629804582153
+    },
+    {
+      "train_step": 1165,
+      "loss": 0.7282911601658978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7282911601658978
+    },
+    {
+      "train_step": 1166,
+      "loss": 0.731111116077807,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.731111116077807
+    },
+    {
+      "train_step": 1167,
+      "loss": 0.8306964720890453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8306964720890453
+    },
+    {
+      "train_step": 1168,
+      "loss": 0.636863425901128,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.636863425901128
+    },
+    {
+      "train_step": 1169,
+      "loss": 0.574232724702333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.574232724702333
+    },
+    {
+      "train_step": 1170,
+      "loss": 0.5095855323128169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5095855323128169
+    },
+    {
+      "train_step": 1171,
+      "loss": 0.5223211446170415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5223211446170415
+    },
+    {
+      "train_step": 1172,
+      "loss": 0.4329673132582019,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4329673132582019
+    },
+    {
+      "train_step": 1173,
+      "loss": 0.7840684041169296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7840684041169296
+    },
+    {
+      "train_step": 1174,
+      "loss": 0.6416346178856528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6416346178856528
+    },
+    {
+      "train_step": 1175,
+      "loss": 0.5211269445213437,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5211269445213437
+    },
+    {
+      "train_step": 1176,
+      "loss": 0.42408409171897166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42408409171897166
+    },
+    {
+      "train_step": 1177,
+      "loss": 0.5712627406093967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5712627406093967
+    },
+    {
+      "train_step": 1178,
+      "loss": 0.7442125144769723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7442125144769723
+    },
+    {
+      "train_step": 1179,
+      "loss": 0.5762516777191556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5762516777191556
+    },
+    {
+      "train_step": 1180,
+      "loss": 0.6061759354461834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6061759354461834
+    },
+    {
+      "train_step": 1181,
+      "loss": 0.5587319022040472,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5587319022040472
+    },
+    {
+      "train_step": 1182,
+      "loss": 0.5306679583085688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5306679583085688
+    },
+    {
+      "train_step": 1183,
+      "loss": 0.4218720619302942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4218720619302942
+    },
+    {
+      "train_step": 1184,
+      "loss": 0.7400866148566637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7400866148566637
+    },
+    {
+      "train_step": 1185,
+      "loss": 0.6027696076320007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6027696076320007
+    },
+    {
+      "train_step": 1186,
+      "loss": 0.5966426698999483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5966426698999483
+    },
+    {
+      "train_step": 1187,
+      "loss": 0.7290264218697816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7290264218697816
+    },
+    {
+      "train_step": 1188,
+      "loss": 0.5494976690134906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5494976690134906
+    },
+    {
+      "train_step": 1189,
+      "loss": 0.4477118481775939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4477118481775939
+    },
+    {
+      "train_step": 1190,
+      "loss": 0.6414916662909976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6414916662909976
+    },
+    {
+      "train_step": 1191,
+      "loss": 0.6366542055495434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6366542055495434
+    },
+    {
+      "train_step": 1192,
+      "loss": 0.5720817767687685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5720817767687685
+    },
+    {
+      "train_step": 1193,
+      "loss": 0.5163181627057929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5163181627057929
+    },
+    {
+      "train_step": 1194,
+      "loss": 0.7221684324471954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7221684324471954
+    },
+    {
+      "train_step": 1195,
+      "loss": 0.5425838081320743,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5425838081320743
+    },
+    {
+      "train_step": 1196,
+      "loss": 0.5041261176151655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5041261176151655
+    },
+    {
+      "train_step": 1197,
+      "loss": 0.6407823409291992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6407823409291992
+    },
+    {
+      "train_step": 1198,
+      "loss": 0.6412337390858942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6412337390858942
+    },
+    {
+      "train_step": 1199,
+      "loss": 0.5972750881880071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5972750881880071
+    },
+    {
+      "train_step": 1200,
+      "loss": 0.6440855312186639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6440855312186639
+    },
+    {
+      "train_step": 1201,
+      "loss": 0.575137746601551,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.575137746601551
+    },
+    {
+      "train_step": 1202,
+      "loss": 0.6549602834271694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6549602834271694
+    },
+    {
+      "train_step": 1203,
+      "loss": 0.45370146815009954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45370146815009954
+    },
+    {
+      "train_step": 1204,
+      "loss": 0.524048453001279,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.524048453001279
+    },
+    {
+      "train_step": 1205,
+      "loss": 0.5295425487994676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5295425487994676
+    },
+    {
+      "train_step": 1206,
+      "loss": 0.8813149784951894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8813149784951894
+    },
+    {
+      "train_step": 1207,
+      "loss": 0.4386667452287118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4386667452287118
+    },
+    {
+      "train_step": 1208,
+      "loss": 0.7704693405926413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7704693405926413
+    },
+    {
+      "train_step": 1209,
+      "loss": 0.5785545580124875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5785545580124875
+    },
+    {
+      "train_step": 1210,
+      "loss": 0.4378601385921212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4378601385921212
+    },
+    {
+      "train_step": 1211,
+      "loss": 0.7169802735462163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7169802735462163
+    },
+    {
+      "train_step": 1212,
+      "loss": 0.4772534169405903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4772534169405903
+    },
+    {
+      "train_step": 1213,
+      "loss": 0.5750041749021136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5750041749021136
+    },
+    {
+      "train_step": 1214,
+      "loss": 0.3907604273613561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3907604273613561
+    },
+    {
+      "train_step": 1215,
+      "loss": 0.4720056660933775,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4720056660933775
+    },
+    {
+      "train_step": 1216,
+      "loss": 0.48031038340529913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48031038340529913
+    },
+    {
+      "train_step": 1217,
+      "loss": 0.528093412903667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.528093412903667
+    },
+    {
+      "train_step": 1218,
+      "loss": 0.4874794276470119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4874794276470119
+    },
+    {
+      "train_step": 1219,
+      "loss": 0.46810797286133565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46810797286133565
+    },
+    {
+      "train_step": 1220,
+      "loss": 0.5751415421697039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5751415421697039
+    },
+    {
+      "train_step": 1221,
+      "loss": 0.5889347165760229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5889347165760229
+    },
+    {
+      "train_step": 1222,
+      "loss": 0.5221385689552348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5221385689552348
+    },
+    {
+      "train_step": 1223,
+      "loss": 0.5271246868617905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5271246868617905
+    },
+    {
+      "train_step": 1224,
+      "loss": 0.7891942014518087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7891942014518087
+    },
+    {
+      "train_step": 1225,
+      "loss": 0.66768992661913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66768992661913
+    },
+    {
+      "train_step": 1226,
+      "loss": 0.46843983410990203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46843983410990203
+    },
+    {
+      "train_step": 1227,
+      "loss": 0.612966337433219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.612966337433219
+    },
+    {
+      "train_step": 1228,
+      "loss": 0.9210489605173989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9210489605173989
+    },
+    {
+      "train_step": 1229,
+      "loss": 0.42664684354216986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42664684354216986
+    },
+    {
+      "train_step": 1230,
+      "loss": 0.5735310106200231,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5735310106200231
+    },
+    {
+      "train_step": 1231,
+      "loss": 0.6230032466010266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6230032466010266
+    },
+    {
+      "train_step": 1232,
+      "loss": 0.6065871121279248,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6065871121279248
+    },
+    {
+      "train_step": 1233,
+      "loss": 0.7268956370120991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7268956370120991
+    },
+    {
+      "train_step": 1234,
+      "loss": 0.6273195437137462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6273195437137462
+    },
+    {
+      "train_step": 1235,
+      "loss": 0.4677989431748798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4677989431748798
+    },
+    {
+      "train_step": 1236,
+      "loss": 0.874536359693857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.874536359693857
+    },
+    {
+      "train_step": 1237,
+      "loss": 0.6342080877851882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6342080877851882
+    },
+    {
+      "train_step": 1238,
+      "loss": 0.4849925235366638,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4849925235366638
+    },
+    {
+      "train_step": 1239,
+      "loss": 0.5894198161258654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5894198161258654
+    },
+    {
+      "train_step": 1240,
+      "loss": 0.7487109950135021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7487109950135021
+    },
+    {
+      "train_step": 1241,
+      "loss": 0.5258254431640393,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5258254431640393
+    },
+    {
+      "train_step": 1242,
+      "loss": 0.6616992887423981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6616992887423981
+    },
+    {
+      "train_step": 1243,
+      "loss": 0.7118374281065223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7118374281065223
+    },
+    {
+      "train_step": 1244,
+      "loss": 0.6418691391717858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6418691391717858
+    },
+    {
+      "train_step": 1245,
+      "loss": 0.5106153628216429,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5106153628216429
+    },
+    {
+      "train_step": 1246,
+      "loss": 0.6584348243994029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6584348243994029
+    },
+    {
+      "train_step": 1247,
+      "loss": 0.5117515378018515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5117515378018515
+    },
+    {
+      "train_step": 1248,
+      "loss": 0.3785190541848192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3785190541848192
+    },
+    {
+      "train_step": 1249,
+      "loss": 0.690232385626466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.690232385626466
+    },
+    {
+      "train_step": 1250,
+      "loss": 0.5634094449226051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5634094449226051
+    },
+    {
+      "train_step": 1251,
+      "loss": 0.457424067951666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.457424067951666
+    },
+    {
+      "train_step": 1252,
+      "loss": 0.6464515747843237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6464515747843237
+    },
+    {
+      "train_step": 1253,
+      "loss": 0.5320711045398003,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5320711045398003
+    },
+    {
+      "train_step": 1254,
+      "loss": 0.6206423661296321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6206423661296321
+    },
+    {
+      "train_step": 1255,
+      "loss": 0.8161850948107714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8161850948107714
+    },
+    {
+      "train_step": 1256,
+      "loss": 0.6048896291119175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6048896291119175
+    },
+    {
+      "train_step": 1257,
+      "loss": 0.5368466855113396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5368466855113396
+    },
+    {
+      "train_step": 1258,
+      "loss": 0.4157926034690253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4157926034690253
+    },
+    {
+      "train_step": 1259,
+      "loss": 0.49928322052158547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49928322052158547
+    },
+    {
+      "train_step": 1260,
+      "loss": 0.4239140207844898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4239140207844898
+    },
+    {
+      "train_step": 1261,
+      "loss": 0.4911748002315074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4911748002315074
+    },
+    {
+      "train_step": 1262,
+      "loss": 0.5927175011729916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5927175011729916
+    },
+    {
+      "train_step": 1263,
+      "loss": 0.8124236526028698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8124236526028698
+    },
+    {
+      "train_step": 1264,
+      "loss": 0.5674359687960682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5674359687960682
+    },
+    {
+      "train_step": 1265,
+      "loss": 0.9529686499123383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9529686499123383
+    },
+    {
+      "train_step": 1266,
+      "loss": 0.5390182423001635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5390182423001635
+    },
+    {
+      "train_step": 1267,
+      "loss": 0.6079547515379162,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6079547515379162
+    },
+    {
+      "train_step": 1268,
+      "loss": 0.38477602571284214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38477602571284214
+    },
+    {
+      "train_step": 1269,
+      "loss": 0.6165368117139747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6165368117139747
+    },
+    {
+      "train_step": 1270,
+      "loss": 0.6806703226395671,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6806703226395671
+    },
+    {
+      "train_step": 1271,
+      "loss": 0.6203247130932293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6203247130932293
+    },
+    {
+      "train_step": 1272,
+      "loss": 0.5147138616771396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5147138616771396
+    },
+    {
+      "train_step": 1273,
+      "loss": 0.43666413957710465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43666413957710465
+    },
+    {
+      "train_step": 1274,
+      "loss": 0.6653188522764613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6653188522764613
+    },
+    {
+      "train_step": 1275,
+      "loss": 0.6505606087397804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6505606087397804
+    },
+    {
+      "train_step": 1276,
+      "loss": 0.5192606388779942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5192606388779942
+    },
+    {
+      "train_step": 1277,
+      "loss": 0.3617648596649068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3617648596649068
+    },
+    {
+      "train_step": 1278,
+      "loss": 0.7252683278093761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7252683278093761
+    },
+    {
+      "train_step": 1279,
+      "loss": 0.6359848930204164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6359848930204164
+    },
+    {
+      "train_step": 1280,
+      "loss": 0.700684873013134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.700684873013134
+    },
+    {
+      "train_step": 1281,
+      "loss": 0.41868829116407635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41868829116407635
+    },
+    {
+      "train_step": 1282,
+      "loss": 0.7293700567553361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7293700567553361
+    },
+    {
+      "train_step": 1283,
+      "loss": 0.6123336410690753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6123336410690753
+    },
+    {
+      "train_step": 1284,
+      "loss": 0.5334437366199646,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5334437366199646
+    },
+    {
+      "train_step": 1285,
+      "loss": 0.7736555283862143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7736555283862143
+    },
+    {
+      "train_step": 1286,
+      "loss": 0.5289270110438384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5289270110438384
+    },
+    {
+      "train_step": 1287,
+      "loss": 0.4791640592366161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4791640592366161
+    },
+    {
+      "train_step": 1288,
+      "loss": 0.6350914229966893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6350914229966893
+    },
+    {
+      "train_step": 1289,
+      "loss": 0.5700707796334858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5700707796334858
+    },
+    {
+      "train_step": 1290,
+      "loss": 0.5565130373102637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5565130373102637
+    },
+    {
+      "train_step": 1291,
+      "loss": 0.4137851195575727,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4137851195575727
+    },
+    {
+      "train_step": 1292,
+      "loss": 0.5677856007884262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5677856007884262
+    },
+    {
+      "train_step": 1293,
+      "loss": 0.6745844109493588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6745844109493588
+    },
+    {
+      "train_step": 1294,
+      "loss": 0.4931256013422375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4931256013422375
+    },
+    {
+      "train_step": 1295,
+      "loss": 0.46049902105904056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46049902105904056
+    },
+    {
+      "train_step": 1296,
+      "loss": 0.7987596453296858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7987596453296858
+    },
+    {
+      "train_step": 1297,
+      "loss": 0.5962221309990905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962221309990905
+    },
+    {
+      "train_step": 1298,
+      "loss": 0.6451704364159937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6451704364159937
+    },
+    {
+      "train_step": 1299,
+      "loss": 0.44130578143250404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44130578143250404
+    },
+    {
+      "train_step": 1300,
+      "loss": 0.5834454365468779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5834454365468779
+    },
+    {
+      "train_step": 1301,
+      "loss": 0.4587244999935237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4587244999935237
+    },
+    {
+      "train_step": 1302,
+      "loss": 0.3652387272647074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3652387272647074
+    },
+    {
+      "train_step": 1303,
+      "loss": 0.6923371645416957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6923371645416957
+    },
+    {
+      "train_step": 1304,
+      "loss": 0.6301065514744919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6301065514744919
+    },
+    {
+      "train_step": 1305,
+      "loss": 0.6929784505556653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6929784505556653
+    },
+    {
+      "train_step": 1306,
+      "loss": 0.626766270206822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.626766270206822
+    },
+    {
+      "train_step": 1307,
+      "loss": 0.7147822526174094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7147822526174094
+    },
+    {
+      "train_step": 1308,
+      "loss": 0.6276225948912014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6276225948912014
+    },
+    {
+      "train_step": 1309,
+      "loss": 0.4864564976315216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4864564976315216
+    },
+    {
+      "train_step": 1310,
+      "loss": 0.6807683908941106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6807683908941106
+    },
+    {
+      "train_step": 1311,
+      "loss": 0.32522647545120464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32522647545120464
+    },
+    {
+      "train_step": 1312,
+      "loss": 0.5723391401170296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5723391401170296
+    },
+    {
+      "train_step": 1313,
+      "loss": 0.5700974920173705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5700974920173705
+    },
+    {
+      "train_step": 1314,
+      "loss": 0.6086105058931573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6086105058931573
+    },
+    {
+      "train_step": 1315,
+      "loss": 0.8027197997708837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8027197997708837
+    },
+    {
+      "train_step": 1316,
+      "loss": 0.6747982557142213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6747982557142213
+    },
+    {
+      "train_step": 1317,
+      "loss": 0.516411751477316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.516411751477316
+    },
+    {
+      "train_step": 1318,
+      "loss": 0.5274993091899784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5274993091899784
+    },
+    {
+      "train_step": 1319,
+      "loss": 0.7431551788050809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7431551788050809
+    },
+    {
+      "train_step": 1320,
+      "loss": 0.7626843238002798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7626843238002798
+    },
+    {
+      "train_step": 1321,
+      "loss": 0.6942139339372354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6942139339372354
+    },
+    {
+      "train_step": 1322,
+      "loss": 0.7219720721726118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7219720721726118
+    },
+    {
+      "train_step": 1323,
+      "loss": 0.5945348031733794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5945348031733794
+    },
+    {
+      "train_step": 1324,
+      "loss": 0.5116677796614704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5116677796614704
+    },
+    {
+      "train_step": 1325,
+      "loss": 0.5121183486160195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5121183486160195
+    },
+    {
+      "train_step": 1326,
+      "loss": 0.4725412519297213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4725412519297213
+    },
+    {
+      "train_step": 1327,
+      "loss": 0.6870286652817629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6870286652817629
+    },
+    {
+      "train_step": 1328,
+      "loss": 0.4874146708525958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4874146708525958
+    },
+    {
+      "train_step": 1329,
+      "loss": 0.37003642120931085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37003642120931085
+    },
+    {
+      "train_step": 1330,
+      "loss": 0.5440934286761696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5440934286761696
+    },
+    {
+      "train_step": 1331,
+      "loss": 0.6160792517101821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6160792517101821
+    },
+    {
+      "train_step": 1332,
+      "loss": 0.4039635316181381,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4039635316181381
+    },
+    {
+      "train_step": 1333,
+      "loss": 0.5441053678865787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441053678865787
+    },
+    {
+      "train_step": 1334,
+      "loss": 0.5421488866719719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5421488866719719
+    },
+    {
+      "train_step": 1335,
+      "loss": 0.6007327287170111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6007327287170111
+    },
+    {
+      "train_step": 1336,
+      "loss": 0.7253189000246477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7253189000246477
+    },
+    {
+      "train_step": 1337,
+      "loss": 0.6645522921380692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6645522921380692
+    },
+    {
+      "train_step": 1338,
+      "loss": 0.5792250601340976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5792250601340976
+    },
+    {
+      "train_step": 1339,
+      "loss": 0.5497348786353257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5497348786353257
+    },
+    {
+      "train_step": 1340,
+      "loss": 0.5077586131514613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5077586131514613
+    },
+    {
+      "train_step": 1341,
+      "loss": 0.6083505990906412,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6083505990906412
+    },
+    {
+      "train_step": 1342,
+      "loss": 0.624517858344108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.624517858344108
+    },
+    {
+      "train_step": 1343,
+      "loss": 0.6352354811931554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6352354811931554
+    },
+    {
+      "train_step": 1344,
+      "loss": 0.6522919407524873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6522919407524873
+    },
+    {
+      "train_step": 1345,
+      "loss": 0.6610374390573234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6610374390573234
+    },
+    {
+      "train_step": 1346,
+      "loss": 0.5500398262085738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5500398262085738
+    },
+    {
+      "train_step": 1347,
+      "loss": 0.6048606180190537,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6048606180190537
+    },
+    {
+      "train_step": 1348,
+      "loss": 0.541592873304062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.541592873304062
+    },
+    {
+      "train_step": 1349,
+      "loss": 0.6293370943926774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6293370943926774
+    },
+    {
+      "train_step": 1350,
+      "loss": 0.5366175129007706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5366175129007706
+    },
+    {
+      "train_step": 1351,
+      "loss": 0.5896547028873704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5896547028873704
+    },
+    {
+      "train_step": 1352,
+      "loss": 0.7138981041324854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7138981041324854
+    },
+    {
+      "train_step": 1353,
+      "loss": 0.6008217905226333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6008217905226333
+    },
+    {
+      "train_step": 1354,
+      "loss": 0.3636575536903295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3636575536903295
+    },
+    {
+      "train_step": 1355,
+      "loss": 0.5702697344880419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5702697344880419
+    },
+    {
+      "train_step": 1356,
+      "loss": 0.5917507517720938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5917507517720938
+    },
+    {
+      "train_step": 1357,
+      "loss": 0.5013908711305023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5013908711305023
+    },
+    {
+      "train_step": 1358,
+      "loss": 0.34986064396835803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34986064396835803
+    },
+    {
+      "train_step": 1359,
+      "loss": 0.46423340568913224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46423340568913224
+    },
+    {
+      "train_step": 1360,
+      "loss": 0.6698769014637236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6698769014637236
+    },
+    {
+      "train_step": 1361,
+      "loss": 0.5314617282041989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5314617282041989
+    },
+    {
+      "train_step": 1362,
+      "loss": 0.4156523115682788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4156523115682788
+    },
+    {
+      "train_step": 1363,
+      "loss": 0.5514197147529839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5514197147529839
+    },
+    {
+      "train_step": 1364,
+      "loss": 0.8293001204989734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8293001204989734
+    },
+    {
+      "train_step": 1365,
+      "loss": 0.7990678269545239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7990678269545239
+    },
+    {
+      "train_step": 1366,
+      "loss": 0.7294306479077418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7294306479077418
+    },
+    {
+      "train_step": 1367,
+      "loss": 0.5790012938841445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5790012938841445
+    },
+    {
+      "train_step": 1368,
+      "loss": 0.6391287674902754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6391287674902754
+    },
+    {
+      "train_step": 1369,
+      "loss": 0.5512098769310021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5512098769310021
+    },
+    {
+      "train_step": 1370,
+      "loss": 0.5840092896327734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5840092896327734
+    },
+    {
+      "train_step": 1371,
+      "loss": 0.3763670041191207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3763670041191207
+    },
+    {
+      "train_step": 1372,
+      "loss": 0.8692826372445777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8692826372445777
+    },
+    {
+      "train_step": 1373,
+      "loss": 0.5630543689797818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5630543689797818
+    },
+    {
+      "train_step": 1374,
+      "loss": 0.7023471300943657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7023471300943657
+    },
+    {
+      "train_step": 1375,
+      "loss": 0.5782405999839402,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5782405999839402
+    },
+    {
+      "train_step": 1376,
+      "loss": 0.5586647436445039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5586647436445039
+    },
+    {
+      "train_step": 1377,
+      "loss": 0.47977036866765455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47977036866765455
+    },
+    {
+      "train_step": 1378,
+      "loss": 0.6317991657661354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6317991657661354
+    },
+    {
+      "train_step": 1379,
+      "loss": 0.6003380828102483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6003380828102483
+    },
+    {
+      "train_step": 1380,
+      "loss": 0.6726201279125452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6726201279125452
+    },
+    {
+      "train_step": 1381,
+      "loss": 0.5997153397005156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5997153397005156
+    },
+    {
+      "train_step": 1382,
+      "loss": 0.41671010654977614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41671010654977614
+    },
+    {
+      "train_step": 1383,
+      "loss": 0.516502186255144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.516502186255144
+    },
+    {
+      "train_step": 1384,
+      "loss": 0.6585767880776978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6585767880776978
+    },
+    {
+      "train_step": 1385,
+      "loss": 0.6262271728189619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6262271728189619
+    },
+    {
+      "train_step": 1386,
+      "loss": 0.46784152222387365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46784152222387365
+    },
+    {
+      "train_step": 1387,
+      "loss": 0.47898724455499364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47898724455499364
+    },
+    {
+      "train_step": 1388,
+      "loss": 0.5666858678963668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5666858678963668
+    },
+    {
+      "train_step": 1389,
+      "loss": 0.42023308444017227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42023308444017227
+    },
+    {
+      "train_step": 1390,
+      "loss": 0.7610127981295779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7610127981295779
+    },
+    {
+      "train_step": 1391,
+      "loss": 0.6646947964387669,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6646947964387669
+    },
+    {
+      "train_step": 1392,
+      "loss": 0.598174778472682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.598174778472682
+    },
+    {
+      "train_step": 1393,
+      "loss": 0.4849422781817217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4849422781817217
+    },
+    {
+      "train_step": 1394,
+      "loss": 0.5894098108833143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5894098108833143
+    },
+    {
+      "train_step": 1395,
+      "loss": 0.49203713415694894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49203713415694894
+    },
+    {
+      "train_step": 1396,
+      "loss": 0.7297456017299,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7297456017299
+    },
+    {
+      "train_step": 1397,
+      "loss": 0.6983079281781883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6983079281781883
+    },
+    {
+      "train_step": 1398,
+      "loss": 0.6050394111912942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6050394111912942
+    },
+    {
+      "train_step": 1399,
+      "loss": 0.38031297178642004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38031297178642004
+    },
+    {
+      "train_step": 1400,
+      "loss": 0.631686640123856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.631686640123856
+    },
+    {
+      "train_step": 1401,
+      "loss": 0.7170005678544852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7170005678544852
+    },
+    {
+      "train_step": 1402,
+      "loss": 0.7165717673486438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7165717673486438
+    },
+    {
+      "train_step": 1403,
+      "loss": 0.6221769314201508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6221769314201508
+    },
+    {
+      "train_step": 1404,
+      "loss": 0.7057715197311313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7057715197311313
+    },
+    {
+      "train_step": 1405,
+      "loss": 0.45343903613872666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45343903613872666
+    },
+    {
+      "train_step": 1406,
+      "loss": 0.7548228647488653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7548228647488653
+    },
+    {
+      "train_step": 1407,
+      "loss": 0.6388978475431203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6388978475431203
+    },
+    {
+      "train_step": 1408,
+      "loss": 0.8359190323029214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8359190323029214
+    },
+    {
+      "train_step": 1409,
+      "loss": 0.567319525595973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.567319525595973
+    },
+    {
+      "train_step": 1410,
+      "loss": 0.7820968730615526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7820968730615526
+    },
+    {
+      "train_step": 1411,
+      "loss": 0.5327127597084376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5327127597084376
+    },
+    {
+      "train_step": 1412,
+      "loss": 0.5458181607445515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5458181607445515
+    },
+    {
+      "train_step": 1413,
+      "loss": 0.5917984223531059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5917984223531059
+    },
+    {
+      "train_step": 1414,
+      "loss": 0.5598303231698881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5598303231698881
+    },
+    {
+      "train_step": 1415,
+      "loss": 0.5632974611520367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5632974611520367
+    },
+    {
+      "train_step": 1416,
+      "loss": 0.5923497660175676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5923497660175676
+    },
+    {
+      "train_step": 1417,
+      "loss": 0.5601812263658987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5601812263658987
+    },
+    {
+      "train_step": 1418,
+      "loss": 0.483917341075623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.483917341075623
+    },
+    {
+      "train_step": 1419,
+      "loss": 0.7395224548289654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7395224548289654
+    },
+    {
+      "train_step": 1420,
+      "loss": 0.6864066522726522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6864066522726522
+    },
+    {
+      "train_step": 1421,
+      "loss": 0.5090321293580284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5090321293580284
+    },
+    {
+      "train_step": 1422,
+      "loss": 0.7423491578108761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7423491578108761
+    },
+    {
+      "train_step": 1423,
+      "loss": 0.7294608681753092,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7294608681753092
+    },
+    {
+      "train_step": 1424,
+      "loss": 0.6507534983550392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6507534983550392
+    },
+    {
+      "train_step": 1425,
+      "loss": 0.6137444751000748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6137444751000748
+    },
+    {
+      "train_step": 1426,
+      "loss": 0.48445191048849445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48445191048849445
+    },
+    {
+      "train_step": 1427,
+      "loss": 0.5499619124939529,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5499619124939529
+    },
+    {
+      "train_step": 1428,
+      "loss": 0.71066825417522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.71066825417522
+    },
+    {
+      "train_step": 1429,
+      "loss": 0.6119646485083762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6119646485083762
+    },
+    {
+      "train_step": 1430,
+      "loss": 0.6383821181747202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6383821181747202
+    },
+    {
+      "train_step": 1431,
+      "loss": 0.5326761649054012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5326761649054012
+    },
+    {
+      "train_step": 1432,
+      "loss": 0.5694513541363658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5694513541363658
+    },
+    {
+      "train_step": 1433,
+      "loss": 0.5480895428461102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5480895428461102
+    },
+    {
+      "train_step": 1434,
+      "loss": 0.6772481258933563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6772481258933563
+    },
+    {
+      "train_step": 1435,
+      "loss": 0.672666109662487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.672666109662487
+    },
+    {
+      "train_step": 1436,
+      "loss": 0.7120418926212659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7120418926212659
+    },
+    {
+      "train_step": 1437,
+      "loss": 0.6207702388607593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6207702388607593
+    },
+    {
+      "train_step": 1438,
+      "loss": 0.5206256345783719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5206256345783719
+    },
+    {
+      "train_step": 1439,
+      "loss": 0.5886537116450258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5886537116450258
+    },
+    {
+      "train_step": 1440,
+      "loss": 0.4350127609530849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4350127609530849
+    },
+    {
+      "train_step": 1441,
+      "loss": 0.3890497691254186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3890497691254186
+    },
+    {
+      "train_step": 1442,
+      "loss": 0.4907319009460171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4907319009460171
+    },
+    {
+      "train_step": 1443,
+      "loss": 0.512106661337842,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.512106661337842
+    },
+    {
+      "train_step": 1444,
+      "loss": 0.48538007951593115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48538007951593115
+    },
+    {
+      "train_step": 1445,
+      "loss": 0.5322842194169319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5322842194169319
+    },
+    {
+      "train_step": 1446,
+      "loss": 0.6148430157289941,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6148430157289941
+    },
+    {
+      "train_step": 1447,
+      "loss": 0.4282453700655837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4282453700655837
+    },
+    {
+      "train_step": 1448,
+      "loss": 0.6127912481965585,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6127912481965585
+    },
+    {
+      "train_step": 1449,
+      "loss": 0.4265163975898574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4265163975898574
+    },
+    {
+      "train_step": 1450,
+      "loss": 0.4682338128678227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4682338128678227
+    },
+    {
+      "train_step": 1451,
+      "loss": 0.7582295642894945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7582295642894945
+    },
+    {
+      "train_step": 1452,
+      "loss": 0.7401498072654318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7401498072654318
+    },
+    {
+      "train_step": 1453,
+      "loss": 0.5669506500186721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5669506500186721
+    },
+    {
+      "train_step": 1454,
+      "loss": 0.5982511866310594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5982511866310594
+    },
+    {
+      "train_step": 1455,
+      "loss": 0.587266115865163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.587266115865163
+    },
+    {
+      "train_step": 1456,
+      "loss": 0.4476916358966844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4476916358966844
+    },
+    {
+      "train_step": 1457,
+      "loss": 0.8271688389638872,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8271688389638872
+    },
+    {
+      "train_step": 1458,
+      "loss": 0.3853896299794627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3853896299794627
+    },
+    {
+      "train_step": 1459,
+      "loss": 0.6551462150843357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6551462150843357
+    },
+    {
+      "train_step": 1460,
+      "loss": 0.492474031563241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.492474031563241
+    },
+    {
+      "train_step": 1461,
+      "loss": 0.4927965995393167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4927965995393167
+    },
+    {
+      "train_step": 1462,
+      "loss": 0.45623801118450413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45623801118450413
+    },
+    {
+      "train_step": 1463,
+      "loss": 0.775095034758298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.775095034758298
+    },
+    {
+      "train_step": 1464,
+      "loss": 0.5784804118783526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5784804118783526
+    },
+    {
+      "train_step": 1465,
+      "loss": 0.6947955892181392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6947955892181392
+    },
+    {
+      "train_step": 1466,
+      "loss": 0.3554868009841513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3554868009841513
+    },
+    {
+      "train_step": 1467,
+      "loss": 0.6542537210614683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6542537210614683
+    },
+    {
+      "train_step": 1468,
+      "loss": 0.6574022646559192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6574022646559192
+    },
+    {
+      "train_step": 1469,
+      "loss": 0.7022125298769074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7022125298769074
+    },
+    {
+      "train_step": 1470,
+      "loss": 0.5471970295896469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5471970295896469
+    },
+    {
+      "train_step": 1471,
+      "loss": 0.8255299113579849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8255299113579849
+    },
+    {
+      "train_step": 1472,
+      "loss": 0.8166026898456873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8166026898456873
+    },
+    {
+      "train_step": 1473,
+      "loss": 0.5912635184002535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5912635184002535
+    },
+    {
+      "train_step": 1474,
+      "loss": 0.6687728501077469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6687728501077469
+    },
+    {
+      "train_step": 1475,
+      "loss": 0.4796149402476647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4796149402476647
+    },
+    {
+      "train_step": 1476,
+      "loss": 0.39895589374228474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39895589374228474
+    },
+    {
+      "train_step": 1477,
+      "loss": 0.5497546215075669,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5497546215075669
+    },
+    {
+      "train_step": 1478,
+      "loss": 0.6019551572883005,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6019551572883005
+    },
+    {
+      "train_step": 1479,
+      "loss": 0.44012963978468617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44012963978468617
+    },
+    {
+      "train_step": 1480,
+      "loss": 0.582904822735366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.582904822735366
+    },
+    {
+      "train_step": 1481,
+      "loss": 0.6355039736079698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6355039736079698
+    },
+    {
+      "train_step": 1482,
+      "loss": 0.4329616170160527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4329616170160527
+    },
+    {
+      "train_step": 1483,
+      "loss": 0.4278655836233508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4278655836233508
+    },
+    {
+      "train_step": 1484,
+      "loss": 0.5415292225089404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5415292225089404
+    },
+    {
+      "train_step": 1485,
+      "loss": 0.6597540276113046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6597540276113046
+    },
+    {
+      "train_step": 1486,
+      "loss": 0.5717513547973853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5717513547973853
+    },
+    {
+      "train_step": 1487,
+      "loss": 0.38206199240423744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38206199240423744
+    },
+    {
+      "train_step": 1488,
+      "loss": 0.5936592205226885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5936592205226885
+    },
+    {
+      "train_step": 1489,
+      "loss": 0.640909348765359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.640909348765359
+    },
+    {
+      "train_step": 1490,
+      "loss": 0.6027869390004821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6027869390004821
+    },
+    {
+      "train_step": 1491,
+      "loss": 0.5239947132845526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5239947132845526
+    },
+    {
+      "train_step": 1492,
+      "loss": 0.37974246671078776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37974246671078776
+    },
+    {
+      "train_step": 1493,
+      "loss": 0.5462729708084109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5462729708084109
+    },
+    {
+      "train_step": 1494,
+      "loss": 0.650870577742522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.650870577742522
+    },
+    {
+      "train_step": 1495,
+      "loss": 0.447756015287123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.447756015287123
+    },
+    {
+      "train_step": 1496,
+      "loss": 0.3997154492768755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3997154492768755
+    },
+    {
+      "train_step": 1497,
+      "loss": 0.631789487663858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.631789487663858
+    },
+    {
+      "train_step": 1498,
+      "loss": 0.40039626571506165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40039626571506165
+    },
+    {
+      "train_step": 1499,
+      "loss": 0.7078896834086756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7078896834086756
+    },
+    {
+      "train_step": 1500,
+      "loss": 0.47725067368641866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47725067368641866
+    },
+    {
+      "train_step": 1501,
+      "loss": 0.4833380444862207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4833380444862207
+    },
+    {
+      "train_step": 1502,
+      "loss": 0.5378258110388887,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5378258110388887
+    },
+    {
+      "train_step": 1503,
+      "loss": 0.5003453556653075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5003453556653075
+    },
+    {
+      "train_step": 1504,
+      "loss": 0.34355935085524614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34355935085524614
+    },
+    {
+      "train_step": 1505,
+      "loss": 0.5525735487404946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5525735487404946
+    },
+    {
+      "train_step": 1506,
+      "loss": 0.5491270688034215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5491270688034215
+    },
+    {
+      "train_step": 1507,
+      "loss": 0.49725789451399466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49725789451399466
+    },
+    {
+      "train_step": 1508,
+      "loss": 0.6487009988278728,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6487009988278728
+    },
+    {
+      "train_step": 1509,
+      "loss": 0.533792090618311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.533792090618311
+    },
+    {
+      "train_step": 1510,
+      "loss": 0.5301873309594163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5301873309594163
+    },
+    {
+      "train_step": 1511,
+      "loss": 0.4462664873355222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4462664873355222
+    },
+    {
+      "train_step": 1512,
+      "loss": 0.33309917981088455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33309917981088455
+    },
+    {
+      "train_step": 1513,
+      "loss": 0.4752826358808696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4752826358808696
+    },
+    {
+      "train_step": 1514,
+      "loss": 0.49853596884179846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49853596884179846
+    },
+    {
+      "train_step": 1515,
+      "loss": 0.6816611576073723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6816611576073723
+    },
+    {
+      "train_step": 1516,
+      "loss": 0.4644832381998326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4644832381998326
+    },
+    {
+      "train_step": 1517,
+      "loss": 0.7038973953425803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7038973953425803
+    },
+    {
+      "train_step": 1518,
+      "loss": 0.6749183646258313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6749183646258313
+    },
+    {
+      "train_step": 1519,
+      "loss": 0.44947025774666793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44947025774666793
+    },
+    {
+      "train_step": 1520,
+      "loss": 0.5425393648016414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5425393648016414
+    },
+    {
+      "train_step": 1521,
+      "loss": 0.578020445229827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.578020445229827
+    },
+    {
+      "train_step": 1522,
+      "loss": 0.7439560547795939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7439560547795939
+    },
+    {
+      "train_step": 1523,
+      "loss": 0.7554668661519732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7554668661519732
+    },
+    {
+      "train_step": 1524,
+      "loss": 0.5504510182493253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5504510182493253
+    },
+    {
+      "train_step": 1525,
+      "loss": 0.6822833397693158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6822833397693158
+    },
+    {
+      "train_step": 1526,
+      "loss": 0.72233321989443,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.72233321989443
+    },
+    {
+      "train_step": 1527,
+      "loss": 0.5301455037173687,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5301455037173687
+    },
+    {
+      "train_step": 1528,
+      "loss": 0.5358998156415165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5358998156415165
+    },
+    {
+      "train_step": 1529,
+      "loss": 0.7367511000679858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7367511000679858
+    },
+    {
+      "train_step": 1530,
+      "loss": 0.49628018862214396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49628018862214396
+    },
+    {
+      "train_step": 1531,
+      "loss": 0.6684307224163845,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6684307224163845
+    },
+    {
+      "train_step": 1532,
+      "loss": 0.5990772042307209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5990772042307209
+    },
+    {
+      "train_step": 1533,
+      "loss": 0.6172291693134528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6172291693134528
+    },
+    {
+      "train_step": 1534,
+      "loss": 0.5850844597368883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5850844597368883
+    },
+    {
+      "train_step": 1535,
+      "loss": 0.4974419043932131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4974419043932131
+    },
+    {
+      "train_step": 1536,
+      "loss": 0.5495864659959813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5495864659959813
+    },
+    {
+      "train_step": 1537,
+      "loss": 0.4728608884205465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4728608884205465
+    },
+    {
+      "train_step": 1538,
+      "loss": 0.5643925092385159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5643925092385159
+    },
+    {
+      "train_step": 1539,
+      "loss": 0.6606901201888628,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6606901201888628
+    },
+    {
+      "train_step": 1540,
+      "loss": 0.7810226052890478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7810226052890478
+    },
+    {
+      "train_step": 1541,
+      "loss": 0.5521511895462542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5521511895462542
+    },
+    {
+      "train_step": 1542,
+      "loss": 0.5403155731006957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5403155731006957
+    },
+    {
+      "train_step": 1543,
+      "loss": 0.47977602907555267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47977602907555267
+    },
+    {
+      "train_step": 1544,
+      "loss": 0.4300629506480056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4300629506480056
+    },
+    {
+      "train_step": 1545,
+      "loss": 0.5103719305028317,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5103719305028317
+    },
+    {
+      "train_step": 1546,
+      "loss": 0.6450671552671168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6450671552671168
+    },
+    {
+      "train_step": 1547,
+      "loss": 0.4092771503825934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4092771503825934
+    },
+    {
+      "train_step": 1548,
+      "loss": 0.5538621289324341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5538621289324341
+    },
+    {
+      "train_step": 1549,
+      "loss": 0.6007583067050182,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6007583067050182
+    },
+    {
+      "train_step": 1550,
+      "loss": 0.49023652762192427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49023652762192427
+    },
+    {
+      "train_step": 1551,
+      "loss": 0.4065729937979731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4065729937979731
+    },
+    {
+      "train_step": 1552,
+      "loss": 0.5926225130770342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5926225130770342
+    },
+    {
+      "train_step": 1553,
+      "loss": 0.4137700424820072,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4137700424820072
+    },
+    {
+      "train_step": 1554,
+      "loss": 0.5923994782408871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5923994782408871
+    },
+    {
+      "train_step": 1555,
+      "loss": 0.6519220132896748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6519220132896748
+    },
+    {
+      "train_step": 1556,
+      "loss": 0.6457806416191626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6457806416191626
+    },
+    {
+      "train_step": 1557,
+      "loss": 0.5015449072472729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5015449072472729
+    },
+    {
+      "train_step": 1558,
+      "loss": 0.3346214595462348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3346214595462348
+    },
+    {
+      "train_step": 1559,
+      "loss": 0.609834844911358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.609834844911358
+    },
+    {
+      "train_step": 1560,
+      "loss": 0.5544240660210836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5544240660210836
+    },
+    {
+      "train_step": 1561,
+      "loss": 0.4922900484296643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4922900484296643
+    },
+    {
+      "train_step": 1562,
+      "loss": 0.6709490088775406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6709490088775406
+    },
+    {
+      "train_step": 1563,
+      "loss": 0.5401942845848453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5401942845848453
+    },
+    {
+      "train_step": 1564,
+      "loss": 0.5167356935812882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5167356935812882
+    },
+    {
+      "train_step": 1565,
+      "loss": 0.5691520227251023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5691520227251023
+    },
+    {
+      "train_step": 1566,
+      "loss": 0.5412193643908796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5412193643908796
+    },
+    {
+      "train_step": 1567,
+      "loss": 0.8240202274606386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8240202274606386
+    },
+    {
+      "train_step": 1568,
+      "loss": 0.5623643156949237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5623643156949237
+    },
+    {
+      "train_step": 1569,
+      "loss": 0.4416264004905476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4416264004905476
+    },
+    {
+      "train_step": 1570,
+      "loss": 0.8698855806797536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8698855806797536
+    },
+    {
+      "train_step": 1571,
+      "loss": 0.5908870501647183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5908870501647183
+    },
+    {
+      "train_step": 1572,
+      "loss": 0.5260609826940268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5260609826940268
+    },
+    {
+      "train_step": 1573,
+      "loss": 0.7132110681552736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7132110681552736
+    },
+    {
+      "train_step": 1574,
+      "loss": 0.4214953168863704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4214953168863704
+    },
+    {
+      "train_step": 1575,
+      "loss": 0.42264680832417695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42264680832417695
+    },
+    {
+      "train_step": 1576,
+      "loss": 0.4551714865877252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4551714865877252
+    },
+    {
+      "train_step": 1577,
+      "loss": 0.5131258778024145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5131258778024145
+    },
+    {
+      "train_step": 1578,
+      "loss": 0.6565775348057483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6565775348057483
+    },
+    {
+      "train_step": 1579,
+      "loss": 0.5881820095092969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5881820095092969
+    },
+    {
+      "train_step": 1580,
+      "loss": 0.6017947768405357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6017947768405357
+    },
+    {
+      "train_step": 1581,
+      "loss": 0.6654908307777405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6654908307777405
+    },
+    {
+      "train_step": 1582,
+      "loss": 0.5365129821784785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5365129821784785
+    },
+    {
+      "train_step": 1583,
+      "loss": 0.5298133342466502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5298133342466502
+    },
+    {
+      "train_step": 1584,
+      "loss": 0.5316731496996948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5316731496996948
+    },
+    {
+      "train_step": 1585,
+      "loss": 0.5162609199342285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5162609199342285
+    },
+    {
+      "train_step": 1586,
+      "loss": 0.5742279318242286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5742279318242286
+    },
+    {
+      "train_step": 1587,
+      "loss": 0.6260553492343217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6260553492343217
+    },
+    {
+      "train_step": 1588,
+      "loss": 0.49368388286061915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49368388286061915
+    },
+    {
+      "train_step": 1589,
+      "loss": 0.5172742097576786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5172742097576786
+    },
+    {
+      "train_step": 1590,
+      "loss": 0.6269779739188404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6269779739188404
+    },
+    {
+      "train_step": 1591,
+      "loss": 0.6339079324237678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6339079324237678
+    },
+    {
+      "train_step": 1592,
+      "loss": 0.45492009657725496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45492009657725496
+    },
+    {
+      "train_step": 1593,
+      "loss": 0.652071722102619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.652071722102619
+    },
+    {
+      "train_step": 1594,
+      "loss": 0.4273244159514127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4273244159514127
+    },
+    {
+      "train_step": 1595,
+      "loss": 0.5454287418453265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5454287418453265
+    },
+    {
+      "train_step": 1596,
+      "loss": 0.37343339277142945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37343339277142945
+    },
+    {
+      "train_step": 1597,
+      "loss": 0.5912586570375902,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5912586570375902
+    },
+    {
+      "train_step": 1598,
+      "loss": 0.7324080028743032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7324080028743032
+    },
+    {
+      "train_step": 1599,
+      "loss": 0.4756740518836038,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4756740518836038
+    },
+    {
+      "train_step": 1600,
+      "loss": 0.5272181627169126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5272181627169126
+    },
+    {
+      "train_step": 1601,
+      "loss": 0.6990100491840113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6990100491840113
+    },
+    {
+      "train_step": 1602,
+      "loss": 0.37596322032546986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37596322032546986
+    },
+    {
+      "train_step": 1603,
+      "loss": 0.6754772319813316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6754772319813316
+    },
+    {
+      "train_step": 1604,
+      "loss": 0.4510951584645242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4510951584645242
+    },
+    {
+      "train_step": 1605,
+      "loss": 0.66662687195543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66662687195543
+    },
+    {
+      "train_step": 1606,
+      "loss": 0.39408827941571456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39408827941571456
+    },
+    {
+      "train_step": 1607,
+      "loss": 0.6880033095289387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6880033095289387
+    },
+    {
+      "train_step": 1608,
+      "loss": 0.43471476313051644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43471476313051644
+    },
+    {
+      "train_step": 1609,
+      "loss": 0.4421909861943702,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4421909861943702
+    },
+    {
+      "train_step": 1610,
+      "loss": 0.7150914917027944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7150914917027944
+    },
+    {
+      "train_step": 1611,
+      "loss": 0.5249621853947087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5249621853947087
+    },
+    {
+      "train_step": 1612,
+      "loss": 0.7554803937754111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7554803937754111
+    },
+    {
+      "train_step": 1613,
+      "loss": 0.3647417622671969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3647417622671969
+    },
+    {
+      "train_step": 1614,
+      "loss": 0.5583184743599724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5583184743599724
+    },
+    {
+      "train_step": 1615,
+      "loss": 0.5470691085210317,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5470691085210317
+    },
+    {
+      "train_step": 1616,
+      "loss": 0.4563131233913786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4563131233913786
+    },
+    {
+      "train_step": 1617,
+      "loss": 0.724191443034683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.724191443034683
+    },
+    {
+      "train_step": 1618,
+      "loss": 0.7732022189835988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7732022189835988
+    },
+    {
+      "train_step": 1619,
+      "loss": 0.6492403586983662,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6492403586983662
+    },
+    {
+      "train_step": 1620,
+      "loss": 0.6692896480012052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6692896480012052
+    },
+    {
+      "train_step": 1621,
+      "loss": 0.7139535113843153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7139535113843153
+    },
+    {
+      "train_step": 1622,
+      "loss": 0.6298560432479604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6298560432479604
+    },
+    {
+      "train_step": 1623,
+      "loss": 0.6961450836251026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6961450836251026
+    },
+    {
+      "train_step": 1624,
+      "loss": 0.4558455300835458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4558455300835458
+    },
+    {
+      "train_step": 1625,
+      "loss": 0.4160231782513491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4160231782513491
+    },
+    {
+      "train_step": 1626,
+      "loss": 0.6933653987369347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6933653987369347
+    },
+    {
+      "train_step": 1627,
+      "loss": 0.3541730874856718,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3541730874856718
+    },
+    {
+      "train_step": 1628,
+      "loss": 0.6718867318874452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6718867318874452
+    },
+    {
+      "train_step": 1629,
+      "loss": 0.6193418742977923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6193418742977923
+    },
+    {
+      "train_step": 1630,
+      "loss": 0.6962795252418179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6962795252418179
+    },
+    {
+      "train_step": 1631,
+      "loss": 0.7214731468749589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7214731468749589
+    },
+    {
+      "train_step": 1632,
+      "loss": 0.7048363697364695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7048363697364695
+    },
+    {
+      "train_step": 1633,
+      "loss": 0.731867536213608,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.731867536213608
+    },
+    {
+      "train_step": 1634,
+      "loss": 0.42544118008095755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42544118008095755
+    },
+    {
+      "train_step": 1635,
+      "loss": 0.4554820990749617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4554820990749617
+    },
+    {
+      "train_step": 1636,
+      "loss": 0.7490502039632383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7490502039632383
+    },
+    {
+      "train_step": 1637,
+      "loss": 0.5379308538024498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5379308538024498
+    },
+    {
+      "train_step": 1638,
+      "loss": 0.4985476249224753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4985476249224753
+    },
+    {
+      "train_step": 1639,
+      "loss": 0.44321119796793257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44321119796793257
+    },
+    {
+      "train_step": 1640,
+      "loss": 0.5845045050451435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5845045050451435
+    },
+    {
+      "train_step": 1641,
+      "loss": 0.6187065994779871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6187065994779871
+    },
+    {
+      "train_step": 1642,
+      "loss": 0.7208108067197407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7208108067197407
+    },
+    {
+      "train_step": 1643,
+      "loss": 0.6540730990633933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6540730990633933
+    },
+    {
+      "train_step": 1644,
+      "loss": 0.6533834187218115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6533834187218115
+    },
+    {
+      "train_step": 1645,
+      "loss": 0.5707373259963245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5707373259963245
+    },
+    {
+      "train_step": 1646,
+      "loss": 0.6273804553557369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6273804553557369
+    },
+    {
+      "train_step": 1647,
+      "loss": 0.6551864953687866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6551864953687866
+    },
+    {
+      "train_step": 1648,
+      "loss": 0.6226133234558873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6226133234558873
+    },
+    {
+      "train_step": 1649,
+      "loss": 0.7053861308866403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7053861308866403
+    },
+    {
+      "train_step": 1650,
+      "loss": 0.38415865633593366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38415865633593366
+    },
+    {
+      "train_step": 1651,
+      "loss": 0.6000957106977285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6000957106977285
+    },
+    {
+      "train_step": 1652,
+      "loss": 0.5612422935356315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5612422935356315
+    },
+    {
+      "train_step": 1653,
+      "loss": 0.5110226259745763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5110226259745763
+    },
+    {
+      "train_step": 1654,
+      "loss": 0.5251093943390556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5251093943390556
+    },
+    {
+      "train_step": 1655,
+      "loss": 0.5086527517431132,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5086527517431132
+    },
+    {
+      "train_step": 1656,
+      "loss": 0.6523039009582386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6523039009582386
+    },
+    {
+      "train_step": 1657,
+      "loss": 0.6134405307364733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6134405307364733
+    },
+    {
+      "train_step": 1658,
+      "loss": 0.7081591511397868,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7081591511397868
+    },
+    {
+      "train_step": 1659,
+      "loss": 0.6359667307297496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6359667307297496
+    },
+    {
+      "train_step": 1660,
+      "loss": 0.3756156546687996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3756156546687996
+    },
+    {
+      "train_step": 1661,
+      "loss": 0.3432021444402112,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3432021444402112
+    },
+    {
+      "train_step": 1662,
+      "loss": 0.7046251513655981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7046251513655981
+    },
+    {
+      "train_step": 1663,
+      "loss": 0.7497246677628702,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7497246677628702
+    },
+    {
+      "train_step": 1664,
+      "loss": 0.7788996931044991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7788996931044991
+    },
+    {
+      "train_step": 1665,
+      "loss": 0.7332374317423915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7332374317423915
+    },
+    {
+      "train_step": 1666,
+      "loss": 0.5777324449707204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5777324449707204
+    },
+    {
+      "train_step": 1667,
+      "loss": 0.7360955961786753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7360955961786753
+    },
+    {
+      "train_step": 1668,
+      "loss": 0.48930146473823244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48930146473823244
+    },
+    {
+      "train_step": 1669,
+      "loss": 0.6130005441512134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6130005441512134
+    },
+    {
+      "train_step": 1670,
+      "loss": 0.8977951874851989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8977951874851989
+    },
+    {
+      "train_step": 1671,
+      "loss": 0.6427896534903568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6427896534903568
+    },
+    {
+      "train_step": 1672,
+      "loss": 0.6068459468525135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6068459468525135
+    },
+    {
+      "train_step": 1673,
+      "loss": 0.5556449919664204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5556449919664204
+    },
+    {
+      "train_step": 1674,
+      "loss": 0.5387863626039251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5387863626039251
+    },
+    {
+      "train_step": 1675,
+      "loss": 0.6253573129121934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6253573129121934
+    },
+    {
+      "train_step": 1676,
+      "loss": 0.4847077038154991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4847077038154991
+    },
+    {
+      "train_step": 1677,
+      "loss": 0.5234033729906121,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5234033729906121
+    },
+    {
+      "train_step": 1678,
+      "loss": 0.4928122669454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4928122669454
+    },
+    {
+      "train_step": 1679,
+      "loss": 0.45047961530922614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45047961530922614
+    },
+    {
+      "train_step": 1680,
+      "loss": 0.5530384146733862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5530384146733862
+    },
+    {
+      "train_step": 1681,
+      "loss": 0.6556927104252206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6556927104252206
+    },
+    {
+      "train_step": 1682,
+      "loss": 0.7629606927725008,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7629606927725008
+    },
+    {
+      "train_step": 1683,
+      "loss": 0.6330616773188792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6330616773188792
+    },
+    {
+      "train_step": 1684,
+      "loss": 0.5284655123065397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5284655123065397
+    },
+    {
+      "train_step": 1685,
+      "loss": 0.7473945912178356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7473945912178356
+    },
+    {
+      "train_step": 1686,
+      "loss": 0.6127124584612756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6127124584612756
+    },
+    {
+      "train_step": 1687,
+      "loss": 0.6256789790506978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6256789790506978
+    },
+    {
+      "train_step": 1688,
+      "loss": 0.5187302239376559,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5187302239376559
+    },
+    {
+      "train_step": 1689,
+      "loss": 0.45923275783774364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45923275783774364
+    },
+    {
+      "train_step": 1690,
+      "loss": 0.48637451295474576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48637451295474576
+    },
+    {
+      "train_step": 1691,
+      "loss": 0.7325570280907931,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7325570280907931
+    },
+    {
+      "train_step": 1692,
+      "loss": 0.6159599710340228,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6159599710340228
+    },
+    {
+      "train_step": 1693,
+      "loss": 0.5154131890605829,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5154131890605829
+    },
+    {
+      "train_step": 1694,
+      "loss": 0.599118266161408,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.599118266161408
+    },
+    {
+      "train_step": 1695,
+      "loss": 0.7266791461779136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7266791461779136
+    },
+    {
+      "train_step": 1696,
+      "loss": 0.6550229808215352,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6550229808215352
+    },
+    {
+      "train_step": 1697,
+      "loss": 0.5948144731512223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5948144731512223
+    },
+    {
+      "train_step": 1698,
+      "loss": 0.763182249477097,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.763182249477097
+    },
+    {
+      "train_step": 1699,
+      "loss": 0.4900777926916362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4900777926916362
+    },
+    {
+      "train_step": 1700,
+      "loss": 0.640637850911245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.640637850911245
+    },
+    {
+      "train_step": 1701,
+      "loss": 0.7401408374294841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7401408374294841
+    },
+    {
+      "train_step": 1702,
+      "loss": 0.5259678036653344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5259678036653344
+    },
+    {
+      "train_step": 1703,
+      "loss": 0.619587502185267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.619587502185267
+    },
+    {
+      "train_step": 1704,
+      "loss": 0.8083101548246475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8083101548246475
+    },
+    {
+      "train_step": 1705,
+      "loss": 0.510031047790109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.510031047790109
+    },
+    {
+      "train_step": 1706,
+      "loss": 0.3647077061144052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3647077061144052
+    },
+    {
+      "train_step": 1707,
+      "loss": 0.43622611803857936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43622611803857936
+    },
+    {
+      "train_step": 1708,
+      "loss": 0.5660576775832459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5660576775832459
+    },
+    {
+      "train_step": 1709,
+      "loss": 0.5345407528261814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5345407528261814
+    },
+    {
+      "train_step": 1710,
+      "loss": 0.3797651801166305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3797651801166305
+    },
+    {
+      "train_step": 1711,
+      "loss": 0.6116824756736561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6116824756736561
+    },
+    {
+      "train_step": 1712,
+      "loss": 0.7166977885051102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7166977885051102
+    },
+    {
+      "train_step": 1713,
+      "loss": 0.4876481521527079,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4876481521527079
+    },
+    {
+      "train_step": 1714,
+      "loss": 0.8253570561267536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8253570561267536
+    },
+    {
+      "train_step": 1715,
+      "loss": 0.6657507367824761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6657507367824761
+    },
+    {
+      "train_step": 1716,
+      "loss": 0.6959728763518586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6959728763518586
+    },
+    {
+      "train_step": 1717,
+      "loss": 0.46024476776602463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46024476776602463
+    },
+    {
+      "train_step": 1718,
+      "loss": 0.5261435727237006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5261435727237006
+    },
+    {
+      "train_step": 1719,
+      "loss": 0.6278281738580174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6278281738580174
+    },
+    {
+      "train_step": 1720,
+      "loss": 0.6177016237957006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6177016237957006
+    },
+    {
+      "train_step": 1721,
+      "loss": 0.670689054837831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.670689054837831
+    },
+    {
+      "train_step": 1722,
+      "loss": 0.5680939677995582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5680939677995582
+    },
+    {
+      "train_step": 1723,
+      "loss": 0.7490351519635069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7490351519635069
+    },
+    {
+      "train_step": 1724,
+      "loss": 0.6526641420259819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6526641420259819
+    },
+    {
+      "train_step": 1725,
+      "loss": 0.5283902284038966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5283902284038966
+    },
+    {
+      "train_step": 1726,
+      "loss": 0.6606099732930196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6606099732930196
+    },
+    {
+      "train_step": 1727,
+      "loss": 0.8170919863279358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8170919863279358
+    },
+    {
+      "train_step": 1728,
+      "loss": 0.5084658395091062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5084658395091062
+    },
+    {
+      "train_step": 1729,
+      "loss": 0.6743488196918558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6743488196918558
+    },
+    {
+      "train_step": 1730,
+      "loss": 0.5339148593718459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5339148593718459
+    },
+    {
+      "train_step": 1731,
+      "loss": 0.6001122525890825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6001122525890825
+    },
+    {
+      "train_step": 1732,
+      "loss": 0.7042851752400914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7042851752400914
+    },
+    {
+      "train_step": 1733,
+      "loss": 0.6703732301677511,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6703732301677511
+    },
+    {
+      "train_step": 1734,
+      "loss": 0.6505591286215169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6505591286215169
+    },
+    {
+      "train_step": 1735,
+      "loss": 0.3984532005772223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3984532005772223
+    },
+    {
+      "train_step": 1736,
+      "loss": 0.7116957778332119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7116957778332119
+    },
+    {
+      "train_step": 1737,
+      "loss": 0.7028489416363405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7028489416363405
+    },
+    {
+      "train_step": 1738,
+      "loss": 0.6165487523911723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6165487523911723
+    },
+    {
+      "train_step": 1739,
+      "loss": 0.5958007477450037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5958007477450037
+    },
+    {
+      "train_step": 1740,
+      "loss": 0.6665246398828835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6665246398828835
+    },
+    {
+      "train_step": 1741,
+      "loss": 0.5879938036935926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5879938036935926
+    },
+    {
+      "train_step": 1742,
+      "loss": 0.3484829001037988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3484829001037988
+    },
+    {
+      "train_step": 1743,
+      "loss": 0.5502496856976401,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5502496856976401
+    },
+    {
+      "train_step": 1744,
+      "loss": 0.581915234685921,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.581915234685921
+    },
+    {
+      "train_step": 1745,
+      "loss": 0.4895537105345468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4895537105345468
+    },
+    {
+      "train_step": 1746,
+      "loss": 0.757774943257034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.757774943257034
+    },
+    {
+      "train_step": 1747,
+      "loss": 0.5820304678319667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5820304678319667
+    },
+    {
+      "train_step": 1748,
+      "loss": 0.5071891035477827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5071891035477827
+    },
+    {
+      "train_step": 1749,
+      "loss": 0.5322749268236188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5322749268236188
+    },
+    {
+      "train_step": 1750,
+      "loss": 0.5867409165564484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5867409165564484
+    },
+    {
+      "train_step": 1751,
+      "loss": 0.5246624155600933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5246624155600933
+    },
+    {
+      "train_step": 1752,
+      "loss": 0.6010888146223576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6010888146223576
+    },
+    {
+      "train_step": 1753,
+      "loss": 0.6437758362598621,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6437758362598621
+    },
+    {
+      "train_step": 1754,
+      "loss": 0.7812209278251403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7812209278251403
+    },
+    {
+      "train_step": 1755,
+      "loss": 0.4808807832992224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4808807832992224
+    },
+    {
+      "train_step": 1756,
+      "loss": 0.8270511167759242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8270511167759242
+    },
+    {
+      "train_step": 1757,
+      "loss": 0.5921046259375583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5921046259375583
+    },
+    {
+      "train_step": 1758,
+      "loss": 0.49784888657477155,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49784888657477155
+    },
+    {
+      "train_step": 1759,
+      "loss": 0.7423966097210156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7423966097210156
+    },
+    {
+      "train_step": 1760,
+      "loss": 0.47569556490619413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47569556490619413
+    },
+    {
+      "train_step": 1761,
+      "loss": 0.3940403443103824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3940403443103824
+    },
+    {
+      "train_step": 1762,
+      "loss": 0.5068812313112833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5068812313112833
+    },
+    {
+      "train_step": 1763,
+      "loss": 0.7398573707880844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7398573707880844
+    },
+    {
+      "train_step": 1764,
+      "loss": 0.6285935495133363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6285935495133363
+    },
+    {
+      "train_step": 1765,
+      "loss": 0.7062557087352695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7062557087352695
+    },
+    {
+      "train_step": 1766,
+      "loss": 0.42496520029870777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42496520029870777
+    },
+    {
+      "train_step": 1767,
+      "loss": 0.38254976040425787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38254976040425787
+    },
+    {
+      "train_step": 1768,
+      "loss": 0.5773646154844064,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5773646154844064
+    },
+    {
+      "train_step": 1769,
+      "loss": 0.47244673117071395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47244673117071395
+    },
+    {
+      "train_step": 1770,
+      "loss": 0.3674900436575632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3674900436575632
+    },
+    {
+      "train_step": 1771,
+      "loss": 0.37770803587956364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37770803587956364
+    },
+    {
+      "train_step": 1772,
+      "loss": 0.5606106575623129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5606106575623129
+    },
+    {
+      "train_step": 1773,
+      "loss": 0.5626910981294543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5626910981294543
+    },
+    {
+      "train_step": 1774,
+      "loss": 0.8636181826146847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8636181826146847
+    },
+    {
+      "train_step": 1775,
+      "loss": 0.7423026413439576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7423026413439576
+    },
+    {
+      "train_step": 1776,
+      "loss": 0.570548234659131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.570548234659131
+    },
+    {
+      "train_step": 1777,
+      "loss": 0.7228533513609233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7228533513609233
+    },
+    {
+      "train_step": 1778,
+      "loss": 0.4889003769128068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4889003769128068
+    },
+    {
+      "train_step": 1779,
+      "loss": 0.5600710238652571,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5600710238652571
+    },
+    {
+      "train_step": 1780,
+      "loss": 0.8334117315481124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8334117315481124
+    },
+    {
+      "train_step": 1781,
+      "loss": 0.627239592637088,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.627239592637088
+    },
+    {
+      "train_step": 1782,
+      "loss": 0.6044260096109209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6044260096109209
+    },
+    {
+      "train_step": 1783,
+      "loss": 0.48901357401809786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48901357401809786
+    },
+    {
+      "train_step": 1784,
+      "loss": 0.4006857980945578,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4006857980945578
+    },
+    {
+      "train_step": 1785,
+      "loss": 0.5579257032180749,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5579257032180749
+    },
+    {
+      "train_step": 1786,
+      "loss": 0.6453521529665251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6453521529665251
+    },
+    {
+      "train_step": 1787,
+      "loss": 0.552632087550878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.552632087550878
+    },
+    {
+      "train_step": 1788,
+      "loss": 0.7013928818290509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7013928818290509
+    },
+    {
+      "train_step": 1789,
+      "loss": 0.4858204865331632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4858204865331632
+    },
+    {
+      "train_step": 1790,
+      "loss": 0.32600997537491466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32600997537491466
+    },
+    {
+      "train_step": 1791,
+      "loss": 0.6370204232650755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6370204232650755
+    },
+    {
+      "train_step": 1792,
+      "loss": 0.5558302334308879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5558302334308879
+    },
+    {
+      "train_step": 1793,
+      "loss": 0.5583808884565168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5583808884565168
+    },
+    {
+      "train_step": 1794,
+      "loss": 0.4553148910592399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4553148910592399
+    },
+    {
+      "train_step": 1795,
+      "loss": 0.6419636153282011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6419636153282011
+    },
+    {
+      "train_step": 1796,
+      "loss": 0.5479381717564626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5479381717564626
+    },
+    {
+      "train_step": 1797,
+      "loss": 0.33855244547388125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33855244547388125
+    },
+    {
+      "train_step": 1798,
+      "loss": 0.5762577318791635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5762577318791635
+    },
+    {
+      "train_step": 1799,
+      "loss": 0.4600991442662281,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4600991442662281
+    },
+    {
+      "train_step": 1800,
+      "loss": 0.5817031923293267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5817031923293267
+    },
+    {
+      "train_step": 1801,
+      "loss": 0.5484235484767607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5484235484767607
+    },
+    {
+      "train_step": 1802,
+      "loss": 0.4829156858250477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4829156858250477
+    },
+    {
+      "train_step": 1803,
+      "loss": 0.6087593142398227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6087593142398227
+    },
+    {
+      "train_step": 1804,
+      "loss": 0.7018197890829241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7018197890829241
+    },
+    {
+      "train_step": 1805,
+      "loss": 0.628632392367681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.628632392367681
+    },
+    {
+      "train_step": 1806,
+      "loss": 0.5097306852823662,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5097306852823662
+    },
+    {
+      "train_step": 1807,
+      "loss": 0.4617776049853053,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4617776049853053
+    },
+    {
+      "train_step": 1808,
+      "loss": 0.6210519863101329,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6210519863101329
+    },
+    {
+      "train_step": 1809,
+      "loss": 0.7082556252203076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7082556252203076
+    },
+    {
+      "train_step": 1810,
+      "loss": 0.45149216108447837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45149216108447837
+    },
+    {
+      "train_step": 1811,
+      "loss": 0.5597542253649178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5597542253649178
+    },
+    {
+      "train_step": 1812,
+      "loss": 0.4906003561249279,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4906003561249279
+    },
+    {
+      "train_step": 1813,
+      "loss": 0.6341383066824655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6341383066824655
+    },
+    {
+      "train_step": 1814,
+      "loss": 0.5565510644660634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5565510644660634
+    },
+    {
+      "train_step": 1815,
+      "loss": 0.7587748955711845,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7587748955711845
+    },
+    {
+      "train_step": 1816,
+      "loss": 0.7328990225099805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7328990225099805
+    },
+    {
+      "train_step": 1817,
+      "loss": 0.7735787507955627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7735787507955627
+    },
+    {
+      "train_step": 1818,
+      "loss": 0.573277183611717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.573277183611717
+    },
+    {
+      "train_step": 1819,
+      "loss": 0.4590067433598924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4590067433598924
+    },
+    {
+      "train_step": 1820,
+      "loss": 0.7344747903814275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7344747903814275
+    },
+    {
+      "train_step": 1821,
+      "loss": 0.8077733361472922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8077733361472922
+    },
+    {
+      "train_step": 1822,
+      "loss": 0.6052804371742194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6052804371742194
+    },
+    {
+      "train_step": 1823,
+      "loss": 0.5389410260367921,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5389410260367921
+    },
+    {
+      "train_step": 1824,
+      "loss": 0.6683586163485419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6683586163485419
+    },
+    {
+      "train_step": 1825,
+      "loss": 0.6535718143054169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6535718143054169
+    },
+    {
+      "train_step": 1826,
+      "loss": 0.6159138416417258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6159138416417258
+    },
+    {
+      "train_step": 1827,
+      "loss": 0.816765548818925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.816765548818925
+    },
+    {
+      "train_step": 1828,
+      "loss": 0.5734282216592118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5734282216592118
+    },
+    {
+      "train_step": 1829,
+      "loss": 0.5497924544165149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5497924544165149
+    },
+    {
+      "train_step": 1830,
+      "loss": 0.5702051443668625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5702051443668625
+    },
+    {
+      "train_step": 1831,
+      "loss": 0.7537822350622561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7537822350622561
+    },
+    {
+      "train_step": 1832,
+      "loss": 0.3921115984173088,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3921115984173088
+    },
+    {
+      "train_step": 1833,
+      "loss": 0.603720831263986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.603720831263986
+    },
+    {
+      "train_step": 1834,
+      "loss": 0.5706242552611174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706242552611174
+    },
+    {
+      "train_step": 1835,
+      "loss": 0.41643573323014677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41643573323014677
+    },
+    {
+      "train_step": 1836,
+      "loss": 0.7151498920033159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7151498920033159
+    },
+    {
+      "train_step": 1837,
+      "loss": 0.7738190521334415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7738190521334415
+    },
+    {
+      "train_step": 1838,
+      "loss": 0.4832318084593766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4832318084593766
+    },
+    {
+      "train_step": 1839,
+      "loss": 0.6008404131621906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6008404131621906
+    },
+    {
+      "train_step": 1840,
+      "loss": 0.5769106553535188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5769106553535188
+    },
+    {
+      "train_step": 1841,
+      "loss": 0.5926563464040069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5926563464040069
+    },
+    {
+      "train_step": 1842,
+      "loss": 0.6019763163833985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6019763163833985
+    },
+    {
+      "train_step": 1843,
+      "loss": 0.4348883282741581,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4348883282741581
+    },
+    {
+      "train_step": 1844,
+      "loss": 0.6149649703307686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6149649703307686
+    },
+    {
+      "train_step": 1845,
+      "loss": 0.5057113854807618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5057113854807618
+    },
+    {
+      "train_step": 1846,
+      "loss": 0.6132526557305246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6132526557305246
+    },
+    {
+      "train_step": 1847,
+      "loss": 0.4963563495655673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4963563495655673
+    },
+    {
+      "train_step": 1848,
+      "loss": 0.5275832059007581,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5275832059007581
+    },
+    {
+      "train_step": 1849,
+      "loss": 0.35432939682871634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35432939682871634
+    },
+    {
+      "train_step": 1850,
+      "loss": 0.9006168740352293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9006168740352293
+    },
+    {
+      "train_step": 1851,
+      "loss": 0.5870591069505751,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5870591069505751
+    },
+    {
+      "train_step": 1852,
+      "loss": 0.5761770953324106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5761770953324106
+    },
+    {
+      "train_step": 1853,
+      "loss": 0.6441192086209366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6441192086209366
+    },
+    {
+      "train_step": 1854,
+      "loss": 0.4433731119953642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4433731119953642
+    },
+    {
+      "train_step": 1855,
+      "loss": 0.7277952989569201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7277952989569201
+    },
+    {
+      "train_step": 1856,
+      "loss": 0.3979469574929302,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3979469574929302
+    },
+    {
+      "train_step": 1857,
+      "loss": 0.7428339169002239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7428339169002239
+    },
+    {
+      "train_step": 1858,
+      "loss": 0.5995777842874112,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5995777842874112
+    },
+    {
+      "train_step": 1859,
+      "loss": 0.6647012448454954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6647012448454954
+    },
+    {
+      "train_step": 1860,
+      "loss": 0.48158657491872026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48158657491872026
+    },
+    {
+      "train_step": 1861,
+      "loss": 0.5950446355958485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5950446355958485
+    },
+    {
+      "train_step": 1862,
+      "loss": 0.5027043025159954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5027043025159954
+    },
+    {
+      "train_step": 1863,
+      "loss": 0.35651641707656284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35651641707656284
+    },
+    {
+      "train_step": 1864,
+      "loss": 0.4536607375447911,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4536607375447911
+    },
+    {
+      "train_step": 1865,
+      "loss": 0.6685300012603413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6685300012603413
+    },
+    {
+      "train_step": 1866,
+      "loss": 0.5637420145110065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5637420145110065
+    },
+    {
+      "train_step": 1867,
+      "loss": 0.6978010796866979,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6978010796866979
+    },
+    {
+      "train_step": 1868,
+      "loss": 0.6250761070270658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6250761070270658
+    },
+    {
+      "train_step": 1869,
+      "loss": 0.5550163666494737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5550163666494737
+    },
+    {
+      "train_step": 1870,
+      "loss": 0.5257175545641478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5257175545641478
+    },
+    {
+      "train_step": 1871,
+      "loss": 0.5354103910194957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5354103910194957
+    },
+    {
+      "train_step": 1872,
+      "loss": 0.4908927436379643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4908927436379643
+    },
+    {
+      "train_step": 1873,
+      "loss": 0.42981417215798434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42981417215798434
+    },
+    {
+      "train_step": 1874,
+      "loss": 0.6587338216955477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6587338216955477
+    },
+    {
+      "train_step": 1875,
+      "loss": 0.5703683650697239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5703683650697239
+    },
+    {
+      "train_step": 1876,
+      "loss": 0.6715231284329831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6715231284329831
+    },
+    {
+      "train_step": 1877,
+      "loss": 0.6444598013978562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6444598013978562
+    },
+    {
+      "train_step": 1878,
+      "loss": 0.5953608542545795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5953608542545795
+    },
+    {
+      "train_step": 1879,
+      "loss": 0.5633613780130213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5633613780130213
+    },
+    {
+      "train_step": 1880,
+      "loss": 0.7019585910907883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7019585910907883
+    },
+    {
+      "train_step": 1881,
+      "loss": 0.6720233873092625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6720233873092625
+    },
+    {
+      "train_step": 1882,
+      "loss": 0.6920170190598781,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6920170190598781
+    },
+    {
+      "train_step": 1883,
+      "loss": 0.46185124929437693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46185124929437693
+    },
+    {
+      "train_step": 1884,
+      "loss": 0.564143987885117,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.564143987885117
+    },
+    {
+      "train_step": 1885,
+      "loss": 0.5432261163830743,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5432261163830743
+    },
+    {
+      "train_step": 1886,
+      "loss": 0.709951151674002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.709951151674002
+    },
+    {
+      "train_step": 1887,
+      "loss": 0.6637421086476955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6637421086476955
+    },
+    {
+      "train_step": 1888,
+      "loss": 0.4440762355334083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4440762355334083
+    },
+    {
+      "train_step": 1889,
+      "loss": 0.5317913781814859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5317913781814859
+    },
+    {
+      "train_step": 1890,
+      "loss": 0.752338204571415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.752338204571415
+    },
+    {
+      "train_step": 1891,
+      "loss": 0.5995461057538202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5995461057538202
+    },
+    {
+      "train_step": 1892,
+      "loss": 0.5655947387945377,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5655947387945377
+    },
+    {
+      "train_step": 1893,
+      "loss": 0.7791565631797431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7791565631797431
+    },
+    {
+      "train_step": 1894,
+      "loss": 0.6648814854749578,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6648814854749578
+    },
+    {
+      "train_step": 1895,
+      "loss": 0.4334999850934094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4334999850934094
+    },
+    {
+      "train_step": 1896,
+      "loss": 0.5653287104513065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5653287104513065
+    },
+    {
+      "train_step": 1897,
+      "loss": 0.4800967284591392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4800967284591392
+    },
+    {
+      "train_step": 1898,
+      "loss": 0.4001851810114692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4001851810114692
+    },
+    {
+      "train_step": 1899,
+      "loss": 0.6038126595961879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6038126595961879
+    },
+    {
+      "train_step": 1900,
+      "loss": 0.5204035545922644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5204035545922644
+    },
+    {
+      "train_step": 1901,
+      "loss": 0.6275476124128698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6275476124128698
+    },
+    {
+      "train_step": 1902,
+      "loss": 0.6226121530588394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6226121530588394
+    },
+    {
+      "train_step": 1903,
+      "loss": 0.372051158671527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.372051158671527
+    },
+    {
+      "train_step": 1904,
+      "loss": 0.4803345815723126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4803345815723126
+    },
+    {
+      "train_step": 1905,
+      "loss": 0.46602546865889766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46602546865889766
+    },
+    {
+      "train_step": 1906,
+      "loss": 0.5369257551364618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5369257551364618
+    },
+    {
+      "train_step": 1907,
+      "loss": 0.45508517570440066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45508517570440066
+    },
+    {
+      "train_step": 1908,
+      "loss": 0.5496436947284363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5496436947284363
+    },
+    {
+      "train_step": 1909,
+      "loss": 0.6391575244662797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6391575244662797
+    },
+    {
+      "train_step": 1910,
+      "loss": 0.7341338367081269,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7341338367081269
+    },
+    {
+      "train_step": 1911,
+      "loss": 0.5665811637789616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5665811637789616
+    },
+    {
+      "train_step": 1912,
+      "loss": 0.5603398555792478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5603398555792478
+    },
+    {
+      "train_step": 1913,
+      "loss": 0.5784720090232127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5784720090232127
+    },
+    {
+      "train_step": 1914,
+      "loss": 0.5891303313546831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5891303313546831
+    },
+    {
+      "train_step": 1915,
+      "loss": 0.4305855189059085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4305855189059085
+    },
+    {
+      "train_step": 1916,
+      "loss": 0.6905285504729022,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6905285504729022
+    },
+    {
+      "train_step": 1917,
+      "loss": 0.5396615539214987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5396615539214987
+    },
+    {
+      "train_step": 1918,
+      "loss": 0.5165383773008798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5165383773008798
+    },
+    {
+      "train_step": 1919,
+      "loss": 0.6461742870322563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6461742870322563
+    },
+    {
+      "train_step": 1920,
+      "loss": 0.6003343976013458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6003343976013458
+    },
+    {
+      "train_step": 1921,
+      "loss": 0.40617547478212496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40617547478212496
+    },
+    {
+      "train_step": 1922,
+      "loss": 0.5250017001608308,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5250017001608308
+    },
+    {
+      "train_step": 1923,
+      "loss": 0.7437645852217402,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7437645852217402
+    },
+    {
+      "train_step": 1924,
+      "loss": 0.5911303092906066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5911303092906066
+    },
+    {
+      "train_step": 1925,
+      "loss": 0.576991815263663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576991815263663
+    },
+    {
+      "train_step": 1926,
+      "loss": 0.655578544896594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.655578544896594
+    },
+    {
+      "train_step": 1927,
+      "loss": 0.5976435617973977,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5976435617973977
+    },
+    {
+      "train_step": 1928,
+      "loss": 0.5965786060713069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5965786060713069
+    },
+    {
+      "train_step": 1929,
+      "loss": 0.5984469577287764,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5984469577287764
+    },
+    {
+      "train_step": 1930,
+      "loss": 0.7248767209117077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7248767209117077
+    },
+    {
+      "train_step": 1931,
+      "loss": 0.5072910981521058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5072910981521058
+    },
+    {
+      "train_step": 1932,
+      "loss": 0.5409948245458768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5409948245458768
+    },
+    {
+      "train_step": 1933,
+      "loss": 0.5018186242157154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5018186242157154
+    },
+    {
+      "train_step": 1934,
+      "loss": 0.5412377483746454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5412377483746454
+    },
+    {
+      "train_step": 1935,
+      "loss": 0.4789667913858685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4789667913858685
+    },
+    {
+      "train_step": 1936,
+      "loss": 0.6051933848116422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6051933848116422
+    },
+    {
+      "train_step": 1937,
+      "loss": 0.5332142807136686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5332142807136686
+    },
+    {
+      "train_step": 1938,
+      "loss": 0.6652799641303603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6652799641303603
+    },
+    {
+      "train_step": 1939,
+      "loss": 0.6407901127852283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6407901127852283
+    },
+    {
+      "train_step": 1940,
+      "loss": 0.5269471163866556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5269471163866556
+    },
+    {
+      "train_step": 1941,
+      "loss": 0.4546497006746765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4546497006746765
+    },
+    {
+      "train_step": 1942,
+      "loss": 0.42637224926120315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42637224926120315
+    },
+    {
+      "train_step": 1943,
+      "loss": 0.5359645663135116,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5359645663135116
+    },
+    {
+      "train_step": 1944,
+      "loss": 0.7994907701860786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7994907701860786
+    },
+    {
+      "train_step": 1945,
+      "loss": 0.6165666899363361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6165666899363361
+    },
+    {
+      "train_step": 1946,
+      "loss": 0.5506840610681213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5506840610681213
+    },
+    {
+      "train_step": 1947,
+      "loss": 0.6320564748514804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6320564748514804
+    },
+    {
+      "train_step": 1948,
+      "loss": 0.5225439865952796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5225439865952796
+    },
+    {
+      "train_step": 1949,
+      "loss": 0.5349083713581564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5349083713581564
+    },
+    {
+      "train_step": 1950,
+      "loss": 0.5618207565400894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5618207565400894
+    },
+    {
+      "train_step": 1951,
+      "loss": 0.5154996546399867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5154996546399867
+    },
+    {
+      "train_step": 1952,
+      "loss": 0.5419437004639713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5419437004639713
+    },
+    {
+      "train_step": 1953,
+      "loss": 0.6732246856797881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6732246856797881
+    },
+    {
+      "train_step": 1954,
+      "loss": 0.6608760619839191,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6608760619839191
+    },
+    {
+      "train_step": 1955,
+      "loss": 0.3872799095590593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3872799095590593
+    },
+    {
+      "train_step": 1956,
+      "loss": 0.3475660238950574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3475660238950574
+    },
+    {
+      "train_step": 1957,
+      "loss": 0.411525886841637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.411525886841637
+    },
+    {
+      "train_step": 1958,
+      "loss": 0.45931495575934644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45931495575934644
+    },
+    {
+      "train_step": 1959,
+      "loss": 0.6499766170980779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6499766170980779
+    },
+    {
+      "train_step": 1960,
+      "loss": 0.6107692761592175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6107692761592175
+    },
+    {
+      "train_step": 1961,
+      "loss": 0.8194751491370519,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8194751491370519
+    },
+    {
+      "train_step": 1962,
+      "loss": 0.5609774447417663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5609774447417663
+    },
+    {
+      "train_step": 1963,
+      "loss": 0.5400725610617209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5400725610617209
+    },
+    {
+      "train_step": 1964,
+      "loss": 0.6377672791639057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6377672791639057
+    },
+    {
+      "train_step": 1965,
+      "loss": 0.5296742849422239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5296742849422239
+    },
+    {
+      "train_step": 1966,
+      "loss": 0.6005738332503614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6005738332503614
+    },
+    {
+      "train_step": 1967,
+      "loss": 0.36946804706699476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36946804706699476
+    },
+    {
+      "train_step": 1968,
+      "loss": 0.7492068166682322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7492068166682322
+    },
+    {
+      "train_step": 1969,
+      "loss": 0.5415561580762785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5415561580762785
+    },
+    {
+      "train_step": 1970,
+      "loss": 0.5253994166265807,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5253994166265807
+    },
+    {
+      "train_step": 1971,
+      "loss": 0.5309425099475036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5309425099475036
+    },
+    {
+      "train_step": 1972,
+      "loss": 0.6045874596485659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6045874596485659
+    },
+    {
+      "train_step": 1973,
+      "loss": 0.5962914274518166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962914274518166
+    },
+    {
+      "train_step": 1974,
+      "loss": 0.5121457870626169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5121457870626169
+    },
+    {
+      "train_step": 1975,
+      "loss": 0.7853241507716012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7853241507716012
+    },
+    {
+      "train_step": 1976,
+      "loss": 0.6206193320542164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6206193320542164
+    },
+    {
+      "train_step": 1977,
+      "loss": 0.48886519398903455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48886519398903455
+    },
+    {
+      "train_step": 1978,
+      "loss": 0.4411155080843233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4411155080843233
+    },
+    {
+      "train_step": 1979,
+      "loss": 0.58975618602318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.58975618602318
+    },
+    {
+      "train_step": 1980,
+      "loss": 0.6171875743603259,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6171875743603259
+    },
+    {
+      "train_step": 1981,
+      "loss": 0.6794596751734714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6794596751734714
+    },
+    {
+      "train_step": 1982,
+      "loss": 0.5794404850093885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5794404850093885
+    },
+    {
+      "train_step": 1983,
+      "loss": 0.7532207944416098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7532207944416098
+    },
+    {
+      "train_step": 1984,
+      "loss": 0.4898171945430774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4898171945430774
+    },
+    {
+      "train_step": 1985,
+      "loss": 0.6581944202239843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6581944202239843
+    },
+    {
+      "train_step": 1986,
+      "loss": 0.519102470206636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.519102470206636
+    },
+    {
+      "train_step": 1987,
+      "loss": 0.6253660892969037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6253660892969037
+    },
+    {
+      "train_step": 1988,
+      "loss": 0.45430029021239526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45430029021239526
+    },
+    {
+      "train_step": 1989,
+      "loss": 0.518472561760292,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.518472561760292
+    },
+    {
+      "train_step": 1990,
+      "loss": 0.5659178250008928,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5659178250008928
+    },
+    {
+      "train_step": 1991,
+      "loss": 0.5677303376134273,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5677303376134273
+    },
+    {
+      "train_step": 1992,
+      "loss": 1.0140196377702118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 1.0140196377702118
+    },
+    {
+      "train_step": 1993,
+      "loss": 0.5122077790728214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5122077790728214
+    },
+    {
+      "train_step": 1994,
+      "loss": 0.39974861047046295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39974861047046295
+    },
+    {
+      "train_step": 1995,
+      "loss": 0.4427669715593237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4427669715593237
+    },
+    {
+      "train_step": 1996,
+      "loss": 0.5036714343508006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5036714343508006
+    },
+    {
+      "train_step": 1997,
+      "loss": 0.4359818919516128,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4359818919516128
+    },
+    {
+      "train_step": 1998,
+      "loss": 0.5308263639445178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5308263639445178
+    },
+    {
+      "train_step": 1999,
+      "loss": 0.6996584737222817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6996584737222817
+    },
+    {
+      "train_step": 2000,
+      "loss": 0.69548492746983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.69548492746983
+    },
+    {
+      "train_step": 2001,
+      "loss": 0.5313750978518839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5313750978518839
+    },
+    {
+      "train_step": 2002,
+      "loss": 0.6903267842828817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6903267842828817
+    },
+    {
+      "train_step": 2003,
+      "loss": 0.6328702314841769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6328702314841769
+    },
+    {
+      "train_step": 2004,
+      "loss": 0.765597320715342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.765597320715342
+    },
+    {
+      "train_step": 2005,
+      "loss": 0.5234612049453794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5234612049453794
+    },
+    {
+      "train_step": 2006,
+      "loss": 0.47081993175387693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47081993175387693
+    },
+    {
+      "train_step": 2007,
+      "loss": 0.40117109278737534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40117109278737534
+    },
+    {
+      "train_step": 2008,
+      "loss": 0.539606206979694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.539606206979694
+    },
+    {
+      "train_step": 2009,
+      "loss": 0.5733113579593386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5733113579593386
+    },
+    {
+      "train_step": 2010,
+      "loss": 0.6044512563439216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6044512563439216
+    },
+    {
+      "train_step": 2011,
+      "loss": 0.6093221709882398,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6093221709882398
+    },
+    {
+      "train_step": 2012,
+      "loss": 0.5461518779663311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5461518779663311
+    },
+    {
+      "train_step": 2013,
+      "loss": 0.5436247049664409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5436247049664409
+    },
+    {
+      "train_step": 2014,
+      "loss": 0.42949747738951466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42949747738951466
+    },
+    {
+      "train_step": 2015,
+      "loss": 0.4192571165860698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4192571165860698
+    },
+    {
+      "train_step": 2016,
+      "loss": 0.6114456405203768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6114456405203768
+    },
+    {
+      "train_step": 2017,
+      "loss": 0.6755251281779285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6755251281779285
+    },
+    {
+      "train_step": 2018,
+      "loss": 0.4784348738097294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4784348738097294
+    },
+    {
+      "train_step": 2019,
+      "loss": 0.4770997869540453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4770997869540453
+    },
+    {
+      "train_step": 2020,
+      "loss": 0.3903865649391971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3903865649391971
+    },
+    {
+      "train_step": 2021,
+      "loss": 0.677034644754944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.677034644754944
+    },
+    {
+      "train_step": 2022,
+      "loss": 0.3434383790873113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3434383790873113
+    },
+    {
+      "train_step": 2023,
+      "loss": 0.3410019887612171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3410019887612171
+    },
+    {
+      "train_step": 2024,
+      "loss": 0.6163596678292154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6163596678292154
+    },
+    {
+      "train_step": 2025,
+      "loss": 0.7519674314224043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7519674314224043
+    },
+    {
+      "train_step": 2026,
+      "loss": 0.7620703737294113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7620703737294113
+    },
+    {
+      "train_step": 2027,
+      "loss": 0.5431350000819511,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5431350000819511
+    },
+    {
+      "train_step": 2028,
+      "loss": 0.6949556776934697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6949556776934697
+    },
+    {
+      "train_step": 2029,
+      "loss": 0.4380874361292518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4380874361292518
+    },
+    {
+      "train_step": 2030,
+      "loss": 0.692627430928219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.692627430928219
+    },
+    {
+      "train_step": 2031,
+      "loss": 0.48297798935999503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48297798935999503
+    },
+    {
+      "train_step": 2032,
+      "loss": 0.4394825758142786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4394825758142786
+    },
+    {
+      "train_step": 2033,
+      "loss": 0.3561409682378896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3561409682378896
+    },
+    {
+      "train_step": 2034,
+      "loss": 0.43129821345312236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43129821345312236
+    },
+    {
+      "train_step": 2035,
+      "loss": 0.6227681118806185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6227681118806185
+    },
+    {
+      "train_step": 2036,
+      "loss": 0.4901533488938832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4901533488938832
+    },
+    {
+      "train_step": 2037,
+      "loss": 0.5000082165241554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5000082165241554
+    },
+    {
+      "train_step": 2038,
+      "loss": 0.61938009655005,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.61938009655005
+    },
+    {
+      "train_step": 2039,
+      "loss": 0.6168109703196686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6168109703196686
+    },
+    {
+      "train_step": 2040,
+      "loss": 0.5651578736689123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5651578736689123
+    },
+    {
+      "train_step": 2041,
+      "loss": 0.5607536595096304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5607536595096304
+    },
+    {
+      "train_step": 2042,
+      "loss": 0.6585141354407852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6585141354407852
+    },
+    {
+      "train_step": 2043,
+      "loss": 0.651621087450492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.651621087450492
+    },
+    {
+      "train_step": 2044,
+      "loss": 0.7256786482651324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7256786482651324
+    },
+    {
+      "train_step": 2045,
+      "loss": 0.43244873699316794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43244873699316794
+    },
+    {
+      "train_step": 2046,
+      "loss": 0.6535717374129589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6535717374129589
+    },
+    {
+      "train_step": 2047,
+      "loss": 0.3822913922974227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3822913922974227
+    },
+    {
+      "train_step": 2048,
+      "loss": 0.6443552833668322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6443552833668322
+    },
+    {
+      "train_step": 2049,
+      "loss": 0.585232052039028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.585232052039028
+    },
+    {
+      "train_step": 2050,
+      "loss": 0.4603327967172379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4603327967172379
+    },
+    {
+      "train_step": 2051,
+      "loss": 0.6070449768348332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6070449768348332
+    },
+    {
+      "train_step": 2052,
+      "loss": 0.4062315639112041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4062315639112041
+    },
+    {
+      "train_step": 2053,
+      "loss": 0.6491044913161919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6491044913161919
+    },
+    {
+      "train_step": 2054,
+      "loss": 0.4870999274437374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4870999274437374
+    },
+    {
+      "train_step": 2055,
+      "loss": 0.5797430532118433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5797430532118433
+    },
+    {
+      "train_step": 2056,
+      "loss": 0.6349862587107472,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6349862587107472
+    },
+    {
+      "train_step": 2057,
+      "loss": 0.8260675753961059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8260675753961059
+    },
+    {
+      "train_step": 2058,
+      "loss": 0.5754767034741514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5754767034741514
+    },
+    {
+      "train_step": 2059,
+      "loss": 0.6356342918874484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6356342918874484
+    },
+    {
+      "train_step": 2060,
+      "loss": 0.5502577939981431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5502577939981431
+    },
+    {
+      "train_step": 2061,
+      "loss": 0.5715657546093015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5715657546093015
+    },
+    {
+      "train_step": 2062,
+      "loss": 0.44524180567537347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44524180567537347
+    },
+    {
+      "train_step": 2063,
+      "loss": 0.6358447566619148,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6358447566619148
+    },
+    {
+      "train_step": 2064,
+      "loss": 0.6445262383420649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6445262383420649
+    },
+    {
+      "train_step": 2065,
+      "loss": 0.646078284417912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.646078284417912
+    },
+    {
+      "train_step": 2066,
+      "loss": 0.6512197325222927,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6512197325222927
+    },
+    {
+      "train_step": 2067,
+      "loss": 0.4856025410850216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4856025410850216
+    },
+    {
+      "train_step": 2068,
+      "loss": 0.7019698119570499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7019698119570499
+    },
+    {
+      "train_step": 2069,
+      "loss": 0.545594552203506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.545594552203506
+    },
+    {
+      "train_step": 2070,
+      "loss": 0.5505226178853014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5505226178853014
+    },
+    {
+      "train_step": 2071,
+      "loss": 0.5023793285387878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5023793285387878
+    },
+    {
+      "train_step": 2072,
+      "loss": 0.45465785620588833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45465785620588833
+    },
+    {
+      "train_step": 2073,
+      "loss": 0.4516818666845821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4516818666845821
+    },
+    {
+      "train_step": 2074,
+      "loss": 0.5646046332053997,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5646046332053997
+    },
+    {
+      "train_step": 2075,
+      "loss": 0.3785637875337361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3785637875337361
+    },
+    {
+      "train_step": 2076,
+      "loss": 0.63105432152598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.63105432152598
+    },
+    {
+      "train_step": 2077,
+      "loss": 0.7524301609628801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7524301609628801
+    },
+    {
+      "train_step": 2078,
+      "loss": 0.36468733101928713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36468733101928713
+    },
+    {
+      "train_step": 2079,
+      "loss": 0.6675301090508474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6675301090508474
+    },
+    {
+      "train_step": 2080,
+      "loss": 0.5711387101370939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5711387101370939
+    },
+    {
+      "train_step": 2081,
+      "loss": 0.37797342236319287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37797342236319287
+    },
+    {
+      "train_step": 2082,
+      "loss": 0.503397083235625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.503397083235625
+    },
+    {
+      "train_step": 2083,
+      "loss": 0.4890154854953861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4890154854953861
+    },
+    {
+      "train_step": 2084,
+      "loss": 0.5674850509552434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5674850509552434
+    },
+    {
+      "train_step": 2085,
+      "loss": 0.5772307694672216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5772307694672216
+    },
+    {
+      "train_step": 2086,
+      "loss": 0.504651075086896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.504651075086896
+    },
+    {
+      "train_step": 2087,
+      "loss": 0.4098708035981332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4098708035981332
+    },
+    {
+      "train_step": 2088,
+      "loss": 0.5302472993282404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5302472993282404
+    },
+    {
+      "train_step": 2089,
+      "loss": 0.5753196633868287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5753196633868287
+    },
+    {
+      "train_step": 2090,
+      "loss": 0.5680681796705987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5680681796705987
+    },
+    {
+      "train_step": 2091,
+      "loss": 0.6515409816381389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6515409816381389
+    },
+    {
+      "train_step": 2092,
+      "loss": 0.7564567417859556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7564567417859556
+    },
+    {
+      "train_step": 2093,
+      "loss": 0.710364534174543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.710364534174543
+    },
+    {
+      "train_step": 2094,
+      "loss": 0.7375495848988799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7375495848988799
+    },
+    {
+      "train_step": 2095,
+      "loss": 0.6112776283906909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6112776283906909
+    },
+    {
+      "train_step": 2096,
+      "loss": 0.45127043713666626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45127043713666626
+    },
+    {
+      "train_step": 2097,
+      "loss": 0.6512035329006065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6512035329006065
+    },
+    {
+      "train_step": 2098,
+      "loss": 0.6518493010056848,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6518493010056848
+    },
+    {
+      "train_step": 2099,
+      "loss": 0.6857906277540232,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6857906277540232
+    },
+    {
+      "train_step": 2100,
+      "loss": 0.4487219426208367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4487219426208367
+    },
+    {
+      "train_step": 2101,
+      "loss": 0.5312247344942241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5312247344942241
+    },
+    {
+      "train_step": 2102,
+      "loss": 0.6584007625797959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6584007625797959
+    },
+    {
+      "train_step": 2103,
+      "loss": 0.6296309111310898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6296309111310898
+    },
+    {
+      "train_step": 2104,
+      "loss": 0.46979725788874993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46979725788874993
+    },
+    {
+      "train_step": 2105,
+      "loss": 0.6667562023502486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6667562023502486
+    },
+    {
+      "train_step": 2106,
+      "loss": 0.5747708363035512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5747708363035512
+    },
+    {
+      "train_step": 2107,
+      "loss": 0.5880311124244858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5880311124244858
+    },
+    {
+      "train_step": 2108,
+      "loss": 0.531121944232403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.531121944232403
+    },
+    {
+      "train_step": 2109,
+      "loss": 0.5454788664575295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5454788664575295
+    },
+    {
+      "train_step": 2110,
+      "loss": 0.4343728463772064,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4343728463772064
+    },
+    {
+      "train_step": 2111,
+      "loss": 0.7728562829619079,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7728562829619079
+    },
+    {
+      "train_step": 2112,
+      "loss": 0.43882888788247004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43882888788247004
+    },
+    {
+      "train_step": 2113,
+      "loss": 0.46160452989092604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46160452989092604
+    },
+    {
+      "train_step": 2114,
+      "loss": 0.698936384789656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.698936384789656
+    },
+    {
+      "train_step": 2115,
+      "loss": 0.6976420164197839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6976420164197839
+    },
+    {
+      "train_step": 2116,
+      "loss": 0.4331907426652305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4331907426652305
+    },
+    {
+      "train_step": 2117,
+      "loss": 0.49398600101891554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49398600101891554
+    },
+    {
+      "train_step": 2118,
+      "loss": 0.7441300428512545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7441300428512545
+    },
+    {
+      "train_step": 2119,
+      "loss": 0.5091816658055139,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5091816658055139
+    },
+    {
+      "train_step": 2120,
+      "loss": 0.4220119543074655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4220119543074655
+    },
+    {
+      "train_step": 2121,
+      "loss": 0.49686322359191554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49686322359191554
+    },
+    {
+      "train_step": 2122,
+      "loss": 0.4621643179924793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4621643179924793
+    },
+    {
+      "train_step": 2123,
+      "loss": 0.7410017102495461,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7410017102495461
+    },
+    {
+      "train_step": 2124,
+      "loss": 0.30480949710468597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.30480949710468597
+    },
+    {
+      "train_step": 2125,
+      "loss": 0.6545808463876057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6545808463876057
+    },
+    {
+      "train_step": 2126,
+      "loss": 0.39998747703273746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39998747703273746
+    },
+    {
+      "train_step": 2127,
+      "loss": 0.6669974111401481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6669974111401481
+    },
+    {
+      "train_step": 2128,
+      "loss": 0.6996935481351331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6996935481351331
+    },
+    {
+      "train_step": 2129,
+      "loss": 0.526442972052805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.526442972052805
+    },
+    {
+      "train_step": 2130,
+      "loss": 0.6960774280687181,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6960774280687181
+    },
+    {
+      "train_step": 2131,
+      "loss": 0.6623306577820895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6623306577820895
+    },
+    {
+      "train_step": 2132,
+      "loss": 0.6462154306261849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6462154306261849
+    },
+    {
+      "train_step": 2133,
+      "loss": 0.5830855328132047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5830855328132047
+    },
+    {
+      "train_step": 2134,
+      "loss": 0.40083392699844506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40083392699844506
+    },
+    {
+      "train_step": 2135,
+      "loss": 0.5285469970342935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5285469970342935
+    },
+    {
+      "train_step": 2136,
+      "loss": 0.6183024103253315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6183024103253315
+    },
+    {
+      "train_step": 2137,
+      "loss": 0.617940591101257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.617940591101257
+    },
+    {
+      "train_step": 2138,
+      "loss": 0.44608637080347513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44608637080347513
+    },
+    {
+      "train_step": 2139,
+      "loss": 0.5870330710181186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5870330710181186
+    },
+    {
+      "train_step": 2140,
+      "loss": 0.8096559443800297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8096559443800297
+    },
+    {
+      "train_step": 2141,
+      "loss": 0.4596811972450903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4596811972450903
+    },
+    {
+      "train_step": 2142,
+      "loss": 0.7302107345592203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7302107345592203
+    },
+    {
+      "train_step": 2143,
+      "loss": 0.5007994756581893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5007994756581893
+    },
+    {
+      "train_step": 2144,
+      "loss": 0.6903462331223915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6903462331223915
+    },
+    {
+      "train_step": 2145,
+      "loss": 0.44216097553131917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44216097553131917
+    },
+    {
+      "train_step": 2146,
+      "loss": 0.7157644581201518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7157644581201518
+    },
+    {
+      "train_step": 2147,
+      "loss": 0.8488503884524683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8488503884524683
+    },
+    {
+      "train_step": 2148,
+      "loss": 0.5413250925622275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5413250925622275
+    },
+    {
+      "train_step": 2149,
+      "loss": 0.504529404963294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.504529404963294
+    },
+    {
+      "train_step": 2150,
+      "loss": 0.6423824665883183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6423824665883183
+    },
+    {
+      "train_step": 2151,
+      "loss": 0.7414252637047123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7414252637047123
+    },
+    {
+      "train_step": 2152,
+      "loss": 0.7656249843858306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7656249843858306
+    },
+    {
+      "train_step": 2153,
+      "loss": 0.7197713039605482,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7197713039605482
+    },
+    {
+      "train_step": 2154,
+      "loss": 0.6490670128713067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6490670128713067
+    },
+    {
+      "train_step": 2155,
+      "loss": 0.5009788374704359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5009788374704359
+    },
+    {
+      "train_step": 2156,
+      "loss": 0.43384848508741497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43384848508741497
+    },
+    {
+      "train_step": 2157,
+      "loss": 0.4138169373088807,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4138169373088807
+    },
+    {
+      "train_step": 2158,
+      "loss": 0.4454836542711472,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4454836542711472
+    },
+    {
+      "train_step": 2159,
+      "loss": 0.5361574888130182,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5361574888130182
+    },
+    {
+      "train_step": 2160,
+      "loss": 0.5971491708863612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5971491708863612
+    },
+    {
+      "train_step": 2161,
+      "loss": 0.6311461937281008,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6311461937281008
+    },
+    {
+      "train_step": 2162,
+      "loss": 0.5054732662060506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5054732662060506
+    },
+    {
+      "train_step": 2163,
+      "loss": 0.7376212704790788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7376212704790788
+    },
+    {
+      "train_step": 2164,
+      "loss": 0.6369758804877621,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6369758804877621
+    },
+    {
+      "train_step": 2165,
+      "loss": 0.663582901551129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.663582901551129
+    },
+    {
+      "train_step": 2166,
+      "loss": 0.6609644750622021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6609644750622021
+    },
+    {
+      "train_step": 2167,
+      "loss": 0.5320094079423765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5320094079423765
+    },
+    {
+      "train_step": 2168,
+      "loss": 0.5411635877759049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5411635877759049
+    },
+    {
+      "train_step": 2169,
+      "loss": 0.6433017345165337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6433017345165337
+    },
+    {
+      "train_step": 2170,
+      "loss": 0.4867251618785624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4867251618785624
+    },
+    {
+      "train_step": 2171,
+      "loss": 0.636636826718461,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.636636826718461
+    },
+    {
+      "train_step": 2172,
+      "loss": 0.5565507007814455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5565507007814455
+    },
+    {
+      "train_step": 2173,
+      "loss": 0.6376184402351271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6376184402351271
+    },
+    {
+      "train_step": 2174,
+      "loss": 0.6361632820656142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6361632820656142
+    },
+    {
+      "train_step": 2175,
+      "loss": 0.6292344511275076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6292344511275076
+    },
+    {
+      "train_step": 2176,
+      "loss": 0.5028424921035385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5028424921035385
+    },
+    {
+      "train_step": 2177,
+      "loss": 0.4170614744976736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4170614744976736
+    },
+    {
+      "train_step": 2178,
+      "loss": 0.417473390992831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.417473390992831
+    },
+    {
+      "train_step": 2179,
+      "loss": 0.7287024683574178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7287024683574178
+    },
+    {
+      "train_step": 2180,
+      "loss": 0.6365004291573438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6365004291573438
+    },
+    {
+      "train_step": 2181,
+      "loss": 0.5549751264310676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5549751264310676
+    },
+    {
+      "train_step": 2182,
+      "loss": 0.38832622969098884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38832622969098884
+    },
+    {
+      "train_step": 2183,
+      "loss": 0.5337613259312917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5337613259312917
+    },
+    {
+      "train_step": 2184,
+      "loss": 0.6120305686797172,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6120305686797172
+    },
+    {
+      "train_step": 2185,
+      "loss": 0.43424930933398936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43424930933398936
+    },
+    {
+      "train_step": 2186,
+      "loss": 0.6907138479451786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6907138479451786
+    },
+    {
+      "train_step": 2187,
+      "loss": 0.8795993179642179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8795993179642179
+    },
+    {
+      "train_step": 2188,
+      "loss": 0.6574232660619492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6574232660619492
+    },
+    {
+      "train_step": 2189,
+      "loss": 0.6941116547881749,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6941116547881749
+    },
+    {
+      "train_step": 2190,
+      "loss": 0.7814285283029981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7814285283029981
+    },
+    {
+      "train_step": 2191,
+      "loss": 0.7342823137273944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7342823137273944
+    },
+    {
+      "train_step": 2192,
+      "loss": 0.43629609918563633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43629609918563633
+    },
+    {
+      "train_step": 2193,
+      "loss": 0.5728935452439143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5728935452439143
+    },
+    {
+      "train_step": 2194,
+      "loss": 0.4014059938036899,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4014059938036899
+    },
+    {
+      "train_step": 2195,
+      "loss": 0.644630889288149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.644630889288149
+    },
+    {
+      "train_step": 2196,
+      "loss": 0.6503213284383209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6503213284383209
+    },
+    {
+      "train_step": 2197,
+      "loss": 0.47996440232992105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47996440232992105
+    },
+    {
+      "train_step": 2198,
+      "loss": 0.37371154139855706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37371154139855706
+    },
+    {
+      "train_step": 2199,
+      "loss": 0.42722493700813535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42722493700813535
+    },
+    {
+      "train_step": 2200,
+      "loss": 0.4616555243672788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4616555243672788
+    },
+    {
+      "train_step": 2201,
+      "loss": 0.6361513335551615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6361513335551615
+    },
+    {
+      "train_step": 2202,
+      "loss": 0.6418610153806918,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6418610153806918
+    },
+    {
+      "train_step": 2203,
+      "loss": 0.7203824087668317,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7203824087668317
+    },
+    {
+      "train_step": 2204,
+      "loss": 0.3861797012321393,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3861797012321393
+    },
+    {
+      "train_step": 2205,
+      "loss": 0.539505393370825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.539505393370825
+    },
+    {
+      "train_step": 2206,
+      "loss": 0.3662349151412369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3662349151412369
+    },
+    {
+      "train_step": 2207,
+      "loss": 0.5866291873027901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5866291873027901
+    },
+    {
+      "train_step": 2208,
+      "loss": 0.8385324429088039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8385324429088039
+    },
+    {
+      "train_step": 2209,
+      "loss": 0.45705297481452384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45705297481452384
+    },
+    {
+      "train_step": 2210,
+      "loss": 0.5453348354445319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5453348354445319
+    },
+    {
+      "train_step": 2211,
+      "loss": 0.5283962713071124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5283962713071124
+    },
+    {
+      "train_step": 2212,
+      "loss": 0.47500975652619076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47500975652619076
+    },
+    {
+      "train_step": 2213,
+      "loss": 0.5128875299120182,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5128875299120182
+    },
+    {
+      "train_step": 2214,
+      "loss": 0.7002252822016084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7002252822016084
+    },
+    {
+      "train_step": 2215,
+      "loss": 0.6450139339436406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6450139339436406
+    },
+    {
+      "train_step": 2216,
+      "loss": 0.8801472707386055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8801472707386055
+    },
+    {
+      "train_step": 2217,
+      "loss": 0.375161855923427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.375161855923427
+    },
+    {
+      "train_step": 2218,
+      "loss": 0.5317841910991886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5317841910991886
+    },
+    {
+      "train_step": 2219,
+      "loss": 0.6908986583161013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6908986583161013
+    },
+    {
+      "train_step": 2220,
+      "loss": 0.693075129837436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.693075129837436
+    },
+    {
+      "train_step": 2221,
+      "loss": 0.5425651244338897,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5425651244338897
+    },
+    {
+      "train_step": 2222,
+      "loss": 0.682779092281186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.682779092281186
+    },
+    {
+      "train_step": 2223,
+      "loss": 0.8704502114470714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8704502114470714
+    },
+    {
+      "train_step": 2224,
+      "loss": 0.5349157473371893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5349157473371893
+    },
+    {
+      "train_step": 2225,
+      "loss": 0.5608878333618686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5608878333618686
+    },
+    {
+      "train_step": 2226,
+      "loss": 0.6016795150566296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6016795150566296
+    },
+    {
+      "train_step": 2227,
+      "loss": 0.5745207862234376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5745207862234376
+    },
+    {
+      "train_step": 2228,
+      "loss": 0.5036576635718076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5036576635718076
+    },
+    {
+      "train_step": 2229,
+      "loss": 0.4049394964713904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4049394964713904
+    },
+    {
+      "train_step": 2230,
+      "loss": 0.4765088510702672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4765088510702672
+    },
+    {
+      "train_step": 2231,
+      "loss": 0.44777460211029696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44777460211029696
+    },
+    {
+      "train_step": 2232,
+      "loss": 0.6369357257999207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6369357257999207
+    },
+    {
+      "train_step": 2233,
+      "loss": 0.7283613480052031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7283613480052031
+    },
+    {
+      "train_step": 2234,
+      "loss": 0.5938428246741738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5938428246741738
+    },
+    {
+      "train_step": 2235,
+      "loss": 0.5154929072114514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5154929072114514
+    },
+    {
+      "train_step": 2236,
+      "loss": 0.576704215707895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576704215707895
+    },
+    {
+      "train_step": 2237,
+      "loss": 0.46795013912718736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46795013912718736
+    },
+    {
+      "train_step": 2238,
+      "loss": 0.38035087894795616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38035087894795616
+    },
+    {
+      "train_step": 2239,
+      "loss": 0.6078945892654314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6078945892654314
+    },
+    {
+      "train_step": 2240,
+      "loss": 0.4815450225360649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4815450225360649
+    },
+    {
+      "train_step": 2241,
+      "loss": 0.665604991045382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.665604991045382
+    },
+    {
+      "train_step": 2242,
+      "loss": 0.5093793327062313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5093793327062313
+    },
+    {
+      "train_step": 2243,
+      "loss": 0.549457783423969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.549457783423969
+    },
+    {
+      "train_step": 2244,
+      "loss": 0.6638253483482924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6638253483482924
+    },
+    {
+      "train_step": 2245,
+      "loss": 0.7306712564926449,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7306712564926449
+    },
+    {
+      "train_step": 2246,
+      "loss": 0.7727264890232306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7727264890232306
+    },
+    {
+      "train_step": 2247,
+      "loss": 0.5884079538117932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5884079538117932
+    },
+    {
+      "train_step": 2248,
+      "loss": 0.5281296942429387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5281296942429387
+    },
+    {
+      "train_step": 2249,
+      "loss": 0.4331602221457512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4331602221457512
+    },
+    {
+      "train_step": 2250,
+      "loss": 0.8194203097097853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8194203097097853
+    },
+    {
+      "train_step": 2251,
+      "loss": 0.5158776303005359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5158776303005359
+    },
+    {
+      "train_step": 2252,
+      "loss": 0.5717344206996254,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5717344206996254
+    },
+    {
+      "train_step": 2253,
+      "loss": 0.5161577483059329,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5161577483059329
+    },
+    {
+      "train_step": 2254,
+      "loss": 0.6326163507564831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6326163507564831
+    },
+    {
+      "train_step": 2255,
+      "loss": 0.5630541837335487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5630541837335487
+    },
+    {
+      "train_step": 2256,
+      "loss": 0.4444458439659036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4444458439659036
+    },
+    {
+      "train_step": 2257,
+      "loss": 0.5515556256161865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5515556256161865
+    },
+    {
+      "train_step": 2258,
+      "loss": 0.5167728212414007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5167728212414007
+    },
+    {
+      "train_step": 2259,
+      "loss": 0.5165602190302991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5165602190302991
+    },
+    {
+      "train_step": 2260,
+      "loss": 0.5142212718228797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5142212718228797
+    },
+    {
+      "train_step": 2261,
+      "loss": 0.35156762192015034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35156762192015034
+    },
+    {
+      "train_step": 2262,
+      "loss": 0.476016052638118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.476016052638118
+    },
+    {
+      "train_step": 2263,
+      "loss": 0.41660055156316883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41660055156316883
+    },
+    {
+      "train_step": 2264,
+      "loss": 0.6621315746315684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6621315746315684
+    },
+    {
+      "train_step": 2265,
+      "loss": 0.5074353298894205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5074353298894205
+    },
+    {
+      "train_step": 2266,
+      "loss": 0.3632748422079317,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3632748422079317
+    },
+    {
+      "train_step": 2267,
+      "loss": 0.44121027248331035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44121027248331035
+    },
+    {
+      "train_step": 2268,
+      "loss": 0.5830158219628946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5830158219628946
+    },
+    {
+      "train_step": 2269,
+      "loss": 0.5441144680590473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441144680590473
+    },
+    {
+      "train_step": 2270,
+      "loss": 0.6326728604233605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6326728604233605
+    },
+    {
+      "train_step": 2271,
+      "loss": 0.5826948059372263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5826948059372263
+    },
+    {
+      "train_step": 2272,
+      "loss": 0.5466828986842658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5466828986842658
+    },
+    {
+      "train_step": 2273,
+      "loss": 0.806435309034245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.806435309034245
+    },
+    {
+      "train_step": 2274,
+      "loss": 0.5079190187603673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5079190187603673
+    },
+    {
+      "train_step": 2275,
+      "loss": 0.45379451655850933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45379451655850933
+    },
+    {
+      "train_step": 2276,
+      "loss": 0.5807253869677294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5807253869677294
+    },
+    {
+      "train_step": 2277,
+      "loss": 0.5933769173863193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5933769173863193
+    },
+    {
+      "train_step": 2278,
+      "loss": 0.6594805994478431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6594805994478431
+    },
+    {
+      "train_step": 2279,
+      "loss": 0.7731415612872726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7731415612872726
+    },
+    {
+      "train_step": 2280,
+      "loss": 0.5556580253517729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5556580253517729
+    },
+    {
+      "train_step": 2281,
+      "loss": 0.41185287003275256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41185287003275256
+    },
+    {
+      "train_step": 2282,
+      "loss": 0.777703112367698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.777703112367698
+    },
+    {
+      "train_step": 2283,
+      "loss": 0.45781411718196663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45781411718196663
+    },
+    {
+      "train_step": 2284,
+      "loss": 0.4823282479878682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4823282479878682
+    },
+    {
+      "train_step": 2285,
+      "loss": 0.4989320484729711,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4989320484729711
+    },
+    {
+      "train_step": 2286,
+      "loss": 0.5982477672830032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5982477672830032
+    },
+    {
+      "train_step": 2287,
+      "loss": 0.5530807620928978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5530807620928978
+    },
+    {
+      "train_step": 2288,
+      "loss": 0.5479558421665639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5479558421665639
+    },
+    {
+      "train_step": 2289,
+      "loss": 0.569589612288162,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.569589612288162
+    },
+    {
+      "train_step": 2290,
+      "loss": 0.63126858931252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.63126858931252
+    },
+    {
+      "train_step": 2291,
+      "loss": 0.44326690398920016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44326690398920016
+    },
+    {
+      "train_step": 2292,
+      "loss": 0.7130109930838777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7130109930838777
+    },
+    {
+      "train_step": 2293,
+      "loss": 0.6735345403292238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6735345403292238
+    },
+    {
+      "train_step": 2294,
+      "loss": 0.4665482347671874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4665482347671874
+    },
+    {
+      "train_step": 2295,
+      "loss": 0.3471375082818699,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3471375082818699
+    },
+    {
+      "train_step": 2296,
+      "loss": 0.5597735349773375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5597735349773375
+    },
+    {
+      "train_step": 2297,
+      "loss": 0.6626938083687385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6626938083687385
+    },
+    {
+      "train_step": 2298,
+      "loss": 0.5509313200570579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5509313200570579
+    },
+    {
+      "train_step": 2299,
+      "loss": 0.3947299183010759,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3947299183010759
+    },
+    {
+      "train_step": 2300,
+      "loss": 0.5801566763884622,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5801566763884622
+    },
+    {
+      "train_step": 2301,
+      "loss": 0.6118244685549612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6118244685549612
+    },
+    {
+      "train_step": 2302,
+      "loss": 0.701460164357687,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.701460164357687
+    },
+    {
+      "train_step": 2303,
+      "loss": 0.5282094825061766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5282094825061766
+    },
+    {
+      "train_step": 2304,
+      "loss": 0.5490678563910691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5490678563910691
+    },
+    {
+      "train_step": 2305,
+      "loss": 0.5662650820127569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5662650820127569
+    },
+    {
+      "train_step": 2306,
+      "loss": 0.40363012194446385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40363012194446385
+    },
+    {
+      "train_step": 2307,
+      "loss": 0.5815974345368596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5815974345368596
+    },
+    {
+      "train_step": 2308,
+      "loss": 0.5750583486657982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5750583486657982
+    },
+    {
+      "train_step": 2309,
+      "loss": 0.46481729169008723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46481729169008723
+    },
+    {
+      "train_step": 2310,
+      "loss": 0.36241790481756764,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36241790481756764
+    },
+    {
+      "train_step": 2311,
+      "loss": 0.7266036455643325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7266036455643325
+    },
+    {
+      "train_step": 2312,
+      "loss": 0.4936786285846008,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4936786285846008
+    },
+    {
+      "train_step": 2313,
+      "loss": 0.5063198511799109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5063198511799109
+    },
+    {
+      "train_step": 2314,
+      "loss": 0.7700095570780581,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7700095570780581
+    },
+    {
+      "train_step": 2315,
+      "loss": 0.6694761508511398,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6694761508511398
+    },
+    {
+      "train_step": 2316,
+      "loss": 0.7453664159957677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7453664159957677
+    },
+    {
+      "train_step": 2317,
+      "loss": 0.5551839522947783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5551839522947783
+    },
+    {
+      "train_step": 2318,
+      "loss": 0.7066503248876688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7066503248876688
+    },
+    {
+      "train_step": 2319,
+      "loss": 0.5095998230131956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5095998230131956
+    },
+    {
+      "train_step": 2320,
+      "loss": 0.5746383160456993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5746383160456993
+    },
+    {
+      "train_step": 2321,
+      "loss": 0.6114113328361199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6114113328361199
+    },
+    {
+      "train_step": 2322,
+      "loss": 0.4637827143751869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4637827143751869
+    },
+    {
+      "train_step": 2323,
+      "loss": 0.646525916139365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.646525916139365
+    },
+    {
+      "train_step": 2324,
+      "loss": 0.5974143213866151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5974143213866151
+    },
+    {
+      "train_step": 2325,
+      "loss": 0.5151256987211059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5151256987211059
+    },
+    {
+      "train_step": 2326,
+      "loss": 0.45411939058026557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45411939058026557
+    },
+    {
+      "train_step": 2327,
+      "loss": 0.6497788993985513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6497788993985513
+    },
+    {
+      "train_step": 2328,
+      "loss": 0.37577367282986146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37577367282986146
+    },
+    {
+      "train_step": 2329,
+      "loss": 0.6669458780199287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6669458780199287
+    },
+    {
+      "train_step": 2330,
+      "loss": 0.5543412202076705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5543412202076705
+    },
+    {
+      "train_step": 2331,
+      "loss": 0.645139444728156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.645139444728156
+    },
+    {
+      "train_step": 2332,
+      "loss": 0.4431295675656408,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4431295675656408
+    },
+    {
+      "train_step": 2333,
+      "loss": 0.7386007444774216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7386007444774216
+    },
+    {
+      "train_step": 2334,
+      "loss": 0.45968051900237744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45968051900237744
+    },
+    {
+      "train_step": 2335,
+      "loss": 0.7108190838502536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7108190838502536
+    },
+    {
+      "train_step": 2336,
+      "loss": 0.5454733379370597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5454733379370597
+    },
+    {
+      "train_step": 2337,
+      "loss": 0.504858434846485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.504858434846485
+    },
+    {
+      "train_step": 2338,
+      "loss": 0.6768533400708167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6768533400708167
+    },
+    {
+      "train_step": 2339,
+      "loss": 0.4623485519127097,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4623485519127097
+    },
+    {
+      "train_step": 2340,
+      "loss": 0.4077765045705198,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4077765045705198
+    },
+    {
+      "train_step": 2341,
+      "loss": 0.5468502749378741,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5468502749378741
+    },
+    {
+      "train_step": 2342,
+      "loss": 0.49476133372507386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49476133372507386
+    },
+    {
+      "train_step": 2343,
+      "loss": 0.47333579910707313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47333579910707313
+    },
+    {
+      "train_step": 2344,
+      "loss": 0.6272950174560863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6272950174560863
+    },
+    {
+      "train_step": 2345,
+      "loss": 0.48932129513432493,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48932129513432493
+    },
+    {
+      "train_step": 2346,
+      "loss": 0.6110601302068852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6110601302068852
+    },
+    {
+      "train_step": 2347,
+      "loss": 0.7305503941605876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7305503941605876
+    },
+    {
+      "train_step": 2348,
+      "loss": 0.2947862928414267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2947862928414267
+    },
+    {
+      "train_step": 2349,
+      "loss": 0.6934499862790158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6934499862790158
+    },
+    {
+      "train_step": 2350,
+      "loss": 0.39679787306669834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39679787306669834
+    },
+    {
+      "train_step": 2351,
+      "loss": 0.5035116458058276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035116458058276
+    },
+    {
+      "train_step": 2352,
+      "loss": 0.816484878003787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.816484878003787
+    },
+    {
+      "train_step": 2353,
+      "loss": 0.5951179780728405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5951179780728405
+    },
+    {
+      "train_step": 2354,
+      "loss": 0.7024380562995519,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7024380562995519
+    },
+    {
+      "train_step": 2355,
+      "loss": 0.5589396853674217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5589396853674217
+    },
+    {
+      "train_step": 2356,
+      "loss": 0.664173043405802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.664173043405802
+    },
+    {
+      "train_step": 2357,
+      "loss": 0.6615126153428701,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6615126153428701
+    },
+    {
+      "train_step": 2358,
+      "loss": 0.6720331236910858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6720331236910858
+    },
+    {
+      "train_step": 2359,
+      "loss": 0.4870080662482907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4870080662482907
+    },
+    {
+      "train_step": 2360,
+      "loss": 0.3969253132478915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3969253132478915
+    },
+    {
+      "train_step": 2361,
+      "loss": 0.5340227632404034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5340227632404034
+    },
+    {
+      "train_step": 2362,
+      "loss": 0.8521441581900491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8521441581900491
+    },
+    {
+      "train_step": 2363,
+      "loss": 0.5132347683970925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5132347683970925
+    },
+    {
+      "train_step": 2364,
+      "loss": 0.616527733560885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.616527733560885
+    },
+    {
+      "train_step": 2365,
+      "loss": 0.45998527604656975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45998527604656975
+    },
+    {
+      "train_step": 2366,
+      "loss": 0.49639012959611356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49639012959611356
+    },
+    {
+      "train_step": 2367,
+      "loss": 0.6014810938960504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6014810938960504
+    },
+    {
+      "train_step": 2368,
+      "loss": 0.5941480317506111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5941480317506111
+    },
+    {
+      "train_step": 2369,
+      "loss": 0.5542470874025784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5542470874025784
+    },
+    {
+      "train_step": 2370,
+      "loss": 0.5984825748756551,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5984825748756551
+    },
+    {
+      "train_step": 2371,
+      "loss": 0.6031941861128689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6031941861128689
+    },
+    {
+      "train_step": 2372,
+      "loss": 0.4942506314259419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4942506314259419
+    },
+    {
+      "train_step": 2373,
+      "loss": 0.4449631604765749,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4449631604765749
+    },
+    {
+      "train_step": 2374,
+      "loss": 0.5684811995150888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5684811995150888
+    },
+    {
+      "train_step": 2375,
+      "loss": 0.8924796175567834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8924796175567834
+    },
+    {
+      "train_step": 2376,
+      "loss": 0.5382413648784087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5382413648784087
+    },
+    {
+      "train_step": 2377,
+      "loss": 0.8414686297927447,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8414686297927447
+    },
+    {
+      "train_step": 2378,
+      "loss": 0.45766547261575596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45766547261575596
+    },
+    {
+      "train_step": 2379,
+      "loss": 0.5893183201025319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5893183201025319
+    },
+    {
+      "train_step": 2380,
+      "loss": 0.3430551281809468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3430551281809468
+    },
+    {
+      "train_step": 2381,
+      "loss": 0.5080284766582882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5080284766582882
+    },
+    {
+      "train_step": 2382,
+      "loss": 0.457714423362822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.457714423362822
+    },
+    {
+      "train_step": 2383,
+      "loss": 0.5348513770289964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5348513770289964
+    },
+    {
+      "train_step": 2384,
+      "loss": 0.419756485952549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.419756485952549
+    },
+    {
+      "train_step": 2385,
+      "loss": 0.6025300294174138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6025300294174138
+    },
+    {
+      "train_step": 2386,
+      "loss": 0.7428479527980919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7428479527980919
+    },
+    {
+      "train_step": 2387,
+      "loss": 0.624766796191008,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.624766796191008
+    },
+    {
+      "train_step": 2388,
+      "loss": 0.8157846571539707,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8157846571539707
+    },
+    {
+      "train_step": 2389,
+      "loss": 0.4276665267150011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4276665267150011
+    },
+    {
+      "train_step": 2390,
+      "loss": 0.7116317018264262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7116317018264262
+    },
+    {
+      "train_step": 2391,
+      "loss": 0.7924637316073656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7924637316073656
+    },
+    {
+      "train_step": 2392,
+      "loss": 0.4709131908287698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4709131908287698
+    },
+    {
+      "train_step": 2393,
+      "loss": 0.7143935184553789,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7143935184553789
+    },
+    {
+      "train_step": 2394,
+      "loss": 0.6574573563994425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6574573563994425
+    },
+    {
+      "train_step": 2395,
+      "loss": 0.671429843211854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.671429843211854
+    },
+    {
+      "train_step": 2396,
+      "loss": 0.6431804711239455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6431804711239455
+    },
+    {
+      "train_step": 2397,
+      "loss": 0.5685479011250718,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5685479011250718
+    },
+    {
+      "train_step": 2398,
+      "loss": 0.664373134671682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.664373134671682
+    },
+    {
+      "train_step": 2399,
+      "loss": 0.6780727585338107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6780727585338107
+    },
+    {
+      "train_step": 2400,
+      "loss": 0.590821192925762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.590821192925762
+    },
+    {
+      "train_step": 2401,
+      "loss": 0.548620843514186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.548620843514186
+    },
+    {
+      "train_step": 2402,
+      "loss": 0.6351425356470211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6351425356470211
+    },
+    {
+      "train_step": 2403,
+      "loss": 0.708650834418768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.708650834418768
+    },
+    {
+      "train_step": 2404,
+      "loss": 0.7016636575950864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7016636575950864
+    },
+    {
+      "train_step": 2405,
+      "loss": 0.7657994612416946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7657994612416946
+    },
+    {
+      "train_step": 2406,
+      "loss": 0.5453589045453837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5453589045453837
+    },
+    {
+      "train_step": 2407,
+      "loss": 0.7206688975672002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7206688975672002
+    },
+    {
+      "train_step": 2408,
+      "loss": 0.4726754565129119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4726754565129119
+    },
+    {
+      "train_step": 2409,
+      "loss": 0.6004899814589887,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6004899814589887
+    },
+    {
+      "train_step": 2410,
+      "loss": 0.4986882270329574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4986882270329574
+    },
+    {
+      "train_step": 2411,
+      "loss": 0.7720854022930898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7720854022930898
+    },
+    {
+      "train_step": 2412,
+      "loss": 0.5121607485016872,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5121607485016872
+    },
+    {
+      "train_step": 2413,
+      "loss": 0.40241012409325394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40241012409325394
+    },
+    {
+      "train_step": 2414,
+      "loss": 0.6262459110645445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6262459110645445
+    },
+    {
+      "train_step": 2415,
+      "loss": 0.7619970373061721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7619970373061721
+    },
+    {
+      "train_step": 2416,
+      "loss": 0.5779401772332897,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5779401772332897
+    },
+    {
+      "train_step": 2417,
+      "loss": 0.42077751625050797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42077751625050797
+    },
+    {
+      "train_step": 2418,
+      "loss": 0.5681037535731467,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5681037535731467
+    },
+    {
+      "train_step": 2419,
+      "loss": 0.6875362856002397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6875362856002397
+    },
+    {
+      "train_step": 2420,
+      "loss": 0.42472471762671704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42472471762671704
+    },
+    {
+      "train_step": 2421,
+      "loss": 0.5406944808959824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5406944808959824
+    },
+    {
+      "train_step": 2422,
+      "loss": 0.7540266132621792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7540266132621792
+    },
+    {
+      "train_step": 2423,
+      "loss": 0.5613207803014113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5613207803014113
+    },
+    {
+      "train_step": 2424,
+      "loss": 0.7572471357497219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7572471357497219
+    },
+    {
+      "train_step": 2425,
+      "loss": 0.5586556087380201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5586556087380201
+    },
+    {
+      "train_step": 2426,
+      "loss": 0.5433584463765013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5433584463765013
+    },
+    {
+      "train_step": 2427,
+      "loss": 0.46051955289870083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46051955289870083
+    },
+    {
+      "train_step": 2428,
+      "loss": 0.7435624273646422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7435624273646422
+    },
+    {
+      "train_step": 2429,
+      "loss": 0.3362441753150501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3362441753150501
+    },
+    {
+      "train_step": 2430,
+      "loss": 0.5546309657458327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5546309657458327
+    },
+    {
+      "train_step": 2431,
+      "loss": 0.5050534079980828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5050534079980828
+    },
+    {
+      "train_step": 2432,
+      "loss": 0.49313862621448756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49313862621448756
+    },
+    {
+      "train_step": 2433,
+      "loss": 0.7137802663191085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7137802663191085
+    },
+    {
+      "train_step": 2434,
+      "loss": 0.6951498961490111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6951498961490111
+    },
+    {
+      "train_step": 2435,
+      "loss": 0.5737802017647364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5737802017647364
+    },
+    {
+      "train_step": 2436,
+      "loss": 0.6404435487785043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6404435487785043
+    },
+    {
+      "train_step": 2437,
+      "loss": 0.5820447963372586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5820447963372586
+    },
+    {
+      "train_step": 2438,
+      "loss": 0.7635429122274403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7635429122274403
+    },
+    {
+      "train_step": 2439,
+      "loss": 0.6714229709637353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6714229709637353
+    },
+    {
+      "train_step": 2440,
+      "loss": 0.5579532906893682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5579532906893682
+    },
+    {
+      "train_step": 2441,
+      "loss": 0.6957855639703958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6957855639703958
+    },
+    {
+      "train_step": 2442,
+      "loss": 0.5800680889510075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5800680889510075
+    },
+    {
+      "train_step": 2443,
+      "loss": 0.4168143128171953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4168143128171953
+    },
+    {
+      "train_step": 2444,
+      "loss": 0.6298674760264241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6298674760264241
+    },
+    {
+      "train_step": 2445,
+      "loss": 0.5517669515528705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5517669515528705
+    },
+    {
+      "train_step": 2446,
+      "loss": 0.5034000600189127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5034000600189127
+    },
+    {
+      "train_step": 2447,
+      "loss": 0.6256678231225447,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6256678231225447
+    },
+    {
+      "train_step": 2448,
+      "loss": 0.762762761083301,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.762762761083301
+    },
+    {
+      "train_step": 2449,
+      "loss": 0.42814229484494615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42814229484494615
+    },
+    {
+      "train_step": 2450,
+      "loss": 0.41753377591720675,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41753377591720675
+    },
+    {
+      "train_step": 2451,
+      "loss": 0.8600797650256149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8600797650256149
+    },
+    {
+      "train_step": 2452,
+      "loss": 0.6720021932880726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6720021932880726
+    },
+    {
+      "train_step": 2453,
+      "loss": 0.5440660077227938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5440660077227938
+    },
+    {
+      "train_step": 2454,
+      "loss": 0.2975126612753756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2975126612753756
+    },
+    {
+      "train_step": 2455,
+      "loss": 0.6127439799440001,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6127439799440001
+    },
+    {
+      "train_step": 2456,
+      "loss": 0.4214693360557349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4214693360557349
+    },
+    {
+      "train_step": 2457,
+      "loss": 0.5296474998985348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5296474998985348
+    },
+    {
+      "train_step": 2458,
+      "loss": 0.8074598821130675,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8074598821130675
+    },
+    {
+      "train_step": 2459,
+      "loss": 0.6720201876580255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6720201876580255
+    },
+    {
+      "train_step": 2460,
+      "loss": 0.49252563011436956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49252563011436956
+    },
+    {
+      "train_step": 2461,
+      "loss": 0.6650386588421143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6650386588421143
+    },
+    {
+      "train_step": 2462,
+      "loss": 0.596553759551643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.596553759551643
+    },
+    {
+      "train_step": 2463,
+      "loss": 0.6230389276119744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6230389276119744
+    },
+    {
+      "train_step": 2464,
+      "loss": 0.48739534456738315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48739534456738315
+    },
+    {
+      "train_step": 2465,
+      "loss": 0.49171193454181894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49171193454181894
+    },
+    {
+      "train_step": 2466,
+      "loss": 0.4489590875728144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4489590875728144
+    },
+    {
+      "train_step": 2467,
+      "loss": 0.5187356706418957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5187356706418957
+    },
+    {
+      "train_step": 2468,
+      "loss": 0.6195014569681088,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6195014569681088
+    },
+    {
+      "train_step": 2469,
+      "loss": 0.48371510224706915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48371510224706915
+    },
+    {
+      "train_step": 2470,
+      "loss": 0.5012073359374802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5012073359374802
+    },
+    {
+      "train_step": 2471,
+      "loss": 0.7723351360185217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7723351360185217
+    },
+    {
+      "train_step": 2472,
+      "loss": 0.7406993949623221,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7406993949623221
+    },
+    {
+      "train_step": 2473,
+      "loss": 0.6556222637151923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6556222637151923
+    },
+    {
+      "train_step": 2474,
+      "loss": 0.6848671074811057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6848671074811057
+    },
+    {
+      "train_step": 2475,
+      "loss": 0.5938009219012028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5938009219012028
+    },
+    {
+      "train_step": 2476,
+      "loss": 0.7360472002934232,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7360472002934232
+    },
+    {
+      "train_step": 2477,
+      "loss": 0.47795832892756557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47795832892756557
+    },
+    {
+      "train_step": 2478,
+      "loss": 0.7202627706547486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7202627706547486
+    },
+    {
+      "train_step": 2479,
+      "loss": 0.5607053091665187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5607053091665187
+    },
+    {
+      "train_step": 2480,
+      "loss": 0.5858412172590828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5858412172590828
+    },
+    {
+      "train_step": 2481,
+      "loss": 0.36257433550944485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36257433550944485
+    },
+    {
+      "train_step": 2482,
+      "loss": 0.7853592510935432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7853592510935432
+    },
+    {
+      "train_step": 2483,
+      "loss": 0.4127009990528036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4127009990528036
+    },
+    {
+      "train_step": 2484,
+      "loss": 0.3816366895209762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3816366895209762
+    },
+    {
+      "train_step": 2485,
+      "loss": 0.4171177752641386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4171177752641386
+    },
+    {
+      "train_step": 2486,
+      "loss": 0.5378529369498237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5378529369498237
+    },
+    {
+      "train_step": 2487,
+      "loss": 0.6286453735296574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6286453735296574
+    },
+    {
+      "train_step": 2488,
+      "loss": 0.7362004458653959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7362004458653959
+    },
+    {
+      "train_step": 2489,
+      "loss": 0.6770501755066134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6770501755066134
+    },
+    {
+      "train_step": 2490,
+      "loss": 0.6793872560309872,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6793872560309872
+    },
+    {
+      "train_step": 2491,
+      "loss": 0.5147353301159272,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5147353301159272
+    },
+    {
+      "train_step": 2492,
+      "loss": 0.32176936825115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32176936825115
+    },
+    {
+      "train_step": 2493,
+      "loss": 0.8199482121112672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8199482121112672
+    },
+    {
+      "train_step": 2494,
+      "loss": 0.5435648745829752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5435648745829752
+    },
+    {
+      "train_step": 2495,
+      "loss": 0.6067208792165946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6067208792165946
+    },
+    {
+      "train_step": 2496,
+      "loss": 0.32111864601683204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32111864601683204
+    },
+    {
+      "train_step": 2497,
+      "loss": 0.5458690172786866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5458690172786866
+    },
+    {
+      "train_step": 2498,
+      "loss": 0.8132398152525822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8132398152525822
+    },
+    {
+      "train_step": 2499,
+      "loss": 0.7694052765162344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7694052765162344
+    },
+    {
+      "train_step": 2500,
+      "loss": 0.5610369579541166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5610369579541166
+    },
+    {
+      "train_step": 2501,
+      "loss": 0.5005865069708858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5005865069708858
+    },
+    {
+      "train_step": 2502,
+      "loss": 0.5382307114829573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5382307114829573
+    },
+    {
+      "train_step": 2503,
+      "loss": 0.7202512654198906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7202512654198906
+    },
+    {
+      "train_step": 2504,
+      "loss": 0.5456001300370396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5456001300370396
+    },
+    {
+      "train_step": 2505,
+      "loss": 0.6573811247422122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6573811247422122
+    },
+    {
+      "train_step": 2506,
+      "loss": 0.6291296953849989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6291296953849989
+    },
+    {
+      "train_step": 2507,
+      "loss": 0.6114891508308041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6114891508308041
+    },
+    {
+      "train_step": 2508,
+      "loss": 0.6551974560495669,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6551974560495669
+    },
+    {
+      "train_step": 2509,
+      "loss": 0.5054549838384513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5054549838384513
+    },
+    {
+      "train_step": 2510,
+      "loss": 0.6474604544411051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6474604544411051
+    },
+    {
+      "train_step": 2511,
+      "loss": 0.6136719073414678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6136719073414678
+    },
+    {
+      "train_step": 2512,
+      "loss": 0.5658943460293946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5658943460293946
+    },
+    {
+      "train_step": 2513,
+      "loss": 0.5986224959554892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5986224959554892
+    },
+    {
+      "train_step": 2514,
+      "loss": 0.7752862042846271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7752862042846271
+    },
+    {
+      "train_step": 2515,
+      "loss": 0.4803943595280079,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4803943595280079
+    },
+    {
+      "train_step": 2516,
+      "loss": 0.8697684250158741,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8697684250158741
+    },
+    {
+      "train_step": 2517,
+      "loss": 0.8706106376984917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8706106376984917
+    },
+    {
+      "train_step": 2518,
+      "loss": 0.5897631592753955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5897631592753955
+    },
+    {
+      "train_step": 2519,
+      "loss": 0.6638966572044215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6638966572044215
+    },
+    {
+      "train_step": 2520,
+      "loss": 0.6117005424869629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6117005424869629
+    },
+    {
+      "train_step": 2521,
+      "loss": 0.5043224316429851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5043224316429851
+    },
+    {
+      "train_step": 2522,
+      "loss": 0.7830032535835914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7830032535835914
+    },
+    {
+      "train_step": 2523,
+      "loss": 0.6859216141151263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6859216141151263
+    },
+    {
+      "train_step": 2524,
+      "loss": 0.5309113780937836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5309113780937836
+    },
+    {
+      "train_step": 2525,
+      "loss": 0.5561375598096407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5561375598096407
+    },
+    {
+      "train_step": 2526,
+      "loss": 0.6531037269154225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6531037269154225
+    },
+    {
+      "train_step": 2527,
+      "loss": 0.6323615829790853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6323615829790853
+    },
+    {
+      "train_step": 2528,
+      "loss": 0.499318729617843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.499318729617843
+    },
+    {
+      "train_step": 2529,
+      "loss": 0.724539989152343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.724539989152343
+    },
+    {
+      "train_step": 2530,
+      "loss": 0.2996947855200717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2996947855200717
+    },
+    {
+      "train_step": 2531,
+      "loss": 0.5669670185839868,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5669670185839868
+    },
+    {
+      "train_step": 2532,
+      "loss": 0.7878361946081114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7878361946081114
+    },
+    {
+      "train_step": 2533,
+      "loss": 0.5296053087209475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5296053087209475
+    },
+    {
+      "train_step": 2534,
+      "loss": 0.7327819903737885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7327819903737885
+    },
+    {
+      "train_step": 2535,
+      "loss": 0.4145736225779576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4145736225779576
+    },
+    {
+      "train_step": 2536,
+      "loss": 0.7936268047539754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7936268047539754
+    },
+    {
+      "train_step": 2537,
+      "loss": 0.596666230472726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.596666230472726
+    },
+    {
+      "train_step": 2538,
+      "loss": 0.5451486009603249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5451486009603249
+    },
+    {
+      "train_step": 2539,
+      "loss": 0.4592827685989619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4592827685989619
+    },
+    {
+      "train_step": 2540,
+      "loss": 0.5429147602268632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5429147602268632
+    },
+    {
+      "train_step": 2541,
+      "loss": 0.7723771635062333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7723771635062333
+    },
+    {
+      "train_step": 2542,
+      "loss": 0.4544674953736055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4544674953736055
+    },
+    {
+      "train_step": 2543,
+      "loss": 0.6894955037570811,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6894955037570811
+    },
+    {
+      "train_step": 2544,
+      "loss": 0.6419554022427186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6419554022427186
+    },
+    {
+      "train_step": 2545,
+      "loss": 0.71112418645196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.71112418645196
+    },
+    {
+      "train_step": 2546,
+      "loss": 0.576026544184308,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576026544184308
+    },
+    {
+      "train_step": 2547,
+      "loss": 0.819451056123954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.819451056123954
+    },
+    {
+      "train_step": 2548,
+      "loss": 0.7054720948285288,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7054720948285288
+    },
+    {
+      "train_step": 2549,
+      "loss": 0.6371175896024901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6371175896024901
+    },
+    {
+      "train_step": 2550,
+      "loss": 0.46247282898522135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46247282898522135
+    },
+    {
+      "train_step": 2551,
+      "loss": 0.663171574370645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.663171574370645
+    },
+    {
+      "train_step": 2552,
+      "loss": 0.5444047524498496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5444047524498496
+    },
+    {
+      "train_step": 2553,
+      "loss": 0.5295162247529358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5295162247529358
+    },
+    {
+      "train_step": 2554,
+      "loss": 0.753076161197572,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.753076161197572
+    },
+    {
+      "train_step": 2555,
+      "loss": 0.6116066594056995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6116066594056995
+    },
+    {
+      "train_step": 2556,
+      "loss": 0.4909169038264972,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4909169038264972
+    },
+    {
+      "train_step": 2557,
+      "loss": 0.5461975183833678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5461975183833678
+    },
+    {
+      "train_step": 2558,
+      "loss": 0.539098358689443,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.539098358689443
+    },
+    {
+      "train_step": 2559,
+      "loss": 0.747314658232311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.747314658232311
+    },
+    {
+      "train_step": 2560,
+      "loss": 0.642716379673303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.642716379673303
+    },
+    {
+      "train_step": 2561,
+      "loss": 0.43199960519300085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43199960519300085
+    },
+    {
+      "train_step": 2562,
+      "loss": 0.6959880713985146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6959880713985146
+    },
+    {
+      "train_step": 2563,
+      "loss": 0.5932357472788752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5932357472788752
+    },
+    {
+      "train_step": 2564,
+      "loss": 0.5975176210946505,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5975176210946505
+    },
+    {
+      "train_step": 2565,
+      "loss": 0.5551406258543337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5551406258543337
+    },
+    {
+      "train_step": 2566,
+      "loss": 0.5468560026626998,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5468560026626998
+    },
+    {
+      "train_step": 2567,
+      "loss": 0.48471890238604315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48471890238604315
+    },
+    {
+      "train_step": 2568,
+      "loss": 0.6005232949526847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6005232949526847
+    },
+    {
+      "train_step": 2569,
+      "loss": 0.5506370082794251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5506370082794251
+    },
+    {
+      "train_step": 2570,
+      "loss": 0.4842519224824985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4842519224824985
+    },
+    {
+      "train_step": 2571,
+      "loss": 0.5930469762738753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5930469762738753
+    },
+    {
+      "train_step": 2572,
+      "loss": 0.6838013610178274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6838013610178274
+    },
+    {
+      "train_step": 2573,
+      "loss": 0.7272938976356362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7272938976356362
+    },
+    {
+      "train_step": 2574,
+      "loss": 0.6236741005976756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6236741005976756
+    },
+    {
+      "train_step": 2575,
+      "loss": 0.5808959047183198,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5808959047183198
+    },
+    {
+      "train_step": 2576,
+      "loss": 0.5283662146542814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5283662146542814
+    },
+    {
+      "train_step": 2577,
+      "loss": 0.7238014986681301,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7238014986681301
+    },
+    {
+      "train_step": 2578,
+      "loss": 0.5324821813167759,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5324821813167759
+    },
+    {
+      "train_step": 2579,
+      "loss": 0.5636630889333653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5636630889333653
+    },
+    {
+      "train_step": 2580,
+      "loss": 0.7069300053064795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7069300053064795
+    },
+    {
+      "train_step": 2581,
+      "loss": 0.5594816383191392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5594816383191392
+    },
+    {
+      "train_step": 2582,
+      "loss": 0.6700573749146028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6700573749146028
+    },
+    {
+      "train_step": 2583,
+      "loss": 0.5618874243618426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5618874243618426
+    },
+    {
+      "train_step": 2584,
+      "loss": 0.8114103526681414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8114103526681414
+    },
+    {
+      "train_step": 2585,
+      "loss": 0.6090244046750084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6090244046750084
+    },
+    {
+      "train_step": 2586,
+      "loss": 0.44427516214608154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44427516214608154
+    },
+    {
+      "train_step": 2587,
+      "loss": 0.6308858089966567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6308858089966567
+    },
+    {
+      "train_step": 2588,
+      "loss": 0.4341104341511481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4341104341511481
+    },
+    {
+      "train_step": 2589,
+      "loss": 0.505676817216385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.505676817216385
+    },
+    {
+      "train_step": 2590,
+      "loss": 0.5017523022228778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5017523022228778
+    },
+    {
+      "train_step": 2591,
+      "loss": 0.33441703557830554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33441703557830554
+    },
+    {
+      "train_step": 2592,
+      "loss": 0.44337056908193256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44337056908193256
+    },
+    {
+      "train_step": 2593,
+      "loss": 0.5671597407519152,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5671597407519152
+    },
+    {
+      "train_step": 2594,
+      "loss": 0.41435307200365895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41435307200365895
+    },
+    {
+      "train_step": 2595,
+      "loss": 0.8076381361796185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8076381361796185
+    },
+    {
+      "train_step": 2596,
+      "loss": 0.7181994197827397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7181994197827397
+    },
+    {
+      "train_step": 2597,
+      "loss": 0.46313284521394543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46313284521394543
+    },
+    {
+      "train_step": 2598,
+      "loss": 0.40774532244903816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40774532244903816
+    },
+    {
+      "train_step": 2599,
+      "loss": 0.5855054466725719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5855054466725719
+    },
+    {
+      "train_step": 2600,
+      "loss": 0.39851934164897435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39851934164897435
+    },
+    {
+      "train_step": 2601,
+      "loss": 0.8180436875753319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8180436875753319
+    },
+    {
+      "train_step": 2602,
+      "loss": 0.5740100773061791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5740100773061791
+    },
+    {
+      "train_step": 2603,
+      "loss": 0.4143147394404319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4143147394404319
+    },
+    {
+      "train_step": 2604,
+      "loss": 0.5240897092995974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5240897092995974
+    },
+    {
+      "train_step": 2605,
+      "loss": 0.6091025001740745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6091025001740745
+    },
+    {
+      "train_step": 2606,
+      "loss": 0.5922613572325363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5922613572325363
+    },
+    {
+      "train_step": 2607,
+      "loss": 0.7741337336110329,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7741337336110329
+    },
+    {
+      "train_step": 2608,
+      "loss": 0.7970481460246539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7970481460246539
+    },
+    {
+      "train_step": 2609,
+      "loss": 0.48042511917891007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48042511917891007
+    },
+    {
+      "train_step": 2610,
+      "loss": 0.8467019054446574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8467019054446574
+    },
+    {
+      "train_step": 2611,
+      "loss": 0.38688114530307693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38688114530307693
+    },
+    {
+      "train_step": 2612,
+      "loss": 0.537969765325381,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.537969765325381
+    },
+    {
+      "train_step": 2613,
+      "loss": 0.34322744597352195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34322744597352195
+    },
+    {
+      "train_step": 2614,
+      "loss": 0.5028701276314747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5028701276314747
+    },
+    {
+      "train_step": 2615,
+      "loss": 0.7397011840168366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7397011840168366
+    },
+    {
+      "train_step": 2616,
+      "loss": 0.5269355616653499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5269355616653499
+    },
+    {
+      "train_step": 2617,
+      "loss": 0.7070617338762784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7070617338762784
+    },
+    {
+      "train_step": 2618,
+      "loss": 0.40572046477020735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40572046477020735
+    },
+    {
+      "train_step": 2619,
+      "loss": 0.5883533720958497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5883533720958497
+    },
+    {
+      "train_step": 2620,
+      "loss": 0.5316146967292266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5316146967292266
+    },
+    {
+      "train_step": 2621,
+      "loss": 0.5475769950165538,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5475769950165538
+    },
+    {
+      "train_step": 2622,
+      "loss": 0.5084034523434527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5084034523434527
+    },
+    {
+      "train_step": 2623,
+      "loss": 0.7100054954488708,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7100054954488708
+    },
+    {
+      "train_step": 2624,
+      "loss": 0.7620622635078086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7620622635078086
+    },
+    {
+      "train_step": 2625,
+      "loss": 0.5407734974070288,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5407734974070288
+    },
+    {
+      "train_step": 2626,
+      "loss": 0.8138883784844785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8138883784844785
+    },
+    {
+      "train_step": 2627,
+      "loss": 0.740178501470009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.740178501470009
+    },
+    {
+      "train_step": 2628,
+      "loss": 0.6104169075304592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6104169075304592
+    },
+    {
+      "train_step": 2629,
+      "loss": 0.6728227504113339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6728227504113339
+    },
+    {
+      "train_step": 2630,
+      "loss": 0.5367537111418055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5367537111418055
+    },
+    {
+      "train_step": 2631,
+      "loss": 0.5312417220577142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5312417220577142
+    },
+    {
+      "train_step": 2632,
+      "loss": 0.41847421447719924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41847421447719924
+    },
+    {
+      "train_step": 2633,
+      "loss": 0.5734609069651211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5734609069651211
+    },
+    {
+      "train_step": 2634,
+      "loss": 0.475714211530723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.475714211530723
+    },
+    {
+      "train_step": 2635,
+      "loss": 0.6929535926546242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6929535926546242
+    },
+    {
+      "train_step": 2636,
+      "loss": 0.6070305612969898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6070305612969898
+    },
+    {
+      "train_step": 2637,
+      "loss": 0.611138945967196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.611138945967196
+    },
+    {
+      "train_step": 2638,
+      "loss": 0.6632103451905295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6632103451905295
+    },
+    {
+      "train_step": 2639,
+      "loss": 0.4886472686524644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4886472686524644
+    },
+    {
+      "train_step": 2640,
+      "loss": 0.6244045355185505,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6244045355185505
+    },
+    {
+      "train_step": 2641,
+      "loss": 0.6307252890020344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6307252890020344
+    },
+    {
+      "train_step": 2642,
+      "loss": 0.5198830935749038,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5198830935749038
+    },
+    {
+      "train_step": 2643,
+      "loss": 0.6461900934384883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6461900934384883
+    },
+    {
+      "train_step": 2644,
+      "loss": 0.548736728481651,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.548736728481651
+    },
+    {
+      "train_step": 2645,
+      "loss": 0.5648858101708041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5648858101708041
+    },
+    {
+      "train_step": 2646,
+      "loss": 0.5022185426440527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5022185426440527
+    },
+    {
+      "train_step": 2647,
+      "loss": 0.4430336950360122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4430336950360122
+    },
+    {
+      "train_step": 2648,
+      "loss": 0.7590312135890411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7590312135890411
+    },
+    {
+      "train_step": 2649,
+      "loss": 0.5729792937792454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5729792937792454
+    },
+    {
+      "train_step": 2650,
+      "loss": 0.6374651781422778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6374651781422778
+    },
+    {
+      "train_step": 2651,
+      "loss": 0.5305839347402244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5305839347402244
+    },
+    {
+      "train_step": 2652,
+      "loss": 0.6220214385894751,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6220214385894751
+    },
+    {
+      "train_step": 2653,
+      "loss": 0.4967446038864137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4967446038864137
+    },
+    {
+      "train_step": 2654,
+      "loss": 0.6328524599779095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6328524599779095
+    },
+    {
+      "train_step": 2655,
+      "loss": 0.7486301985661421,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7486301985661421
+    },
+    {
+      "train_step": 2656,
+      "loss": 0.8347928976294298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8347928976294298
+    },
+    {
+      "train_step": 2657,
+      "loss": 0.4587732033383617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4587732033383617
+    },
+    {
+      "train_step": 2658,
+      "loss": 0.7772513210386759,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7772513210386759
+    },
+    {
+      "train_step": 2659,
+      "loss": 0.5735600800302358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5735600800302358
+    },
+    {
+      "train_step": 2660,
+      "loss": 0.6750739596807088,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6750739596807088
+    },
+    {
+      "train_step": 2661,
+      "loss": 0.5426262288806387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5426262288806387
+    },
+    {
+      "train_step": 2662,
+      "loss": 0.7249969757792624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7249969757792624
+    },
+    {
+      "train_step": 2663,
+      "loss": 0.6173079349225524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6173079349225524
+    },
+    {
+      "train_step": 2664,
+      "loss": 0.429487007951207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.429487007951207
+    },
+    {
+      "train_step": 2665,
+      "loss": 0.5092407215731262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5092407215731262
+    },
+    {
+      "train_step": 2666,
+      "loss": 0.5724751208245784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5724751208245784
+    },
+    {
+      "train_step": 2667,
+      "loss": 0.7602797249410752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7602797249410752
+    },
+    {
+      "train_step": 2668,
+      "loss": 0.6707698493540387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6707698493540387
+    },
+    {
+      "train_step": 2669,
+      "loss": 0.6111736870451865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6111736870451865
+    },
+    {
+      "train_step": 2670,
+      "loss": 0.5719470872428712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5719470872428712
+    },
+    {
+      "train_step": 2671,
+      "loss": 0.5008923292469973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5008923292469973
+    },
+    {
+      "train_step": 2672,
+      "loss": 0.3091492190771811,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3091492190771811
+    },
+    {
+      "train_step": 2673,
+      "loss": 0.5218698968824217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5218698968824217
+    },
+    {
+      "train_step": 2674,
+      "loss": 0.5112349301446318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5112349301446318
+    },
+    {
+      "train_step": 2675,
+      "loss": 0.451442906477891,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.451442906477891
+    },
+    {
+      "train_step": 2676,
+      "loss": 0.6274427267048018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6274427267048018
+    },
+    {
+      "train_step": 2677,
+      "loss": 0.5718955136527321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5718955136527321
+    },
+    {
+      "train_step": 2678,
+      "loss": 0.5205619046811852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5205619046811852
+    },
+    {
+      "train_step": 2679,
+      "loss": 0.6574494903442111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6574494903442111
+    },
+    {
+      "train_step": 2680,
+      "loss": 0.6254549721865564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6254549721865564
+    },
+    {
+      "train_step": 2681,
+      "loss": 0.30942371796117046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.30942371796117046
+    },
+    {
+      "train_step": 2682,
+      "loss": 0.8236149258488589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8236149258488589
+    },
+    {
+      "train_step": 2683,
+      "loss": 0.5890841916028787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5890841916028787
+    },
+    {
+      "train_step": 2684,
+      "loss": 0.4749605063418772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4749605063418772
+    },
+    {
+      "train_step": 2685,
+      "loss": 0.501756030366604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.501756030366604
+    },
+    {
+      "train_step": 2686,
+      "loss": 0.4608243139718233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4608243139718233
+    },
+    {
+      "train_step": 2687,
+      "loss": 0.7027505678950474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7027505678950474
+    },
+    {
+      "train_step": 2688,
+      "loss": 0.6215760807342289,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6215760807342289
+    },
+    {
+      "train_step": 2689,
+      "loss": 0.5880918946041325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5880918946041325
+    },
+    {
+      "train_step": 2690,
+      "loss": 0.4014778280243479,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4014778280243479
+    },
+    {
+      "train_step": 2691,
+      "loss": 0.7410666400658262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7410666400658262
+    },
+    {
+      "train_step": 2692,
+      "loss": 0.3880375182454379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3880375182454379
+    },
+    {
+      "train_step": 2693,
+      "loss": 0.5706111657061419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706111657061419
+    },
+    {
+      "train_step": 2694,
+      "loss": 0.4648621692455399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4648621692455399
+    },
+    {
+      "train_step": 2695,
+      "loss": 0.6076442635111975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6076442635111975
+    },
+    {
+      "train_step": 2696,
+      "loss": 0.3938343435655863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3938343435655863
+    },
+    {
+      "train_step": 2697,
+      "loss": 0.6264948434850268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6264948434850268
+    },
+    {
+      "train_step": 2698,
+      "loss": 0.5319644327873414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5319644327873414
+    },
+    {
+      "train_step": 2699,
+      "loss": 0.7048580754557189,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7048580754557189
+    },
+    {
+      "train_step": 2700,
+      "loss": 0.5832946758683478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5832946758683478
+    },
+    {
+      "train_step": 2701,
+      "loss": 0.5437661726470524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5437661726470524
+    },
+    {
+      "train_step": 2702,
+      "loss": 0.5578067442989176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5578067442989176
+    },
+    {
+      "train_step": 2703,
+      "loss": 0.4201557182180259,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4201557182180259
+    },
+    {
+      "train_step": 2704,
+      "loss": 0.6769913559674663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6769913559674663
+    },
+    {
+      "train_step": 2705,
+      "loss": 0.3899816330202619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3899816330202619
+    },
+    {
+      "train_step": 2706,
+      "loss": 0.6270863050650729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6270863050650729
+    },
+    {
+      "train_step": 2707,
+      "loss": 0.4561995073869095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4561995073869095
+    },
+    {
+      "train_step": 2708,
+      "loss": 0.5974249705135202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5974249705135202
+    },
+    {
+      "train_step": 2709,
+      "loss": 0.4227409216596511,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4227409216596511
+    },
+    {
+      "train_step": 2710,
+      "loss": 0.66007204662033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66007204662033
+    },
+    {
+      "train_step": 2711,
+      "loss": 0.6938471292245331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6938471292245331
+    },
+    {
+      "train_step": 2712,
+      "loss": 0.3926937545149142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3926937545149142
+    },
+    {
+      "train_step": 2713,
+      "loss": 0.4637420625215433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4637420625215433
+    },
+    {
+      "train_step": 2714,
+      "loss": 0.5541100166614874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5541100166614874
+    },
+    {
+      "train_step": 2715,
+      "loss": 0.5437271511853822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5437271511853822
+    },
+    {
+      "train_step": 2716,
+      "loss": 0.622844977305604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.622844977305604
+    },
+    {
+      "train_step": 2717,
+      "loss": 0.35283499217478187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35283499217478187
+    },
+    {
+      "train_step": 2718,
+      "loss": 0.28319366665990525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.28319366665990525
+    },
+    {
+      "train_step": 2719,
+      "loss": 0.30245464857615456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.30245464857615456
+    },
+    {
+      "train_step": 2720,
+      "loss": 0.8095258991423528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8095258991423528
+    },
+    {
+      "train_step": 2721,
+      "loss": 0.531550812842,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.531550812842
+    },
+    {
+      "train_step": 2722,
+      "loss": 0.3623258558098655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3623258558098655
+    },
+    {
+      "train_step": 2723,
+      "loss": 0.5117654346540695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5117654346540695
+    },
+    {
+      "train_step": 2724,
+      "loss": 0.5473426098954884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5473426098954884
+    },
+    {
+      "train_step": 2725,
+      "loss": 0.5620088880697106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5620088880697106
+    },
+    {
+      "train_step": 2726,
+      "loss": 0.7084451249633423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7084451249633423
+    },
+    {
+      "train_step": 2727,
+      "loss": 0.6902076836799668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6902076836799668
+    },
+    {
+      "train_step": 2728,
+      "loss": 0.576267135575499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576267135575499
+    },
+    {
+      "train_step": 2729,
+      "loss": 0.6383596121262375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6383596121262375
+    },
+    {
+      "train_step": 2730,
+      "loss": 0.6419129719075908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6419129719075908
+    },
+    {
+      "train_step": 2731,
+      "loss": 0.31589364143621623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31589364143621623
+    },
+    {
+      "train_step": 2732,
+      "loss": 0.7511750770356909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7511750770356909
+    },
+    {
+      "train_step": 2733,
+      "loss": 0.5448990072161324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5448990072161324
+    },
+    {
+      "train_step": 2734,
+      "loss": 0.5664917176947624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5664917176947624
+    },
+    {
+      "train_step": 2735,
+      "loss": 0.802952662591385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.802952662591385
+    },
+    {
+      "train_step": 2736,
+      "loss": 0.617805415351924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.617805415351924
+    },
+    {
+      "train_step": 2737,
+      "loss": 0.6600341645922042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6600341645922042
+    },
+    {
+      "train_step": 2738,
+      "loss": 0.42525067511474424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42525067511474424
+    },
+    {
+      "train_step": 2739,
+      "loss": 0.523815921626964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.523815921626964
+    },
+    {
+      "train_step": 2740,
+      "loss": 0.4311806560754371,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4311806560754371
+    },
+    {
+      "train_step": 2741,
+      "loss": 0.7528301982017245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7528301982017245
+    },
+    {
+      "train_step": 2742,
+      "loss": 0.6142389831451972,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6142389831451972
+    },
+    {
+      "train_step": 2743,
+      "loss": 0.5539168927658463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5539168927658463
+    },
+    {
+      "train_step": 2744,
+      "loss": 0.776536165731276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.776536165731276
+    },
+    {
+      "train_step": 2745,
+      "loss": 0.4682044939003275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4682044939003275
+    },
+    {
+      "train_step": 2746,
+      "loss": 0.651142426029914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.651142426029914
+    },
+    {
+      "train_step": 2747,
+      "loss": 0.7600883818955232,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7600883818955232
+    },
+    {
+      "train_step": 2748,
+      "loss": 0.45774285454631475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45774285454631475
+    },
+    {
+      "train_step": 2749,
+      "loss": 0.5053119163807288,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5053119163807288
+    },
+    {
+      "train_step": 2750,
+      "loss": 0.6985796184643721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6985796184643721
+    },
+    {
+      "train_step": 2751,
+      "loss": 0.6385603638314936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6385603638314936
+    },
+    {
+      "train_step": 2752,
+      "loss": 0.6522817792400166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6522817792400166
+    },
+    {
+      "train_step": 2753,
+      "loss": 0.4714683058013257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4714683058013257
+    },
+    {
+      "train_step": 2754,
+      "loss": 0.5038373604596678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5038373604596678
+    },
+    {
+      "train_step": 2755,
+      "loss": 0.7717471863723164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7717471863723164
+    },
+    {
+      "train_step": 2756,
+      "loss": 0.44205764637497447,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44205764637497447
+    },
+    {
+      "train_step": 2757,
+      "loss": 0.577245767099319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.577245767099319
+    },
+    {
+      "train_step": 2758,
+      "loss": 0.7965297667791389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7965297667791389
+    },
+    {
+      "train_step": 2759,
+      "loss": 0.6916710669496992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6916710669496992
+    },
+    {
+      "train_step": 2760,
+      "loss": 0.5737654778677916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5737654778677916
+    },
+    {
+      "train_step": 2761,
+      "loss": 0.5322389869053916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5322389869053916
+    },
+    {
+      "train_step": 2762,
+      "loss": 0.5158217736178434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5158217736178434
+    },
+    {
+      "train_step": 2763,
+      "loss": 0.44808176036835035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44808176036835035
+    },
+    {
+      "train_step": 2764,
+      "loss": 0.7921359942523021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7921359942523021
+    },
+    {
+      "train_step": 2765,
+      "loss": 0.5806056470670342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5806056470670342
+    },
+    {
+      "train_step": 2766,
+      "loss": 0.7298410699835631,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7298410699835631
+    },
+    {
+      "train_step": 2767,
+      "loss": 0.6233710102599438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6233710102599438
+    },
+    {
+      "train_step": 2768,
+      "loss": 0.46428622604148134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46428622604148134
+    },
+    {
+      "train_step": 2769,
+      "loss": 0.3978826484892785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3978826484892785
+    },
+    {
+      "train_step": 2770,
+      "loss": 0.4423091499098369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4423091499098369
+    },
+    {
+      "train_step": 2771,
+      "loss": 0.48676330486421104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48676330486421104
+    },
+    {
+      "train_step": 2772,
+      "loss": 0.4999482721622407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4999482721622407
+    },
+    {
+      "train_step": 2773,
+      "loss": 0.6249942505397637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6249942505397637
+    },
+    {
+      "train_step": 2774,
+      "loss": 0.5301100961111739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5301100961111739
+    },
+    {
+      "train_step": 2775,
+      "loss": 0.5270643071189456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5270643071189456
+    },
+    {
+      "train_step": 2776,
+      "loss": 0.6930511519173175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6930511519173175
+    },
+    {
+      "train_step": 2777,
+      "loss": 0.6340202444770775,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6340202444770775
+    },
+    {
+      "train_step": 2778,
+      "loss": 0.4079883816753698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4079883816753698
+    },
+    {
+      "train_step": 2779,
+      "loss": 0.6093031472383841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6093031472383841
+    },
+    {
+      "train_step": 2780,
+      "loss": 0.49166099699146804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49166099699146804
+    },
+    {
+      "train_step": 2781,
+      "loss": 0.505577254822657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.505577254822657
+    },
+    {
+      "train_step": 2782,
+      "loss": 0.6448106239834438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6448106239834438
+    },
+    {
+      "train_step": 2783,
+      "loss": 0.5252235765746347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5252235765746347
+    },
+    {
+      "train_step": 2784,
+      "loss": 0.6528106842608494,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6528106842608494
+    },
+    {
+      "train_step": 2785,
+      "loss": 0.6943861102936929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6943861102936929
+    },
+    {
+      "train_step": 2786,
+      "loss": 0.3651678631033392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3651678631033392
+    },
+    {
+      "train_step": 2787,
+      "loss": 0.6810484453299214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6810484453299214
+    },
+    {
+      "train_step": 2788,
+      "loss": 0.5225535436322318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5225535436322318
+    },
+    {
+      "train_step": 2789,
+      "loss": 0.5979635952073543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5979635952073543
+    },
+    {
+      "train_step": 2790,
+      "loss": 0.6349530799190409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6349530799190409
+    },
+    {
+      "train_step": 2791,
+      "loss": 0.5610604228484066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5610604228484066
+    },
+    {
+      "train_step": 2792,
+      "loss": 0.5073547278647565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5073547278647565
+    },
+    {
+      "train_step": 2793,
+      "loss": 0.5718457600960228,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5718457600960228
+    },
+    {
+      "train_step": 2794,
+      "loss": 0.49036284864657137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49036284864657137
+    },
+    {
+      "train_step": 2795,
+      "loss": 0.4680503275244783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4680503275244783
+    },
+    {
+      "train_step": 2796,
+      "loss": 0.48318299935120756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48318299935120756
+    },
+    {
+      "train_step": 2797,
+      "loss": 0.6611912600736034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6611912600736034
+    },
+    {
+      "train_step": 2798,
+      "loss": 0.6632482822852238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6632482822852238
+    },
+    {
+      "train_step": 2799,
+      "loss": 0.7096145239821721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7096145239821721
+    },
+    {
+      "train_step": 2800,
+      "loss": 0.6598705121590329,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6598705121590329
+    },
+    {
+      "train_step": 2801,
+      "loss": 0.42493187760527057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42493187760527057
+    },
+    {
+      "train_step": 2802,
+      "loss": 0.6477110219060326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6477110219060326
+    },
+    {
+      "train_step": 2803,
+      "loss": 0.48341179408706847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48341179408706847
+    },
+    {
+      "train_step": 2804,
+      "loss": 0.7215806093369883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7215806093369883
+    },
+    {
+      "train_step": 2805,
+      "loss": 0.6675022451728965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6675022451728965
+    },
+    {
+      "train_step": 2806,
+      "loss": 0.6015718101447772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6015718101447772
+    },
+    {
+      "train_step": 2807,
+      "loss": 0.5701617532359476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5701617532359476
+    },
+    {
+      "train_step": 2808,
+      "loss": 0.5491450474983083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5491450474983083
+    },
+    {
+      "train_step": 2809,
+      "loss": 0.5093011620327507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5093011620327507
+    },
+    {
+      "train_step": 2810,
+      "loss": 0.5762219025944001,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5762219025944001
+    },
+    {
+      "train_step": 2811,
+      "loss": 0.576095698451099,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576095698451099
+    },
+    {
+      "train_step": 2812,
+      "loss": 0.679372120848325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.679372120848325
+    },
+    {
+      "train_step": 2813,
+      "loss": 0.6440472937933288,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6440472937933288
+    },
+    {
+      "train_step": 2814,
+      "loss": 0.5871012386935763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5871012386935763
+    },
+    {
+      "train_step": 2815,
+      "loss": 0.6448720822417751,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6448720822417751
+    },
+    {
+      "train_step": 2816,
+      "loss": 0.3758215436085561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3758215436085561
+    },
+    {
+      "train_step": 2817,
+      "loss": 0.48189268844601996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48189268844601996
+    },
+    {
+      "train_step": 2818,
+      "loss": 0.3949202291368763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3949202291368763
+    },
+    {
+      "train_step": 2819,
+      "loss": 0.6120633086531714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6120633086531714
+    },
+    {
+      "train_step": 2820,
+      "loss": 0.37553590194271297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37553590194271297
+    },
+    {
+      "train_step": 2821,
+      "loss": 0.6037415389248336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6037415389248336
+    },
+    {
+      "train_step": 2822,
+      "loss": 0.598001614099908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.598001614099908
+    },
+    {
+      "train_step": 2823,
+      "loss": 0.7285174428412892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7285174428412892
+    },
+    {
+      "train_step": 2824,
+      "loss": 0.4264218300766774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4264218300766774
+    },
+    {
+      "train_step": 2825,
+      "loss": 0.6137066137966806,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6137066137966806
+    },
+    {
+      "train_step": 2826,
+      "loss": 0.7266461618828213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7266461618828213
+    },
+    {
+      "train_step": 2827,
+      "loss": 0.5580454369437393,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5580454369437393
+    },
+    {
+      "train_step": 2828,
+      "loss": 0.7606565876205642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7606565876205642
+    },
+    {
+      "train_step": 2829,
+      "loss": 0.5550274935352901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5550274935352901
+    },
+    {
+      "train_step": 2830,
+      "loss": 0.6601059237512178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6601059237512178
+    },
+    {
+      "train_step": 2831,
+      "loss": 0.6840727784111108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6840727784111108
+    },
+    {
+      "train_step": 2832,
+      "loss": 0.6660392147019705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6660392147019705
+    },
+    {
+      "train_step": 2833,
+      "loss": 0.5301214081225172,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5301214081225172
+    },
+    {
+      "train_step": 2834,
+      "loss": 0.3488353126224032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3488353126224032
+    },
+    {
+      "train_step": 2835,
+      "loss": 0.5440188279266658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5440188279266658
+    },
+    {
+      "train_step": 2836,
+      "loss": 0.56249016278138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.56249016278138
+    },
+    {
+      "train_step": 2837,
+      "loss": 0.4999621555478019,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4999621555478019
+    },
+    {
+      "train_step": 2838,
+      "loss": 0.5830997101530582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5830997101530582
+    },
+    {
+      "train_step": 2839,
+      "loss": 0.5684884122843838,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5684884122843838
+    },
+    {
+      "train_step": 2840,
+      "loss": 0.652655819749394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.652655819749394
+    },
+    {
+      "train_step": 2841,
+      "loss": 0.6308977278394547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6308977278394547
+    },
+    {
+      "train_step": 2842,
+      "loss": 0.45519233759372274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45519233759372274
+    },
+    {
+      "train_step": 2843,
+      "loss": 0.6304689702310348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6304689702310348
+    },
+    {
+      "train_step": 2844,
+      "loss": 0.5597609903744091,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5597609903744091
+    },
+    {
+      "train_step": 2845,
+      "loss": 0.5175888639155586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5175888639155586
+    },
+    {
+      "train_step": 2846,
+      "loss": 0.7680358250110686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7680358250110686
+    },
+    {
+      "train_step": 2847,
+      "loss": 0.5389546331539022,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5389546331539022
+    },
+    {
+      "train_step": 2848,
+      "loss": 0.6637294744332967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6637294744332967
+    },
+    {
+      "train_step": 2849,
+      "loss": 0.37556555400022723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37556555400022723
+    },
+    {
+      "train_step": 2850,
+      "loss": 0.3941722888240275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3941722888240275
+    },
+    {
+      "train_step": 2851,
+      "loss": 0.75249223116367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.75249223116367
+    },
+    {
+      "train_step": 2852,
+      "loss": 0.6238102859738955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6238102859738955
+    },
+    {
+      "train_step": 2853,
+      "loss": 0.4904328866787957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4904328866787957
+    },
+    {
+      "train_step": 2854,
+      "loss": 0.47784231909721925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47784231909721925
+    },
+    {
+      "train_step": 2855,
+      "loss": 0.5890385872706537,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5890385872706537
+    },
+    {
+      "train_step": 2856,
+      "loss": 0.6100783225474646,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6100783225474646
+    },
+    {
+      "train_step": 2857,
+      "loss": 0.555265836611639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.555265836611639
+    },
+    {
+      "train_step": 2858,
+      "loss": 0.3893790362271481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3893790362271481
+    },
+    {
+      "train_step": 2859,
+      "loss": 0.5112716750311429,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5112716750311429
+    },
+    {
+      "train_step": 2860,
+      "loss": 0.584761297694369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.584761297694369
+    },
+    {
+      "train_step": 2861,
+      "loss": 0.49230387199575537,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49230387199575537
+    },
+    {
+      "train_step": 2862,
+      "loss": 0.7900689872841449,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7900689872841449
+    },
+    {
+      "train_step": 2863,
+      "loss": 0.45875946790572364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45875946790572364
+    },
+    {
+      "train_step": 2864,
+      "loss": 0.4868181568731766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4868181568731766
+    },
+    {
+      "train_step": 2865,
+      "loss": 0.4065841509605344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4065841509605344
+    },
+    {
+      "train_step": 2866,
+      "loss": 0.590355968396498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.590355968396498
+    },
+    {
+      "train_step": 2867,
+      "loss": 0.5784112630508667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5784112630508667
+    },
+    {
+      "train_step": 2868,
+      "loss": 0.45181949351502604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45181949351502604
+    },
+    {
+      "train_step": 2869,
+      "loss": 0.5107624599437717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5107624599437717
+    },
+    {
+      "train_step": 2870,
+      "loss": 0.5043580355090489,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5043580355090489
+    },
+    {
+      "train_step": 2871,
+      "loss": 0.5549466391110103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5549466391110103
+    },
+    {
+      "train_step": 2872,
+      "loss": 0.44617553429959717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44617553429959717
+    },
+    {
+      "train_step": 2873,
+      "loss": 0.4757594783257077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4757594783257077
+    },
+    {
+      "train_step": 2874,
+      "loss": 0.6608707182242131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6608707182242131
+    },
+    {
+      "train_step": 2875,
+      "loss": 0.5524239187578269,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5524239187578269
+    },
+    {
+      "train_step": 2876,
+      "loss": 0.6787743896339913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6787743896339913
+    },
+    {
+      "train_step": 2877,
+      "loss": 0.7113615131383135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7113615131383135
+    },
+    {
+      "train_step": 2878,
+      "loss": 0.4421739287029491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4421739287029491
+    },
+    {
+      "train_step": 2879,
+      "loss": 0.556133192349721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.556133192349721
+    },
+    {
+      "train_step": 2880,
+      "loss": 0.5562632460943213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5562632460943213
+    },
+    {
+      "train_step": 2881,
+      "loss": 0.3400785167691046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3400785167691046
+    },
+    {
+      "train_step": 2882,
+      "loss": 0.48174536864140655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48174536864140655
+    },
+    {
+      "train_step": 2883,
+      "loss": 0.6557401394074794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6557401394074794
+    },
+    {
+      "train_step": 2884,
+      "loss": 0.5033763175130039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5033763175130039
+    },
+    {
+      "train_step": 2885,
+      "loss": 0.5294283645706952,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5294283645706952
+    },
+    {
+      "train_step": 2886,
+      "loss": 0.6501501719924792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6501501719924792
+    },
+    {
+      "train_step": 2887,
+      "loss": 0.653083882886818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.653083882886818
+    },
+    {
+      "train_step": 2888,
+      "loss": 0.7193129857181852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7193129857181852
+    },
+    {
+      "train_step": 2889,
+      "loss": 0.36602315784325956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36602315784325956
+    },
+    {
+      "train_step": 2890,
+      "loss": 0.7349736165102025,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7349736165102025
+    },
+    {
+      "train_step": 2891,
+      "loss": 0.5186295006338223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5186295006338223
+    },
+    {
+      "train_step": 2892,
+      "loss": 0.3349370841049552,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3349370841049552
+    },
+    {
+      "train_step": 2893,
+      "loss": 0.595394631078688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.595394631078688
+    },
+    {
+      "train_step": 2894,
+      "loss": 0.6666117289415492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6666117289415492
+    },
+    {
+      "train_step": 2895,
+      "loss": 0.71961775561834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.71961775561834
+    },
+    {
+      "train_step": 2896,
+      "loss": 0.6284169711780514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6284169711780514
+    },
+    {
+      "train_step": 2897,
+      "loss": 0.4951513244919209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4951513244919209
+    },
+    {
+      "train_step": 2898,
+      "loss": 0.6916907877698113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6916907877698113
+    },
+    {
+      "train_step": 2899,
+      "loss": 0.33613563906218674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33613563906218674
+    },
+    {
+      "train_step": 2900,
+      "loss": 0.5293293043163272,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5293293043163272
+    },
+    {
+      "train_step": 2901,
+      "loss": 0.5867799620498401,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5867799620498401
+    },
+    {
+      "train_step": 2902,
+      "loss": 0.6007627714461432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6007627714461432
+    },
+    {
+      "train_step": 2903,
+      "loss": 0.707840013750236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.707840013750236
+    },
+    {
+      "train_step": 2904,
+      "loss": 0.647385608625352,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.647385608625352
+    },
+    {
+      "train_step": 2905,
+      "loss": 0.4947140418275789,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4947140418275789
+    },
+    {
+      "train_step": 2906,
+      "loss": 0.44456303252608104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44456303252608104
+    },
+    {
+      "train_step": 2907,
+      "loss": 0.5098527693437362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5098527693437362
+    },
+    {
+      "train_step": 2908,
+      "loss": 0.538854873293229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.538854873293229
+    },
+    {
+      "train_step": 2909,
+      "loss": 0.5985339941654544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5985339941654544
+    },
+    {
+      "train_step": 2910,
+      "loss": 0.5716130058888262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5716130058888262
+    },
+    {
+      "train_step": 2911,
+      "loss": 0.5007481785340011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5007481785340011
+    },
+    {
+      "train_step": 2912,
+      "loss": 0.4853184569138258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4853184569138258
+    },
+    {
+      "train_step": 2913,
+      "loss": 0.34782401545064134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34782401545064134
+    },
+    {
+      "train_step": 2914,
+      "loss": 0.5681760990311782,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5681760990311782
+    },
+    {
+      "train_step": 2915,
+      "loss": 0.5047358423975157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5047358423975157
+    },
+    {
+      "train_step": 2916,
+      "loss": 0.6791185546553513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6791185546553513
+    },
+    {
+      "train_step": 2917,
+      "loss": 0.7267041886157651,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7267041886157651
+    },
+    {
+      "train_step": 2918,
+      "loss": 0.6747874393920729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6747874393920729
+    },
+    {
+      "train_step": 2919,
+      "loss": 0.3980227388732233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3980227388732233
+    },
+    {
+      "train_step": 2920,
+      "loss": 0.5345815110722372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5345815110722372
+    },
+    {
+      "train_step": 2921,
+      "loss": 0.5379182371653366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5379182371653366
+    },
+    {
+      "train_step": 2922,
+      "loss": 0.7634866961931318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7634866961931318
+    },
+    {
+      "train_step": 2923,
+      "loss": 0.429492472359811,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.429492472359811
+    },
+    {
+      "train_step": 2924,
+      "loss": 0.7730189942125979,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7730189942125979
+    },
+    {
+      "train_step": 2925,
+      "loss": 0.5165492749547055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5165492749547055
+    },
+    {
+      "train_step": 2926,
+      "loss": 0.6765652678091513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6765652678091513
+    },
+    {
+      "train_step": 2927,
+      "loss": 0.5422385329988846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5422385329988846
+    },
+    {
+      "train_step": 2928,
+      "loss": 0.6267376709403035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6267376709403035
+    },
+    {
+      "train_step": 2929,
+      "loss": 0.6848146133815406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6848146133815406
+    },
+    {
+      "train_step": 2930,
+      "loss": 0.40463472855619187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40463472855619187
+    },
+    {
+      "train_step": 2931,
+      "loss": 0.5301623188586031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5301623188586031
+    },
+    {
+      "train_step": 2932,
+      "loss": 0.6366339167222093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6366339167222093
+    },
+    {
+      "train_step": 2933,
+      "loss": 0.4264842066825061,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4264842066825061
+    },
+    {
+      "train_step": 2934,
+      "loss": 0.47055913891176343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47055913891176343
+    },
+    {
+      "train_step": 2935,
+      "loss": 0.49853819589282405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49853819589282405
+    },
+    {
+      "train_step": 2936,
+      "loss": 0.7790519637140362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7790519637140362
+    },
+    {
+      "train_step": 2937,
+      "loss": 0.7132116981037457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7132116981037457
+    },
+    {
+      "train_step": 2938,
+      "loss": 0.6777072435491582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6777072435491582
+    },
+    {
+      "train_step": 2939,
+      "loss": 0.6022075391683976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6022075391683976
+    },
+    {
+      "train_step": 2940,
+      "loss": 0.5890504570718719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5890504570718719
+    },
+    {
+      "train_step": 2941,
+      "loss": 0.5873539835730647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5873539835730647
+    },
+    {
+      "train_step": 2942,
+      "loss": 0.43363773067582306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43363773067582306
+    },
+    {
+      "train_step": 2943,
+      "loss": 0.7306048062998671,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7306048062998671
+    },
+    {
+      "train_step": 2944,
+      "loss": 0.6166677302378778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6166677302378778
+    },
+    {
+      "train_step": 2945,
+      "loss": 0.5684178882390188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5684178882390188
+    },
+    {
+      "train_step": 2946,
+      "loss": 0.6724028386415677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6724028386415677
+    },
+    {
+      "train_step": 2947,
+      "loss": 0.970779369098274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.970779369098274
+    },
+    {
+      "train_step": 2948,
+      "loss": 0.7196654375847303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7196654375847303
+    },
+    {
+      "train_step": 2949,
+      "loss": 0.5581863009280463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5581863009280463
+    },
+    {
+      "train_step": 2950,
+      "loss": 0.524367808509817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.524367808509817
+    },
+    {
+      "train_step": 2951,
+      "loss": 0.5709460072209589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5709460072209589
+    },
+    {
+      "train_step": 2952,
+      "loss": 0.6600084751788179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6600084751788179
+    },
+    {
+      "train_step": 2953,
+      "loss": 0.5867978842456147,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5867978842456147
+    },
+    {
+      "train_step": 2954,
+      "loss": 0.6650420732534998,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6650420732534998
+    },
+    {
+      "train_step": 2955,
+      "loss": 0.7051069887407312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7051069887407312
+    },
+    {
+      "train_step": 2956,
+      "loss": 0.5996029941577571,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5996029941577571
+    },
+    {
+      "train_step": 2957,
+      "loss": 0.6587781577278985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6587781577278985
+    },
+    {
+      "train_step": 2958,
+      "loss": 0.5859584576404012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5859584576404012
+    },
+    {
+      "train_step": 2959,
+      "loss": 0.5874848078491215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5874848078491215
+    },
+    {
+      "train_step": 2960,
+      "loss": 0.6086181516811004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6086181516811004
+    },
+    {
+      "train_step": 2961,
+      "loss": 0.4881755150189102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4881755150189102
+    },
+    {
+      "train_step": 2962,
+      "loss": 0.5796566570734595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5796566570734595
+    },
+    {
+      "train_step": 2963,
+      "loss": 0.5925744997386352,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5925744997386352
+    },
+    {
+      "train_step": 2964,
+      "loss": 0.6969911823403111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6969911823403111
+    },
+    {
+      "train_step": 2965,
+      "loss": 0.5531945302721177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5531945302721177
+    },
+    {
+      "train_step": 2966,
+      "loss": 0.610702636465254,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.610702636465254
+    },
+    {
+      "train_step": 2967,
+      "loss": 0.5933972872191446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5933972872191446
+    },
+    {
+      "train_step": 2968,
+      "loss": 0.5585298618019012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5585298618019012
+    },
+    {
+      "train_step": 2969,
+      "loss": 0.6809169218282154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6809169218282154
+    },
+    {
+      "train_step": 2970,
+      "loss": 0.6954957152645904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6954957152645904
+    },
+    {
+      "train_step": 2971,
+      "loss": 0.48380423935980876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48380423935980876
+    },
+    {
+      "train_step": 2972,
+      "loss": 0.4550302458796457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4550302458796457
+    },
+    {
+      "train_step": 2973,
+      "loss": 0.6475059172004436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6475059172004436
+    },
+    {
+      "train_step": 2974,
+      "loss": 0.49928856728918103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49928856728918103
+    },
+    {
+      "train_step": 2975,
+      "loss": 0.2620345629882218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2620345629882218
+    },
+    {
+      "train_step": 2976,
+      "loss": 0.3377318460232523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3377318460232523
+    },
+    {
+      "train_step": 2977,
+      "loss": 0.4532225939946374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4532225939946374
+    },
+    {
+      "train_step": 2978,
+      "loss": 0.40501951919493323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40501951919493323
+    },
+    {
+      "train_step": 2979,
+      "loss": 0.6631789728546506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6631789728546506
+    },
+    {
+      "train_step": 2980,
+      "loss": 0.777773486800704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.777773486800704
+    },
+    {
+      "train_step": 2981,
+      "loss": 0.639169514402976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.639169514402976
+    },
+    {
+      "train_step": 2982,
+      "loss": 0.5668608407210403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5668608407210403
+    },
+    {
+      "train_step": 2983,
+      "loss": 0.6507884592838832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6507884592838832
+    },
+    {
+      "train_step": 2984,
+      "loss": 0.5904194309345793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5904194309345793
+    },
+    {
+      "train_step": 2985,
+      "loss": 0.5481575973032863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5481575973032863
+    },
+    {
+      "train_step": 2986,
+      "loss": 0.37812757905947275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37812757905947275
+    },
+    {
+      "train_step": 2987,
+      "loss": 0.6981900872041968,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6981900872041968
+    },
+    {
+      "train_step": 2988,
+      "loss": 0.6968497527894689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6968497527894689
+    },
+    {
+      "train_step": 2989,
+      "loss": 0.5751952334592021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5751952334592021
+    },
+    {
+      "train_step": 2990,
+      "loss": 0.5883861064981228,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5883861064981228
+    },
+    {
+      "train_step": 2991,
+      "loss": 0.6632605399570369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6632605399570369
+    },
+    {
+      "train_step": 2992,
+      "loss": 0.3411591695516391,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3411591695516391
+    },
+    {
+      "train_step": 2993,
+      "loss": 0.6658218237783442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6658218237783442
+    },
+    {
+      "train_step": 2994,
+      "loss": 0.4522383973086296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4522383973086296
+    },
+    {
+      "train_step": 2995,
+      "loss": 0.6796528896918175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6796528896918175
+    },
+    {
+      "train_step": 2996,
+      "loss": 0.7389688807864778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7389688807864778
+    },
+    {
+      "train_step": 2997,
+      "loss": 0.5388468930386173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5388468930386173
+    },
+    {
+      "train_step": 2998,
+      "loss": 0.23513576402393233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.23513576402393233
+    },
+    {
+      "train_step": 2999,
+      "loss": 0.5421047503481504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5421047503481504
+    },
+    {
+      "train_step": 3000,
+      "loss": 0.43551506645237137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43551506645237137
+    },
+    {
+      "train_step": 3001,
+      "loss": 0.48209161165910736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48209161165910736
+    },
+    {
+      "train_step": 3002,
+      "loss": 0.558049328327036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.558049328327036
+    },
+    {
+      "train_step": 3003,
+      "loss": 0.6026952545758376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6026952545758376
+    },
+    {
+      "train_step": 3004,
+      "loss": 0.5266425399407161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5266425399407161
+    },
+    {
+      "train_step": 3005,
+      "loss": 0.3707189432469976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3707189432469976
+    },
+    {
+      "train_step": 3006,
+      "loss": 0.45921936082428483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45921936082428483
+    },
+    {
+      "train_step": 3007,
+      "loss": 0.567995933578639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.567995933578639
+    },
+    {
+      "train_step": 3008,
+      "loss": 0.4784770252998979,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4784770252998979
+    },
+    {
+      "train_step": 3009,
+      "loss": 0.7047569441074832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7047569441074832
+    },
+    {
+      "train_step": 3010,
+      "loss": 0.4115679102130758,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4115679102130758
+    },
+    {
+      "train_step": 3011,
+      "loss": 0.5871954363945985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5871954363945985
+    },
+    {
+      "train_step": 3012,
+      "loss": 0.5408750127903235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5408750127903235
+    },
+    {
+      "train_step": 3013,
+      "loss": 0.6043350172693156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6043350172693156
+    },
+    {
+      "train_step": 3014,
+      "loss": 0.6051029058537948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6051029058537948
+    },
+    {
+      "train_step": 3015,
+      "loss": 0.5979108308098692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5979108308098692
+    },
+    {
+      "train_step": 3016,
+      "loss": 0.6526560213289293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6526560213289293
+    },
+    {
+      "train_step": 3017,
+      "loss": 0.7339890845975974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7339890845975974
+    },
+    {
+      "train_step": 3018,
+      "loss": 0.7239011897167611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7239011897167611
+    },
+    {
+      "train_step": 3019,
+      "loss": 0.4967628311238245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4967628311238245
+    },
+    {
+      "train_step": 3020,
+      "loss": 0.6325381213200065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6325381213200065
+    },
+    {
+      "train_step": 3021,
+      "loss": 0.4631555445849166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4631555445849166
+    },
+    {
+      "train_step": 3022,
+      "loss": 0.5407095387021271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5407095387021271
+    },
+    {
+      "train_step": 3023,
+      "loss": 0.4782035444533193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4782035444533193
+    },
+    {
+      "train_step": 3024,
+      "loss": 0.5465089768317648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5465089768317648
+    },
+    {
+      "train_step": 3025,
+      "loss": 0.5153584590912701,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5153584590912701
+    },
+    {
+      "train_step": 3026,
+      "loss": 0.5337141665535317,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5337141665535317
+    },
+    {
+      "train_step": 3027,
+      "loss": 0.5574701849941808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5574701849941808
+    },
+    {
+      "train_step": 3028,
+      "loss": 0.44646770043433454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44646770043433454
+    },
+    {
+      "train_step": 3029,
+      "loss": 0.6609598944985947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6609598944985947
+    },
+    {
+      "train_step": 3030,
+      "loss": 0.4961860733314948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4961860733314948
+    },
+    {
+      "train_step": 3031,
+      "loss": 0.7197672681747641,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7197672681747641
+    },
+    {
+      "train_step": 3032,
+      "loss": 0.5559458338516269,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5559458338516269
+    },
+    {
+      "train_step": 3033,
+      "loss": 0.6194364572240636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6194364572240636
+    },
+    {
+      "train_step": 3034,
+      "loss": 0.5759530813632212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5759530813632212
+    },
+    {
+      "train_step": 3035,
+      "loss": 0.5197673368084544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5197673368084544
+    },
+    {
+      "train_step": 3036,
+      "loss": 0.5827509489634556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5827509489634556
+    },
+    {
+      "train_step": 3037,
+      "loss": 0.4806593004258465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4806593004258465
+    },
+    {
+      "train_step": 3038,
+      "loss": 0.6774574205845488,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6774574205845488
+    },
+    {
+      "train_step": 3039,
+      "loss": 0.34455904024369766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34455904024369766
+    },
+    {
+      "train_step": 3040,
+      "loss": 0.4700794225204029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4700794225204029
+    },
+    {
+      "train_step": 3041,
+      "loss": 0.7741510523831024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7741510523831024
+    },
+    {
+      "train_step": 3042,
+      "loss": 0.6857850444791704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6857850444791704
+    },
+    {
+      "train_step": 3043,
+      "loss": 0.7197136392035289,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7197136392035289
+    },
+    {
+      "train_step": 3044,
+      "loss": 0.7043516719965607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7043516719965607
+    },
+    {
+      "train_step": 3045,
+      "loss": 0.5368381705262606,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5368381705262606
+    },
+    {
+      "train_step": 3046,
+      "loss": 0.5579966944225578,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5579966944225578
+    },
+    {
+      "train_step": 3047,
+      "loss": 0.4514963695697214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4514963695697214
+    },
+    {
+      "train_step": 3048,
+      "loss": 0.522036352955362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.522036352955362
+    },
+    {
+      "train_step": 3049,
+      "loss": 0.5901847965018202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5901847965018202
+    },
+    {
+      "train_step": 3050,
+      "loss": 0.596862278140841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.596862278140841
+    },
+    {
+      "train_step": 3051,
+      "loss": 0.5333454899790517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5333454899790517
+    },
+    {
+      "train_step": 3052,
+      "loss": 0.5122472352853272,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5122472352853272
+    },
+    {
+      "train_step": 3053,
+      "loss": 0.6091259510484907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6091259510484907
+    },
+    {
+      "train_step": 3054,
+      "loss": 0.55469977292081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.55469977292081
+    },
+    {
+      "train_step": 3055,
+      "loss": 0.512022128612613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.512022128612613
+    },
+    {
+      "train_step": 3056,
+      "loss": 0.5333844175568361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5333844175568361
+    },
+    {
+      "train_step": 3057,
+      "loss": 0.5071174952403718,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5071174952403718
+    },
+    {
+      "train_step": 3058,
+      "loss": 0.6024276739288923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6024276739288923
+    },
+    {
+      "train_step": 3059,
+      "loss": 0.6600126481608882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6600126481608882
+    },
+    {
+      "train_step": 3060,
+      "loss": 0.3616053603949877,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3616053603949877
+    },
+    {
+      "train_step": 3061,
+      "loss": 0.6162403539322113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6162403539322113
+    },
+    {
+      "train_step": 3062,
+      "loss": 0.657486930263294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.657486930263294
+    },
+    {
+      "train_step": 3063,
+      "loss": 0.4237173678511911,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4237173678511911
+    },
+    {
+      "train_step": 3064,
+      "loss": 0.5252336141323297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5252336141323297
+    },
+    {
+      "train_step": 3065,
+      "loss": 0.5703655147183034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5703655147183034
+    },
+    {
+      "train_step": 3066,
+      "loss": 0.4717131532387616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4717131532387616
+    },
+    {
+      "train_step": 3067,
+      "loss": 0.472359413636955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.472359413636955
+    },
+    {
+      "train_step": 3068,
+      "loss": 0.6216027975477878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6216027975477878
+    },
+    {
+      "train_step": 3069,
+      "loss": 0.3008128086859276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3008128086859276
+    },
+    {
+      "train_step": 3070,
+      "loss": 0.7800152565852841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7800152565852841
+    },
+    {
+      "train_step": 3071,
+      "loss": 0.5507333826681257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5507333826681257
+    },
+    {
+      "train_step": 3072,
+      "loss": 0.48482452383770236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48482452383770236
+    },
+    {
+      "train_step": 3073,
+      "loss": 0.556280445368241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.556280445368241
+    },
+    {
+      "train_step": 3074,
+      "loss": 0.5303011003682087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5303011003682087
+    },
+    {
+      "train_step": 3075,
+      "loss": 0.443527826177411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.443527826177411
+    },
+    {
+      "train_step": 3076,
+      "loss": 0.7035873444337463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7035873444337463
+    },
+    {
+      "train_step": 3077,
+      "loss": 0.41477569782775175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41477569782775175
+    },
+    {
+      "train_step": 3078,
+      "loss": 0.7227764230513389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7227764230513389
+    },
+    {
+      "train_step": 3079,
+      "loss": 0.5741205906206222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5741205906206222
+    },
+    {
+      "train_step": 3080,
+      "loss": 0.5578058894542922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5578058894542922
+    },
+    {
+      "train_step": 3081,
+      "loss": 0.4663491621803858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4663491621803858
+    },
+    {
+      "train_step": 3082,
+      "loss": 0.5544192897203425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5544192897203425
+    },
+    {
+      "train_step": 3083,
+      "loss": 0.5057462716988265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5057462716988265
+    },
+    {
+      "train_step": 3084,
+      "loss": 0.7716333839781583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7716333839781583
+    },
+    {
+      "train_step": 3085,
+      "loss": 0.6426019366624754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6426019366624754
+    },
+    {
+      "train_step": 3086,
+      "loss": 0.6253356456528136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6253356456528136
+    },
+    {
+      "train_step": 3087,
+      "loss": 0.5660635881310729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5660635881310729
+    },
+    {
+      "train_step": 3088,
+      "loss": 0.5437454532105254,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5437454532105254
+    },
+    {
+      "train_step": 3089,
+      "loss": 0.6277783702374417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6277783702374417
+    },
+    {
+      "train_step": 3090,
+      "loss": 0.6255380433469822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6255380433469822
+    },
+    {
+      "train_step": 3091,
+      "loss": 0.8646775661343138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8646775661343138
+    },
+    {
+      "train_step": 3092,
+      "loss": 0.6998960868630657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6998960868630657
+    },
+    {
+      "train_step": 3093,
+      "loss": 0.44064776285792084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44064776285792084
+    },
+    {
+      "train_step": 3094,
+      "loss": 0.8529394274610631,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8529394274610631
+    },
+    {
+      "train_step": 3095,
+      "loss": 0.42517501046593553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42517501046593553
+    },
+    {
+      "train_step": 3096,
+      "loss": 0.5904183081034805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5904183081034805
+    },
+    {
+      "train_step": 3097,
+      "loss": 0.3971462107125567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3971462107125567
+    },
+    {
+      "train_step": 3098,
+      "loss": 0.7821241941995215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7821241941995215
+    },
+    {
+      "train_step": 3099,
+      "loss": 0.45741808671824497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45741808671824497
+    },
+    {
+      "train_step": 3100,
+      "loss": 0.5883635429416342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5883635429416342
+    },
+    {
+      "train_step": 3101,
+      "loss": 0.41327509828810366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41327509828810366
+    },
+    {
+      "train_step": 3102,
+      "loss": 0.5280122105900119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5280122105900119
+    },
+    {
+      "train_step": 3103,
+      "loss": 0.6487148051727644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6487148051727644
+    },
+    {
+      "train_step": 3104,
+      "loss": 0.5951949982850284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5951949982850284
+    },
+    {
+      "train_step": 3105,
+      "loss": 0.36442713913850644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36442713913850644
+    },
+    {
+      "train_step": 3106,
+      "loss": 0.6942312511454541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6942312511454541
+    },
+    {
+      "train_step": 3107,
+      "loss": 0.5434870883968218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5434870883968218
+    },
+    {
+      "train_step": 3108,
+      "loss": 0.7334390008644535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7334390008644535
+    },
+    {
+      "train_step": 3109,
+      "loss": 0.43169753697873575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43169753697873575
+    },
+    {
+      "train_step": 3110,
+      "loss": 0.671338849829835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.671338849829835
+    },
+    {
+      "train_step": 3111,
+      "loss": 0.7124607963976528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7124607963976528
+    },
+    {
+      "train_step": 3112,
+      "loss": 0.5067081963251125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5067081963251125
+    },
+    {
+      "train_step": 3113,
+      "loss": 0.5728532360287248,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5728532360287248
+    },
+    {
+      "train_step": 3114,
+      "loss": 0.5412691579992354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5412691579992354
+    },
+    {
+      "train_step": 3115,
+      "loss": 0.631256912062089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.631256912062089
+    },
+    {
+      "train_step": 3116,
+      "loss": 0.3113894112950694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3113894112950694
+    },
+    {
+      "train_step": 3117,
+      "loss": 0.7816992622624315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7816992622624315
+    },
+    {
+      "train_step": 3118,
+      "loss": 0.4409908650881847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4409908650881847
+    },
+    {
+      "train_step": 3119,
+      "loss": 0.6774794561674953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6774794561674953
+    },
+    {
+      "train_step": 3120,
+      "loss": 0.5898512471917485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5898512471917485
+    },
+    {
+      "train_step": 3121,
+      "loss": 0.423601785998597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.423601785998597
+    },
+    {
+      "train_step": 3122,
+      "loss": 0.7962491789564625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7962491789564625
+    },
+    {
+      "train_step": 3123,
+      "loss": 0.4523154471225149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4523154471225149
+    },
+    {
+      "train_step": 3124,
+      "loss": 0.4360304112470136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4360304112470136
+    },
+    {
+      "train_step": 3125,
+      "loss": 0.5337362971896563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5337362971896563
+    },
+    {
+      "train_step": 3126,
+      "loss": 0.5097171651438988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5097171651438988
+    },
+    {
+      "train_step": 3127,
+      "loss": 0.5012217199628908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5012217199628908
+    },
+    {
+      "train_step": 3128,
+      "loss": 0.4543200401686155,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4543200401686155
+    },
+    {
+      "train_step": 3129,
+      "loss": 0.7192821424210417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7192821424210417
+    },
+    {
+      "train_step": 3130,
+      "loss": 0.5317334073277051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5317334073277051
+    },
+    {
+      "train_step": 3131,
+      "loss": 0.5388908839904748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5388908839904748
+    },
+    {
+      "train_step": 3132,
+      "loss": 0.5771774421595105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5771774421595105
+    },
+    {
+      "train_step": 3133,
+      "loss": 0.7003328004316045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7003328004316045
+    },
+    {
+      "train_step": 3134,
+      "loss": 0.4187565676390105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4187565676390105
+    },
+    {
+      "train_step": 3135,
+      "loss": 0.5221748083042408,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5221748083042408
+    },
+    {
+      "train_step": 3136,
+      "loss": 0.6102968357801527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6102968357801527
+    },
+    {
+      "train_step": 3137,
+      "loss": 0.6396996464941819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6396996464941819
+    },
+    {
+      "train_step": 3138,
+      "loss": 0.6830211538478862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6830211538478862
+    },
+    {
+      "train_step": 3139,
+      "loss": 0.36187823745988923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36187823745988923
+    },
+    {
+      "train_step": 3140,
+      "loss": 0.3154481602140999,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3154481602140999
+    },
+    {
+      "train_step": 3141,
+      "loss": 0.579135196663596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.579135196663596
+    },
+    {
+      "train_step": 3142,
+      "loss": 0.4977696907039921,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4977696907039921
+    },
+    {
+      "train_step": 3143,
+      "loss": 0.4297390113655542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4297390113655542
+    },
+    {
+      "train_step": 3144,
+      "loss": 0.5156744587192574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5156744587192574
+    },
+    {
+      "train_step": 3145,
+      "loss": 0.44717767153039223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44717767153039223
+    },
+    {
+      "train_step": 3146,
+      "loss": 0.63748532619645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.63748532619645
+    },
+    {
+      "train_step": 3147,
+      "loss": 0.7113983913949629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7113983913949629
+    },
+    {
+      "train_step": 3148,
+      "loss": 0.7164798738924681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7164798738924681
+    },
+    {
+      "train_step": 3149,
+      "loss": 0.4304132342270174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4304132342270174
+    },
+    {
+      "train_step": 3150,
+      "loss": 0.6809731792757014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6809731792757014
+    },
+    {
+      "train_step": 3151,
+      "loss": 0.43404109705928035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43404109705928035
+    },
+    {
+      "train_step": 3152,
+      "loss": 0.5321528514358274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5321528514358274
+    },
+    {
+      "train_step": 3153,
+      "loss": 0.5702838135094089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5702838135094089
+    },
+    {
+      "train_step": 3154,
+      "loss": 0.5283043930431182,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5283043930431182
+    },
+    {
+      "train_step": 3155,
+      "loss": 0.5949276200618409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5949276200618409
+    },
+    {
+      "train_step": 3156,
+      "loss": 0.4465668702341103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4465668702341103
+    },
+    {
+      "train_step": 3157,
+      "loss": 0.3403376221100388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3403376221100388
+    },
+    {
+      "train_step": 3158,
+      "loss": 0.5012459394460218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5012459394460218
+    },
+    {
+      "train_step": 3159,
+      "loss": 0.6289083791614188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6289083791614188
+    },
+    {
+      "train_step": 3160,
+      "loss": 0.6430935026091893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6430935026091893
+    },
+    {
+      "train_step": 3161,
+      "loss": 0.4624010419033324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4624010419033324
+    },
+    {
+      "train_step": 3162,
+      "loss": 0.5352987003511516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5352987003511516
+    },
+    {
+      "train_step": 3163,
+      "loss": 0.5889192565172818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5889192565172818
+    },
+    {
+      "train_step": 3164,
+      "loss": 0.5035107488358623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035107488358623
+    },
+    {
+      "train_step": 3165,
+      "loss": 0.4651217669225841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4651217669225841
+    },
+    {
+      "train_step": 3166,
+      "loss": 0.7356103567287903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7356103567287903
+    },
+    {
+      "train_step": 3167,
+      "loss": 0.6924078405730358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6924078405730358
+    },
+    {
+      "train_step": 3168,
+      "loss": 0.6062333780329252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6062333780329252
+    },
+    {
+      "train_step": 3169,
+      "loss": 0.5207431092292882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5207431092292882
+    },
+    {
+      "train_step": 3170,
+      "loss": 0.47639951745346343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47639951745346343
+    },
+    {
+      "train_step": 3171,
+      "loss": 0.702151523656582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.702151523656582
+    },
+    {
+      "train_step": 3172,
+      "loss": 0.573567874339118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.573567874339118
+    },
+    {
+      "train_step": 3173,
+      "loss": 0.5644508492220597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5644508492220597
+    },
+    {
+      "train_step": 3174,
+      "loss": 0.5128020262408968,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5128020262408968
+    },
+    {
+      "train_step": 3175,
+      "loss": 0.651762697276157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.651762697276157
+    },
+    {
+      "train_step": 3176,
+      "loss": 0.6495387150913157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6495387150913157
+    },
+    {
+      "train_step": 3177,
+      "loss": 0.4444418375677361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4444418375677361
+    },
+    {
+      "train_step": 3178,
+      "loss": 0.5637473331023889,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5637473331023889
+    },
+    {
+      "train_step": 3179,
+      "loss": 0.669797531624477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.669797531624477
+    },
+    {
+      "train_step": 3180,
+      "loss": 0.6955252452356252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6955252452356252
+    },
+    {
+      "train_step": 3181,
+      "loss": 0.6738354332508397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6738354332508397
+    },
+    {
+      "train_step": 3182,
+      "loss": 0.352403611290833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.352403611290833
+    },
+    {
+      "train_step": 3183,
+      "loss": 0.6118302015480598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6118302015480598
+    },
+    {
+      "train_step": 3184,
+      "loss": 0.5470675702995458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5470675702995458
+    },
+    {
+      "train_step": 3185,
+      "loss": 0.4897898377998245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4897898377998245
+    },
+    {
+      "train_step": 3186,
+      "loss": 0.6223985546322433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6223985546322433
+    },
+    {
+      "train_step": 3187,
+      "loss": 0.548933733379853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.548933733379853
+    },
+    {
+      "train_step": 3188,
+      "loss": 0.3707434770452186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3707434770452186
+    },
+    {
+      "train_step": 3189,
+      "loss": 0.5027514423990166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5027514423990166
+    },
+    {
+      "train_step": 3190,
+      "loss": 0.5775980679041619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5775980679041619
+    },
+    {
+      "train_step": 3191,
+      "loss": 0.7240376964127779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7240376964127779
+    },
+    {
+      "train_step": 3192,
+      "loss": 0.48787528283366244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48787528283366244
+    },
+    {
+      "train_step": 3193,
+      "loss": 0.42549508010421644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42549508010421644
+    },
+    {
+      "train_step": 3194,
+      "loss": 0.6823479971522028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6823479971522028
+    },
+    {
+      "train_step": 3195,
+      "loss": 0.48699408057308213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48699408057308213
+    },
+    {
+      "train_step": 3196,
+      "loss": 0.6162446309899344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6162446309899344
+    },
+    {
+      "train_step": 3197,
+      "loss": 0.5394157176969978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5394157176969978
+    },
+    {
+      "train_step": 3198,
+      "loss": 0.2757339642393798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2757339642393798
+    },
+    {
+      "train_step": 3199,
+      "loss": 0.7110884567731229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7110884567731229
+    },
+    {
+      "train_step": 3200,
+      "loss": 0.6554773335337897,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6554773335337897
+    },
+    {
+      "train_step": 3201,
+      "loss": 0.5213540039935596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5213540039935596
+    },
+    {
+      "train_step": 3202,
+      "loss": 0.6583435113607978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6583435113607978
+    },
+    {
+      "train_step": 3203,
+      "loss": 0.7245849012689257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7245849012689257
+    },
+    {
+      "train_step": 3204,
+      "loss": 0.5559722388634996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5559722388634996
+    },
+    {
+      "train_step": 3205,
+      "loss": 0.5094581558165403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5094581558165403
+    },
+    {
+      "train_step": 3206,
+      "loss": 0.52158797333131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.52158797333131
+    },
+    {
+      "train_step": 3207,
+      "loss": 0.6678651620844309,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6678651620844309
+    },
+    {
+      "train_step": 3208,
+      "loss": 0.6854677185335164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6854677185335164
+    },
+    {
+      "train_step": 3209,
+      "loss": 0.6157696767437636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6157696767437636
+    },
+    {
+      "train_step": 3210,
+      "loss": 0.5655113347779758,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5655113347779758
+    },
+    {
+      "train_step": 3211,
+      "loss": 0.7694181976012434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7694181976012434
+    },
+    {
+      "train_step": 3212,
+      "loss": 0.3813203194328301,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3813203194328301
+    },
+    {
+      "train_step": 3213,
+      "loss": 0.4857555784540971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4857555784540971
+    },
+    {
+      "train_step": 3214,
+      "loss": 0.6023700448319557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6023700448319557
+    },
+    {
+      "train_step": 3215,
+      "loss": 0.5293244675413785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5293244675413785
+    },
+    {
+      "train_step": 3216,
+      "loss": 0.548847118463981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.548847118463981
+    },
+    {
+      "train_step": 3217,
+      "loss": 0.6348728019496543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6348728019496543
+    },
+    {
+      "train_step": 3218,
+      "loss": 0.6874798801017751,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6874798801017751
+    },
+    {
+      "train_step": 3219,
+      "loss": 0.532777913086486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.532777913086486
+    },
+    {
+      "train_step": 3220,
+      "loss": 0.6227364981622322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6227364981622322
+    },
+    {
+      "train_step": 3221,
+      "loss": 0.6189192654783439,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6189192654783439
+    },
+    {
+      "train_step": 3222,
+      "loss": 0.8370275979540729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8370275979540729
+    },
+    {
+      "train_step": 3223,
+      "loss": 0.5428174088726997,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5428174088726997
+    },
+    {
+      "train_step": 3224,
+      "loss": 0.6640336797904111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6640336797904111
+    },
+    {
+      "train_step": 3225,
+      "loss": 0.46536107286831596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46536107286831596
+    },
+    {
+      "train_step": 3226,
+      "loss": 0.74486166494738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.74486166494738
+    },
+    {
+      "train_step": 3227,
+      "loss": 0.5092038474256748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5092038474256748
+    },
+    {
+      "train_step": 3228,
+      "loss": 0.362764325837144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.362764325837144
+    },
+    {
+      "train_step": 3229,
+      "loss": 0.5893295870726314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5893295870726314
+    },
+    {
+      "train_step": 3230,
+      "loss": 0.696428815699568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.696428815699568
+    },
+    {
+      "train_step": 3231,
+      "loss": 0.5479646956873663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5479646956873663
+    },
+    {
+      "train_step": 3232,
+      "loss": 0.5280297235542425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5280297235542425
+    },
+    {
+      "train_step": 3233,
+      "loss": 0.36296068635577805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36296068635577805
+    },
+    {
+      "train_step": 3234,
+      "loss": 0.5396729007742742,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5396729007742742
+    },
+    {
+      "train_step": 3235,
+      "loss": 0.6899615925343374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6899615925343374
+    },
+    {
+      "train_step": 3236,
+      "loss": 0.6556996772899267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6556996772899267
+    },
+    {
+      "train_step": 3237,
+      "loss": 0.6317055457712919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6317055457712919
+    },
+    {
+      "train_step": 3238,
+      "loss": 0.45002272334346305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45002272334346305
+    },
+    {
+      "train_step": 3239,
+      "loss": 0.6225512800989199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6225512800989199
+    },
+    {
+      "train_step": 3240,
+      "loss": 0.4888725432336722,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4888725432336722
+    },
+    {
+      "train_step": 3241,
+      "loss": 0.5327328441085138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5327328441085138
+    },
+    {
+      "train_step": 3242,
+      "loss": 0.5804246316231758,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5804246316231758
+    },
+    {
+      "train_step": 3243,
+      "loss": 0.4953323836044524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4953323836044524
+    },
+    {
+      "train_step": 3244,
+      "loss": 0.5405663086833453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5405663086833453
+    },
+    {
+      "train_step": 3245,
+      "loss": 0.6259532420685534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6259532420685534
+    },
+    {
+      "train_step": 3246,
+      "loss": 0.5563308767931207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5563308767931207
+    },
+    {
+      "train_step": 3247,
+      "loss": 0.6987210956818821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6987210956818821
+    },
+    {
+      "train_step": 3248,
+      "loss": 0.4981741354509295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4981741354509295
+    },
+    {
+      "train_step": 3249,
+      "loss": 0.48029656106519925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48029656106519925
+    },
+    {
+      "train_step": 3250,
+      "loss": 0.5086705323120616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5086705323120616
+    },
+    {
+      "train_step": 3251,
+      "loss": 0.5463901501024478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5463901501024478
+    },
+    {
+      "train_step": 3252,
+      "loss": 0.7861158691596258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7861158691596258
+    },
+    {
+      "train_step": 3253,
+      "loss": 0.46865554534457754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46865554534457754
+    },
+    {
+      "train_step": 3254,
+      "loss": 0.46778678536132046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46778678536132046
+    },
+    {
+      "train_step": 3255,
+      "loss": 0.5357693691996561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5357693691996561
+    },
+    {
+      "train_step": 3256,
+      "loss": 0.3657917563519689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3657917563519689
+    },
+    {
+      "train_step": 3257,
+      "loss": 0.6892450480713214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6892450480713214
+    },
+    {
+      "train_step": 3258,
+      "loss": 0.4123004131301924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4123004131301924
+    },
+    {
+      "train_step": 3259,
+      "loss": 0.5306535503319351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5306535503319351
+    },
+    {
+      "train_step": 3260,
+      "loss": 0.6588917071971714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6588917071971714
+    },
+    {
+      "train_step": 3261,
+      "loss": 0.52384909220501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.52384909220501
+    },
+    {
+      "train_step": 3262,
+      "loss": 0.6570198166408342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6570198166408342
+    },
+    {
+      "train_step": 3263,
+      "loss": 0.5064878260742853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5064878260742853
+    },
+    {
+      "train_step": 3264,
+      "loss": 0.5829541314565779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5829541314565779
+    },
+    {
+      "train_step": 3265,
+      "loss": 0.704843104828059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.704843104828059
+    },
+    {
+      "train_step": 3266,
+      "loss": 0.564740608861117,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.564740608861117
+    },
+    {
+      "train_step": 3267,
+      "loss": 0.4336482293002499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4336482293002499
+    },
+    {
+      "train_step": 3268,
+      "loss": 0.5448235613159242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5448235613159242
+    },
+    {
+      "train_step": 3269,
+      "loss": 0.5534893120681916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5534893120681916
+    },
+    {
+      "train_step": 3270,
+      "loss": 0.6138007238650368,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6138007238650368
+    },
+    {
+      "train_step": 3271,
+      "loss": 0.5570598918351791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5570598918351791
+    },
+    {
+      "train_step": 3272,
+      "loss": 0.5825893654070636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5825893654070636
+    },
+    {
+      "train_step": 3273,
+      "loss": 0.5383324996813783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5383324996813783
+    },
+    {
+      "train_step": 3274,
+      "loss": 0.5606836012732174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5606836012732174
+    },
+    {
+      "train_step": 3275,
+      "loss": 0.4712955499699342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4712955499699342
+    },
+    {
+      "train_step": 3276,
+      "loss": 0.48911073845269026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48911073845269026
+    },
+    {
+      "train_step": 3277,
+      "loss": 0.5251706003684589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5251706003684589
+    },
+    {
+      "train_step": 3278,
+      "loss": 0.5494111793993314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5494111793993314
+    },
+    {
+      "train_step": 3279,
+      "loss": 0.7449433951676107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7449433951676107
+    },
+    {
+      "train_step": 3280,
+      "loss": 0.7155216303839385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7155216303839385
+    },
+    {
+      "train_step": 3281,
+      "loss": 0.5728244883705128,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5728244883705128
+    },
+    {
+      "train_step": 3282,
+      "loss": 0.6047840047700778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6047840047700778
+    },
+    {
+      "train_step": 3283,
+      "loss": 0.7044835664317268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7044835664317268
+    },
+    {
+      "train_step": 3284,
+      "loss": 0.4761931066111874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4761931066111874
+    },
+    {
+      "train_step": 3285,
+      "loss": 0.6757188175561716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6757188175561716
+    },
+    {
+      "train_step": 3286,
+      "loss": 0.7166808158388713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7166808158388713
+    },
+    {
+      "train_step": 3287,
+      "loss": 0.47565333376704383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47565333376704383
+    },
+    {
+      "train_step": 3288,
+      "loss": 0.5853476566368796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5853476566368796
+    },
+    {
+      "train_step": 3289,
+      "loss": 0.6872535142962695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6872535142962695
+    },
+    {
+      "train_step": 3290,
+      "loss": 0.5914741340442949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5914741340442949
+    },
+    {
+      "train_step": 3291,
+      "loss": 0.6357297496473456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6357297496473456
+    },
+    {
+      "train_step": 3292,
+      "loss": 0.5892179011984159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5892179011984159
+    },
+    {
+      "train_step": 3293,
+      "loss": 0.6421429103639541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6421429103639541
+    },
+    {
+      "train_step": 3294,
+      "loss": 0.6453331134273533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6453331134273533
+    },
+    {
+      "train_step": 3295,
+      "loss": 0.5129056682706744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5129056682706744
+    },
+    {
+      "train_step": 3296,
+      "loss": 0.6038325241286695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6038325241286695
+    },
+    {
+      "train_step": 3297,
+      "loss": 0.5441213987599107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441213987599107
+    },
+    {
+      "train_step": 3298,
+      "loss": 0.6372333653941045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6372333653941045
+    },
+    {
+      "train_step": 3299,
+      "loss": 0.6185310535490904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6185310535490904
+    },
+    {
+      "train_step": 3300,
+      "loss": 0.7911310194257883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7911310194257883
+    },
+    {
+      "train_step": 3301,
+      "loss": 0.6387585151417194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6387585151417194
+    },
+    {
+      "train_step": 3302,
+      "loss": 0.3670667689376205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3670667689376205
+    },
+    {
+      "train_step": 3303,
+      "loss": 0.5668672982098248,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5668672982098248
+    },
+    {
+      "train_step": 3304,
+      "loss": 0.6182771542900676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6182771542900676
+    },
+    {
+      "train_step": 3305,
+      "loss": 0.5652639322609319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5652639322609319
+    },
+    {
+      "train_step": 3306,
+      "loss": 0.5523263304234338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5523263304234338
+    },
+    {
+      "train_step": 3307,
+      "loss": 0.635953361017028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.635953361017028
+    },
+    {
+      "train_step": 3308,
+      "loss": 0.548676250415128,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.548676250415128
+    },
+    {
+      "train_step": 3309,
+      "loss": 0.6262397657684068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6262397657684068
+    },
+    {
+      "train_step": 3310,
+      "loss": 0.6377068816349382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6377068816349382
+    },
+    {
+      "train_step": 3311,
+      "loss": 0.6856041096874956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6856041096874956
+    },
+    {
+      "train_step": 3312,
+      "loss": 0.5246571437416722,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5246571437416722
+    },
+    {
+      "train_step": 3313,
+      "loss": 0.8608180783371822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8608180783371822
+    },
+    {
+      "train_step": 3314,
+      "loss": 0.5402515932995554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5402515932995554
+    },
+    {
+      "train_step": 3315,
+      "loss": 0.509709177796709,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.509709177796709
+    },
+    {
+      "train_step": 3316,
+      "loss": 0.7149857391114294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7149857391114294
+    },
+    {
+      "train_step": 3317,
+      "loss": 0.6788844966205334,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6788844966205334
+    },
+    {
+      "train_step": 3318,
+      "loss": 0.49185633208744767,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49185633208744767
+    },
+    {
+      "train_step": 3319,
+      "loss": 0.4742924982955874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4742924982955874
+    },
+    {
+      "train_step": 3320,
+      "loss": 0.7074950926670933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7074950926670933
+    },
+    {
+      "train_step": 3321,
+      "loss": 0.5067418885413297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5067418885413297
+    },
+    {
+      "train_step": 3322,
+      "loss": 0.6098006692462364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6098006692462364
+    },
+    {
+      "train_step": 3323,
+      "loss": 0.646415804785626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.646415804785626
+    },
+    {
+      "train_step": 3324,
+      "loss": 0.6062640307309054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6062640307309054
+    },
+    {
+      "train_step": 3325,
+      "loss": 0.43787488468241653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43787488468241653
+    },
+    {
+      "train_step": 3326,
+      "loss": 0.6397552275883869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6397552275883869
+    },
+    {
+      "train_step": 3327,
+      "loss": 0.5500584514628175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5500584514628175
+    },
+    {
+      "train_step": 3328,
+      "loss": 0.5592759981630916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5592759981630916
+    },
+    {
+      "train_step": 3329,
+      "loss": 0.46769327339755856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46769327339755856
+    },
+    {
+      "train_step": 3330,
+      "loss": 0.7116798282836551,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7116798282836551
+    },
+    {
+      "train_step": 3331,
+      "loss": 0.5344936519616247,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5344936519616247
+    },
+    {
+      "train_step": 3332,
+      "loss": 0.40002208390213034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40002208390213034
+    },
+    {
+      "train_step": 3333,
+      "loss": 0.6162581929774128,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6162581929774128
+    },
+    {
+      "train_step": 3334,
+      "loss": 0.6443541339572325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6443541339572325
+    },
+    {
+      "train_step": 3335,
+      "loss": 0.5531196493994587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5531196493994587
+    },
+    {
+      "train_step": 3336,
+      "loss": 0.5316076957819516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5316076957819516
+    },
+    {
+      "train_step": 3337,
+      "loss": 0.7660703832685906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7660703832685906
+    },
+    {
+      "train_step": 3338,
+      "loss": 0.6387982891273314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6387982891273314
+    },
+    {
+      "train_step": 3339,
+      "loss": 0.5761917993214102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5761917993214102
+    },
+    {
+      "train_step": 3340,
+      "loss": 0.5288174014816351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5288174014816351
+    },
+    {
+      "train_step": 3341,
+      "loss": 0.40432034278695383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40432034278695383
+    },
+    {
+      "train_step": 3342,
+      "loss": 0.545056011421625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.545056011421625
+    },
+    {
+      "train_step": 3343,
+      "loss": 0.6728698514466895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6728698514466895
+    },
+    {
+      "train_step": 3344,
+      "loss": 0.462964071587486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.462964071587486
+    },
+    {
+      "train_step": 3345,
+      "loss": 0.6601178747910653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6601178747910653
+    },
+    {
+      "train_step": 3346,
+      "loss": 0.858603341113151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.858603341113151
+    },
+    {
+      "train_step": 3347,
+      "loss": 0.47395264524221975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47395264524221975
+    },
+    {
+      "train_step": 3348,
+      "loss": 0.554110392406904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.554110392406904
+    },
+    {
+      "train_step": 3349,
+      "loss": 0.6333327491839706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6333327491839706
+    },
+    {
+      "train_step": 3350,
+      "loss": 0.485813085404862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.485813085404862
+    },
+    {
+      "train_step": 3351,
+      "loss": 0.45131711678939834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45131711678939834
+    },
+    {
+      "train_step": 3352,
+      "loss": 0.48610004142548857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48610004142548857
+    },
+    {
+      "train_step": 3353,
+      "loss": 0.4329905536576476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4329905536576476
+    },
+    {
+      "train_step": 3354,
+      "loss": 0.5743248340762086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5743248340762086
+    },
+    {
+      "train_step": 3355,
+      "loss": 0.691954837515423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.691954837515423
+    },
+    {
+      "train_step": 3356,
+      "loss": 0.36430129498159813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36430129498159813
+    },
+    {
+      "train_step": 3357,
+      "loss": 0.6710969039293855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6710969039293855
+    },
+    {
+      "train_step": 3358,
+      "loss": 0.5156784667278304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5156784667278304
+    },
+    {
+      "train_step": 3359,
+      "loss": 0.5706658948605057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706658948605057
+    },
+    {
+      "train_step": 3360,
+      "loss": 0.6484408633699286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6484408633699286
+    },
+    {
+      "train_step": 3361,
+      "loss": 0.5925253132392183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5925253132392183
+    },
+    {
+      "train_step": 3362,
+      "loss": 0.6027845110106905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6027845110106905
+    },
+    {
+      "train_step": 3363,
+      "loss": 0.5452905488961408,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5452905488961408
+    },
+    {
+      "train_step": 3364,
+      "loss": 0.5623678708086141,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5623678708086141
+    },
+    {
+      "train_step": 3365,
+      "loss": 0.3867409188785824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3867409188785824
+    },
+    {
+      "train_step": 3366,
+      "loss": 0.45387485484043316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45387485484043316
+    },
+    {
+      "train_step": 3367,
+      "loss": 0.7377688244020644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7377688244020644
+    },
+    {
+      "train_step": 3368,
+      "loss": 0.7018398664040542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7018398664040542
+    },
+    {
+      "train_step": 3369,
+      "loss": 0.8770745499888778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8770745499888778
+    },
+    {
+      "train_step": 3370,
+      "loss": 0.8202521526517037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8202521526517037
+    },
+    {
+      "train_step": 3371,
+      "loss": 0.45534989779094653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45534989779094653
+    },
+    {
+      "train_step": 3372,
+      "loss": 0.6466174766748919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6466174766748919
+    },
+    {
+      "train_step": 3373,
+      "loss": 0.5636260287301876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5636260287301876
+    },
+    {
+      "train_step": 3374,
+      "loss": 0.37512289433267104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37512289433267104
+    },
+    {
+      "train_step": 3375,
+      "loss": 0.4791895188041938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4791895188041938
+    },
+    {
+      "train_step": 3376,
+      "loss": 0.3450458486850726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3450458486850726
+    },
+    {
+      "train_step": 3377,
+      "loss": 0.8171330866079048,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8171330866079048
+    },
+    {
+      "train_step": 3378,
+      "loss": 0.6506422554490859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6506422554490859
+    },
+    {
+      "train_step": 3379,
+      "loss": 0.5239741815896114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5239741815896114
+    },
+    {
+      "train_step": 3380,
+      "loss": 0.6539857263642661,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6539857263642661
+    },
+    {
+      "train_step": 3381,
+      "loss": 0.409506810759521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.409506810759521
+    },
+    {
+      "train_step": 3382,
+      "loss": 0.491273662732968,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.491273662732968
+    },
+    {
+      "train_step": 3383,
+      "loss": 0.7636457101883444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7636457101883444
+    },
+    {
+      "train_step": 3384,
+      "loss": 0.40479480656727285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40479480656727285
+    },
+    {
+      "train_step": 3385,
+      "loss": 0.5401091310016227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5401091310016227
+    },
+    {
+      "train_step": 3386,
+      "loss": 0.44795460721795466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44795460721795466
+    },
+    {
+      "train_step": 3387,
+      "loss": 0.5687069616109528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5687069616109528
+    },
+    {
+      "train_step": 3388,
+      "loss": 0.41181642673511953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41181642673511953
+    },
+    {
+      "train_step": 3389,
+      "loss": 0.4553301269382439,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4553301269382439
+    },
+    {
+      "train_step": 3390,
+      "loss": 0.46948302266592773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46948302266592773
+    },
+    {
+      "train_step": 3391,
+      "loss": 0.5689063796504512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5689063796504512
+    },
+    {
+      "train_step": 3392,
+      "loss": 0.6475873621266717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6475873621266717
+    },
+    {
+      "train_step": 3393,
+      "loss": 0.7600737987358528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7600737987358528
+    },
+    {
+      "train_step": 3394,
+      "loss": 0.5970913327789032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5970913327789032
+    },
+    {
+      "train_step": 3395,
+      "loss": 0.5515221525837516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5515221525837516
+    },
+    {
+      "train_step": 3396,
+      "loss": 0.4566214109549703,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4566214109549703
+    },
+    {
+      "train_step": 3397,
+      "loss": 0.5870457633884955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5870457633884955
+    },
+    {
+      "train_step": 3398,
+      "loss": 0.4735088812980796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4735088812980796
+    },
+    {
+      "train_step": 3399,
+      "loss": 0.507229837637025,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.507229837637025
+    },
+    {
+      "train_step": 3400,
+      "loss": 0.5524959521905227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5524959521905227
+    },
+    {
+      "train_step": 3401,
+      "loss": 0.7766496421377103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7766496421377103
+    },
+    {
+      "train_step": 3402,
+      "loss": 0.5486167193468159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5486167193468159
+    },
+    {
+      "train_step": 3403,
+      "loss": 0.5766632482846914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5766632482846914
+    },
+    {
+      "train_step": 3404,
+      "loss": 0.5429178135389044,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5429178135389044
+    },
+    {
+      "train_step": 3405,
+      "loss": 0.4299620299131202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4299620299131202
+    },
+    {
+      "train_step": 3406,
+      "loss": 0.3370428730323659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3370428730323659
+    },
+    {
+      "train_step": 3407,
+      "loss": 0.589090670576158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.589090670576158
+    },
+    {
+      "train_step": 3408,
+      "loss": 0.5480218490988747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5480218490988747
+    },
+    {
+      "train_step": 3409,
+      "loss": 0.6535389886002594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6535389886002594
+    },
+    {
+      "train_step": 3410,
+      "loss": 0.493088142011318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.493088142011318
+    },
+    {
+      "train_step": 3411,
+      "loss": 0.5106442467733412,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5106442467733412
+    },
+    {
+      "train_step": 3412,
+      "loss": 0.7668304505884502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7668304505884502
+    },
+    {
+      "train_step": 3413,
+      "loss": 0.49068647378048086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49068647378048086
+    },
+    {
+      "train_step": 3414,
+      "loss": 0.49808353657274346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49808353657274346
+    },
+    {
+      "train_step": 3415,
+      "loss": 0.6115883711736592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6115883711736592
+    },
+    {
+      "train_step": 3416,
+      "loss": 0.5804344036655268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5804344036655268
+    },
+    {
+      "train_step": 3417,
+      "loss": 0.6890884527240816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6890884527240816
+    },
+    {
+      "train_step": 3418,
+      "loss": 0.3629857642754491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3629857642754491
+    },
+    {
+      "train_step": 3419,
+      "loss": 0.585908808856783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.585908808856783
+    },
+    {
+      "train_step": 3420,
+      "loss": 0.5835214173552125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5835214173552125
+    },
+    {
+      "train_step": 3421,
+      "loss": 0.5322986777149303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5322986777149303
+    },
+    {
+      "train_step": 3422,
+      "loss": 0.4853631694841244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4853631694841244
+    },
+    {
+      "train_step": 3423,
+      "loss": 0.4030988529003474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4030988529003474
+    },
+    {
+      "train_step": 3424,
+      "loss": 0.42364176832563766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42364176832563766
+    },
+    {
+      "train_step": 3425,
+      "loss": 0.4003455597552627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4003455597552627
+    },
+    {
+      "train_step": 3426,
+      "loss": 0.3634619256154587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3634619256154587
+    },
+    {
+      "train_step": 3427,
+      "loss": 0.5119294698329659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5119294698329659
+    },
+    {
+      "train_step": 3428,
+      "loss": 0.59206930940314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.59206930940314
+    },
+    {
+      "train_step": 3429,
+      "loss": 0.5568683549025176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5568683549025176
+    },
+    {
+      "train_step": 3430,
+      "loss": 0.472209575548484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.472209575548484
+    },
+    {
+      "train_step": 3431,
+      "loss": 0.5565747373897105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5565747373897105
+    },
+    {
+      "train_step": 3432,
+      "loss": 0.4801397384781858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4801397384781858
+    },
+    {
+      "train_step": 3433,
+      "loss": 0.6547441586639529,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6547441586639529
+    },
+    {
+      "train_step": 3434,
+      "loss": 0.48698005209347356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48698005209347356
+    },
+    {
+      "train_step": 3435,
+      "loss": 0.5648164417121937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5648164417121937
+    },
+    {
+      "train_step": 3436,
+      "loss": 0.431143897383215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.431143897383215
+    },
+    {
+      "train_step": 3437,
+      "loss": 0.31621323258013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31621323258013
+    },
+    {
+      "train_step": 3438,
+      "loss": 0.5927665596145802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5927665596145802
+    },
+    {
+      "train_step": 3439,
+      "loss": 0.436284556977626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.436284556977626
+    },
+    {
+      "train_step": 3440,
+      "loss": 0.6555259157435386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6555259157435386
+    },
+    {
+      "train_step": 3441,
+      "loss": 0.7240933016574458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7240933016574458
+    },
+    {
+      "train_step": 3442,
+      "loss": 0.4097102999772855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4097102999772855
+    },
+    {
+      "train_step": 3443,
+      "loss": 0.5035812169791346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035812169791346
+    },
+    {
+      "train_step": 3444,
+      "loss": 0.5516904767494499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5516904767494499
+    },
+    {
+      "train_step": 3445,
+      "loss": 0.5565628454355555,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5565628454355555
+    },
+    {
+      "train_step": 3446,
+      "loss": 0.7401042305291871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7401042305291871
+    },
+    {
+      "train_step": 3447,
+      "loss": 0.617041603686988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.617041603686988
+    },
+    {
+      "train_step": 3448,
+      "loss": 0.6529365861353595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6529365861353595
+    },
+    {
+      "train_step": 3449,
+      "loss": 0.6283151857140269,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6283151857140269
+    },
+    {
+      "train_step": 3450,
+      "loss": 0.6425403237189968,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6425403237189968
+    },
+    {
+      "train_step": 3451,
+      "loss": 0.4824306170640016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4824306170640016
+    },
+    {
+      "train_step": 3452,
+      "loss": 0.6254529375087101,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6254529375087101
+    },
+    {
+      "train_step": 3453,
+      "loss": 0.46597300499277955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46597300499277955
+    },
+    {
+      "train_step": 3454,
+      "loss": 0.5808391222956293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5808391222956293
+    },
+    {
+      "train_step": 3455,
+      "loss": 0.5698644666604374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5698644666604374
+    },
+    {
+      "train_step": 3456,
+      "loss": 0.5845001308968139,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5845001308968139
+    },
+    {
+      "train_step": 3457,
+      "loss": 0.45497517901414114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45497517901414114
+    },
+    {
+      "train_step": 3458,
+      "loss": 0.3559940363341431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3559940363341431
+    },
+    {
+      "train_step": 3459,
+      "loss": 0.6973412740988101,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6973412740988101
+    },
+    {
+      "train_step": 3460,
+      "loss": 0.44328058350015426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44328058350015426
+    },
+    {
+      "train_step": 3461,
+      "loss": 0.5318922869057225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5318922869057225
+    },
+    {
+      "train_step": 3462,
+      "loss": 0.4988252470088993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4988252470088993
+    },
+    {
+      "train_step": 3463,
+      "loss": 0.5122042334998779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5122042334998779
+    },
+    {
+      "train_step": 3464,
+      "loss": 0.553761601433444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.553761601433444
+    },
+    {
+      "train_step": 3465,
+      "loss": 0.5546309797874982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5546309797874982
+    },
+    {
+      "train_step": 3466,
+      "loss": 0.5969536860713215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5969536860713215
+    },
+    {
+      "train_step": 3467,
+      "loss": 0.43545295393236366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43545295393236366
+    },
+    {
+      "train_step": 3468,
+      "loss": 0.4067410492309822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4067410492309822
+    },
+    {
+      "train_step": 3469,
+      "loss": 0.678240748970818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.678240748970818
+    },
+    {
+      "train_step": 3470,
+      "loss": 0.5503491511145848,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5503491511145848
+    },
+    {
+      "train_step": 3471,
+      "loss": 0.5959154433014215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5959154433014215
+    },
+    {
+      "train_step": 3472,
+      "loss": 0.6660554593918971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6660554593918971
+    },
+    {
+      "train_step": 3473,
+      "loss": 0.5017966678564703,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5017966678564703
+    },
+    {
+      "train_step": 3474,
+      "loss": 0.7892854575248807,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7892854575248807
+    },
+    {
+      "train_step": 3475,
+      "loss": 0.7185380262707363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7185380262707363
+    },
+    {
+      "train_step": 3476,
+      "loss": 0.4994144806153213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4994144806153213
+    },
+    {
+      "train_step": 3477,
+      "loss": 0.5392756987896056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5392756987896056
+    },
+    {
+      "train_step": 3478,
+      "loss": 0.5783515652974559,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5783515652974559
+    },
+    {
+      "train_step": 3479,
+      "loss": 0.6432183494466495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6432183494466495
+    },
+    {
+      "train_step": 3480,
+      "loss": 0.626157230108017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.626157230108017
+    },
+    {
+      "train_step": 3481,
+      "loss": 0.7924002227322259,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7924002227322259
+    },
+    {
+      "train_step": 3482,
+      "loss": 0.5685700801031588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5685700801031588
+    },
+    {
+      "train_step": 3483,
+      "loss": 0.7940614513476716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7940614513476716
+    },
+    {
+      "train_step": 3484,
+      "loss": 0.4351766456736642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4351766456736642
+    },
+    {
+      "train_step": 3485,
+      "loss": 0.5984738265205024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5984738265205024
+    },
+    {
+      "train_step": 3486,
+      "loss": 0.41010480738694466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41010480738694466
+    },
+    {
+      "train_step": 3487,
+      "loss": 0.779537670363791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.779537670363791
+    },
+    {
+      "train_step": 3488,
+      "loss": 0.7896329863195851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7896329863195851
+    },
+    {
+      "train_step": 3489,
+      "loss": 0.47397215857084884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47397215857084884
+    },
+    {
+      "train_step": 3490,
+      "loss": 0.4185449835684234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4185449835684234
+    },
+    {
+      "train_step": 3491,
+      "loss": 0.37377712812923225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37377712812923225
+    },
+    {
+      "train_step": 3492,
+      "loss": 0.654350019245932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.654350019245932
+    },
+    {
+      "train_step": 3493,
+      "loss": 0.5096012667905788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5096012667905788
+    },
+    {
+      "train_step": 3494,
+      "loss": 0.637146166845113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.637146166845113
+    },
+    {
+      "train_step": 3495,
+      "loss": 0.5576720960083263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5576720960083263
+    },
+    {
+      "train_step": 3496,
+      "loss": 0.5232457929608274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5232457929608274
+    },
+    {
+      "train_step": 3497,
+      "loss": 0.7206898040115204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7206898040115204
+    },
+    {
+      "train_step": 3498,
+      "loss": 0.7049381147736924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7049381147736924
+    },
+    {
+      "train_step": 3499,
+      "loss": 0.6176800298279628,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6176800298279628
+    },
+    {
+      "train_step": 3500,
+      "loss": 0.5957103361242962,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5957103361242962
+    },
+    {
+      "train_step": 3501,
+      "loss": 0.5018127625530748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5018127625530748
+    },
+    {
+      "train_step": 3502,
+      "loss": 0.6118485992274907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6118485992274907
+    },
+    {
+      "train_step": 3503,
+      "loss": 0.5170435717393624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5170435717393624
+    },
+    {
+      "train_step": 3504,
+      "loss": 0.7760391831428123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7760391831428123
+    },
+    {
+      "train_step": 3505,
+      "loss": 0.6715073311206892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6715073311206892
+    },
+    {
+      "train_step": 3506,
+      "loss": 0.4461786867759464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4461786867759464
+    },
+    {
+      "train_step": 3507,
+      "loss": 0.5159883171298413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5159883171298413
+    },
+    {
+      "train_step": 3508,
+      "loss": 0.4613375706659891,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4613375706659891
+    },
+    {
+      "train_step": 3509,
+      "loss": 0.5672624577591541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5672624577591541
+    },
+    {
+      "train_step": 3510,
+      "loss": 0.46164609539374873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46164609539374873
+    },
+    {
+      "train_step": 3511,
+      "loss": 0.6958840377190418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6958840377190418
+    },
+    {
+      "train_step": 3512,
+      "loss": 0.4732268983423433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4732268983423433
+    },
+    {
+      "train_step": 3513,
+      "loss": 0.7717247439316557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7717247439316557
+    },
+    {
+      "train_step": 3514,
+      "loss": 0.44739821014762216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44739821014762216
+    },
+    {
+      "train_step": 3515,
+      "loss": 0.41227152613791024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41227152613791024
+    },
+    {
+      "train_step": 3516,
+      "loss": 0.47184962995287777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47184962995287777
+    },
+    {
+      "train_step": 3517,
+      "loss": 0.5841386153998166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5841386153998166
+    },
+    {
+      "train_step": 3518,
+      "loss": 0.5403316768368249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5403316768368249
+    },
+    {
+      "train_step": 3519,
+      "loss": 0.5872600718706636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5872600718706636
+    },
+    {
+      "train_step": 3520,
+      "loss": 0.6262387476605981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6262387476605981
+    },
+    {
+      "train_step": 3521,
+      "loss": 0.5018891317622126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5018891317622126
+    },
+    {
+      "train_step": 3522,
+      "loss": 0.6410695367744667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6410695367744667
+    },
+    {
+      "train_step": 3523,
+      "loss": 0.5977690333231387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5977690333231387
+    },
+    {
+      "train_step": 3524,
+      "loss": 0.5162870462205376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5162870462205376
+    },
+    {
+      "train_step": 3525,
+      "loss": 0.26620260923796424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.26620260923796424
+    },
+    {
+      "train_step": 3526,
+      "loss": 0.5661085740521323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5661085740521323
+    },
+    {
+      "train_step": 3527,
+      "loss": 0.7015482782710767,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7015482782710767
+    },
+    {
+      "train_step": 3528,
+      "loss": 0.6112187790858707,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6112187790858707
+    },
+    {
+      "train_step": 3529,
+      "loss": 0.5585490710161269,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5585490710161269
+    },
+    {
+      "train_step": 3530,
+      "loss": 0.6595711374149579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6595711374149579
+    },
+    {
+      "train_step": 3531,
+      "loss": 0.4527203166111532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4527203166111532
+    },
+    {
+      "train_step": 3532,
+      "loss": 0.4491191597052051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4491191597052051
+    },
+    {
+      "train_step": 3533,
+      "loss": 0.7520229357995034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7520229357995034
+    },
+    {
+      "train_step": 3534,
+      "loss": 0.5614815544111632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5614815544111632
+    },
+    {
+      "train_step": 3535,
+      "loss": 0.8435025913923062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8435025913923062
+    },
+    {
+      "train_step": 3536,
+      "loss": 0.5873596489552223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5873596489552223
+    },
+    {
+      "train_step": 3537,
+      "loss": 0.6484546943602738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6484546943602738
+    },
+    {
+      "train_step": 3538,
+      "loss": 0.6377316961540611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6377316961540611
+    },
+    {
+      "train_step": 3539,
+      "loss": 0.5498004430514253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5498004430514253
+    },
+    {
+      "train_step": 3540,
+      "loss": 0.47729145054350874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47729145054350874
+    },
+    {
+      "train_step": 3541,
+      "loss": 0.5521836171819642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5521836171819642
+    },
+    {
+      "train_step": 3542,
+      "loss": 0.583040425997434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.583040425997434
+    },
+    {
+      "train_step": 3543,
+      "loss": 0.5500271483461314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5500271483461314
+    },
+    {
+      "train_step": 3544,
+      "loss": 0.45791924302955245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45791924302955245
+    },
+    {
+      "train_step": 3545,
+      "loss": 0.5551158624535832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5551158624535832
+    },
+    {
+      "train_step": 3546,
+      "loss": 0.6145571919877781,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6145571919877781
+    },
+    {
+      "train_step": 3547,
+      "loss": 0.558359239909479,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.558359239909479
+    },
+    {
+      "train_step": 3548,
+      "loss": 0.6068328264799425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6068328264799425
+    },
+    {
+      "train_step": 3549,
+      "loss": 0.8315688096990075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8315688096990075
+    },
+    {
+      "train_step": 3550,
+      "loss": 0.7366340281162473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7366340281162473
+    },
+    {
+      "train_step": 3551,
+      "loss": 0.6336251531341269,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6336251531341269
+    },
+    {
+      "train_step": 3552,
+      "loss": 0.7324315936998568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7324315936998568
+    },
+    {
+      "train_step": 3553,
+      "loss": 0.6883246014902523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6883246014902523
+    },
+    {
+      "train_step": 3554,
+      "loss": 0.6594221721732085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6594221721732085
+    },
+    {
+      "train_step": 3555,
+      "loss": 0.3735722684814045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3735722684814045
+    },
+    {
+      "train_step": 3556,
+      "loss": 0.4712701261104797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4712701261104797
+    },
+    {
+      "train_step": 3557,
+      "loss": 0.8058010545379052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8058010545379052
+    },
+    {
+      "train_step": 3558,
+      "loss": 0.5187004193212195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5187004193212195
+    },
+    {
+      "train_step": 3559,
+      "loss": 0.6245366717657551,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6245366717657551
+    },
+    {
+      "train_step": 3560,
+      "loss": 0.528806504828614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.528806504828614
+    },
+    {
+      "train_step": 3561,
+      "loss": 0.5262055296439323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5262055296439323
+    },
+    {
+      "train_step": 3562,
+      "loss": 0.6088061461709361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6088061461709361
+    },
+    {
+      "train_step": 3563,
+      "loss": 0.48223672366552656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48223672366552656
+    },
+    {
+      "train_step": 3564,
+      "loss": 0.5918501484100714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5918501484100714
+    },
+    {
+      "train_step": 3565,
+      "loss": 0.566614659338432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.566614659338432
+    },
+    {
+      "train_step": 3566,
+      "loss": 0.5383346558717331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5383346558717331
+    },
+    {
+      "train_step": 3567,
+      "loss": 0.717319119772117,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.717319119772117
+    },
+    {
+      "train_step": 3568,
+      "loss": 0.641433621612798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.641433621612798
+    },
+    {
+      "train_step": 3569,
+      "loss": 0.6883913521471084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6883913521471084
+    },
+    {
+      "train_step": 3570,
+      "loss": 0.5231566701451102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5231566701451102
+    },
+    {
+      "train_step": 3571,
+      "loss": 0.6194601521313899,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6194601521313899
+    },
+    {
+      "train_step": 3572,
+      "loss": 0.5760960069581949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5760960069581949
+    },
+    {
+      "train_step": 3573,
+      "loss": 0.5299372633785009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5299372633785009
+    },
+    {
+      "train_step": 3574,
+      "loss": 0.3550713277248957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3550713277248957
+    },
+    {
+      "train_step": 3575,
+      "loss": 0.5334849859110549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5334849859110549
+    },
+    {
+      "train_step": 3576,
+      "loss": 0.5793798368844226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5793798368844226
+    },
+    {
+      "train_step": 3577,
+      "loss": 0.6834495994859692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6834495994859692
+    },
+    {
+      "train_step": 3578,
+      "loss": 0.701321455796209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.701321455796209
+    },
+    {
+      "train_step": 3579,
+      "loss": 0.37556910275700367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37556910275700367
+    },
+    {
+      "train_step": 3580,
+      "loss": 0.48541548168903526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48541548168903526
+    },
+    {
+      "train_step": 3581,
+      "loss": 0.5408505692355786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5408505692355786
+    },
+    {
+      "train_step": 3582,
+      "loss": 0.6997772241358168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6997772241358168
+    },
+    {
+      "train_step": 3583,
+      "loss": 0.6485851364886295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6485851364886295
+    },
+    {
+      "train_step": 3584,
+      "loss": 0.5494299731141348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5494299731141348
+    },
+    {
+      "train_step": 3585,
+      "loss": 0.6235722042165102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6235722042165102
+    },
+    {
+      "train_step": 3586,
+      "loss": 0.6251292579822697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6251292579822697
+    },
+    {
+      "train_step": 3587,
+      "loss": 0.6338063206365638,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6338063206365638
+    },
+    {
+      "train_step": 3588,
+      "loss": 0.27088420574287203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.27088420574287203
+    },
+    {
+      "train_step": 3589,
+      "loss": 0.7269119410976665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7269119410976665
+    },
+    {
+      "train_step": 3590,
+      "loss": 0.38159149488057525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38159149488057525
+    },
+    {
+      "train_step": 3591,
+      "loss": 0.7105016045326485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7105016045326485
+    },
+    {
+      "train_step": 3592,
+      "loss": 0.5800123499375245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5800123499375245
+    },
+    {
+      "train_step": 3593,
+      "loss": 0.8554273670485989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8554273670485989
+    },
+    {
+      "train_step": 3594,
+      "loss": 0.5636921910912115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5636921910912115
+    },
+    {
+      "train_step": 3595,
+      "loss": 0.8246026867537883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8246026867537883
+    },
+    {
+      "train_step": 3596,
+      "loss": 0.5791821389482186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5791821389482186
+    },
+    {
+      "train_step": 3597,
+      "loss": 0.3963448736229351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3963448736229351
+    },
+    {
+      "train_step": 3598,
+      "loss": 0.5525957731876687,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5525957731876687
+    },
+    {
+      "train_step": 3599,
+      "loss": 0.6383710904127696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6383710904127696
+    },
+    {
+      "train_step": 3600,
+      "loss": 0.6080218748153008,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6080218748153008
+    },
+    {
+      "train_step": 3601,
+      "loss": 0.46761295958743,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46761295958743
+    },
+    {
+      "train_step": 3602,
+      "loss": 0.48818807764044464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48818807764044464
+    },
+    {
+      "train_step": 3603,
+      "loss": 0.6064310450919739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6064310450919739
+    },
+    {
+      "train_step": 3604,
+      "loss": 0.6097326068047766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6097326068047766
+    },
+    {
+      "train_step": 3605,
+      "loss": 0.7254160574492199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7254160574492199
+    },
+    {
+      "train_step": 3606,
+      "loss": 0.5463896465979905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5463896465979905
+    },
+    {
+      "train_step": 3607,
+      "loss": 0.5737246815654151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5737246815654151
+    },
+    {
+      "train_step": 3608,
+      "loss": 0.3488969610358184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3488969610358184
+    },
+    {
+      "train_step": 3609,
+      "loss": 0.462703827602829,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.462703827602829
+    },
+    {
+      "train_step": 3610,
+      "loss": 0.6135215761343447,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6135215761343447
+    },
+    {
+      "train_step": 3611,
+      "loss": 0.52700816921791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.52700816921791
+    },
+    {
+      "train_step": 3612,
+      "loss": 0.6271451694396204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6271451694396204
+    },
+    {
+      "train_step": 3613,
+      "loss": 0.5549698421243415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5549698421243415
+    },
+    {
+      "train_step": 3614,
+      "loss": 0.5896119149744391,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5896119149744391
+    },
+    {
+      "train_step": 3615,
+      "loss": 0.32528726631650917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32528726631650917
+    },
+    {
+      "train_step": 3616,
+      "loss": 0.5253787710066374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5253787710066374
+    },
+    {
+      "train_step": 3617,
+      "loss": 0.42553459947139266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42553459947139266
+    },
+    {
+      "train_step": 3618,
+      "loss": 0.6397492558358879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6397492558358879
+    },
+    {
+      "train_step": 3619,
+      "loss": 0.6113115592135211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6113115592135211
+    },
+    {
+      "train_step": 3620,
+      "loss": 0.6452029202369577,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6452029202369577
+    },
+    {
+      "train_step": 3621,
+      "loss": 0.5190612185717033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5190612185717033
+    },
+    {
+      "train_step": 3622,
+      "loss": 0.5048243131724895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5048243131724895
+    },
+    {
+      "train_step": 3623,
+      "loss": 0.5174849570527323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5174849570527323
+    },
+    {
+      "train_step": 3624,
+      "loss": 0.5509951600914604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5509951600914604
+    },
+    {
+      "train_step": 3625,
+      "loss": 0.531235571110969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.531235571110969
+    },
+    {
+      "train_step": 3626,
+      "loss": 0.4724820679685063,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4724820679685063
+    },
+    {
+      "train_step": 3627,
+      "loss": 0.4600090322836049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4600090322836049
+    },
+    {
+      "train_step": 3628,
+      "loss": 0.7406220467381379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7406220467381379
+    },
+    {
+      "train_step": 3629,
+      "loss": 0.5878275529854758,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5878275529854758
+    },
+    {
+      "train_step": 3630,
+      "loss": 0.5795694587554553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5795694587554553
+    },
+    {
+      "train_step": 3631,
+      "loss": 0.6684846196634423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6684846196634423
+    },
+    {
+      "train_step": 3632,
+      "loss": 0.5574520992835199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5574520992835199
+    },
+    {
+      "train_step": 3633,
+      "loss": 0.4499759442386684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4499759442386684
+    },
+    {
+      "train_step": 3634,
+      "loss": 0.42964889057932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42964889057932
+    },
+    {
+      "train_step": 3635,
+      "loss": 0.42824446552957096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42824446552957096
+    },
+    {
+      "train_step": 3636,
+      "loss": 0.5718174816585416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5718174816585416
+    },
+    {
+      "train_step": 3637,
+      "loss": 0.6429032415910484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6429032415910484
+    },
+    {
+      "train_step": 3638,
+      "loss": 0.6194697272509574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6194697272509574
+    },
+    {
+      "train_step": 3639,
+      "loss": 0.570308227943854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.570308227943854
+    },
+    {
+      "train_step": 3640,
+      "loss": 0.6666441276728027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6666441276728027
+    },
+    {
+      "train_step": 3641,
+      "loss": 0.44495032979221993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44495032979221993
+    },
+    {
+      "train_step": 3642,
+      "loss": 0.43954375768624904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43954375768624904
+    },
+    {
+      "train_step": 3643,
+      "loss": 0.5108819451158801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5108819451158801
+    },
+    {
+      "train_step": 3644,
+      "loss": 0.7320644514375622,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7320644514375622
+    },
+    {
+      "train_step": 3645,
+      "loss": 0.588463231078196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.588463231078196
+    },
+    {
+      "train_step": 3646,
+      "loss": 0.5720811528292765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5720811528292765
+    },
+    {
+      "train_step": 3647,
+      "loss": 0.5162118618154554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5162118618154554
+    },
+    {
+      "train_step": 3648,
+      "loss": 0.525543760559396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.525543760559396
+    },
+    {
+      "train_step": 3649,
+      "loss": 0.5264698976533443,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5264698976533443
+    },
+    {
+      "train_step": 3650,
+      "loss": 0.5041538428838004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5041538428838004
+    },
+    {
+      "train_step": 3651,
+      "loss": 0.5540040451009758,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5540040451009758
+    },
+    {
+      "train_step": 3652,
+      "loss": 0.5360156879203186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5360156879203186
+    },
+    {
+      "train_step": 3653,
+      "loss": 0.522576959021166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.522576959021166
+    },
+    {
+      "train_step": 3654,
+      "loss": 0.48250328283581273,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48250328283581273
+    },
+    {
+      "train_step": 3655,
+      "loss": 0.5156323223848076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5156323223848076
+    },
+    {
+      "train_step": 3656,
+      "loss": 0.5430211580340993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5430211580340993
+    },
+    {
+      "train_step": 3657,
+      "loss": 0.5482990006004045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5482990006004045
+    },
+    {
+      "train_step": 3658,
+      "loss": 0.593099779975396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.593099779975396
+    },
+    {
+      "train_step": 3659,
+      "loss": 0.5059853090663131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5059853090663131
+    },
+    {
+      "train_step": 3660,
+      "loss": 0.48439324479935425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48439324479935425
+    },
+    {
+      "train_step": 3661,
+      "loss": 0.5259490873698556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5259490873698556
+    },
+    {
+      "train_step": 3662,
+      "loss": 0.6659721090620497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6659721090620497
+    },
+    {
+      "train_step": 3663,
+      "loss": 0.517718243571402,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.517718243571402
+    },
+    {
+      "train_step": 3664,
+      "loss": 0.6714479816310515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6714479816310515
+    },
+    {
+      "train_step": 3665,
+      "loss": 0.39878092102204865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39878092102204865
+    },
+    {
+      "train_step": 3666,
+      "loss": 0.6370463787795111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6370463787795111
+    },
+    {
+      "train_step": 3667,
+      "loss": 0.5035861322647904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035861322647904
+    },
+    {
+      "train_step": 3668,
+      "loss": 0.43289810196237605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43289810196237605
+    },
+    {
+      "train_step": 3669,
+      "loss": 0.4515700331139631,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4515700331139631
+    },
+    {
+      "train_step": 3670,
+      "loss": 0.5251258761566497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5251258761566497
+    },
+    {
+      "train_step": 3671,
+      "loss": 0.42120497291795667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42120497291795667
+    },
+    {
+      "train_step": 3672,
+      "loss": 0.40888136484582593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40888136484582593
+    },
+    {
+      "train_step": 3673,
+      "loss": 0.7060511010346836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7060511010346836
+    },
+    {
+      "train_step": 3674,
+      "loss": 0.5725148592930632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5725148592930632
+    },
+    {
+      "train_step": 3675,
+      "loss": 0.43002186868345604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43002186868345604
+    },
+    {
+      "train_step": 3676,
+      "loss": 0.7354235125144803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7354235125144803
+    },
+    {
+      "train_step": 3677,
+      "loss": 0.7749006046746993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7749006046746993
+    },
+    {
+      "train_step": 3678,
+      "loss": 0.5500631259818467,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5500631259818467
+    },
+    {
+      "train_step": 3679,
+      "loss": 0.6238928594126164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6238928594126164
+    },
+    {
+      "train_step": 3680,
+      "loss": 0.553347448584924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.553347448584924
+    },
+    {
+      "train_step": 3681,
+      "loss": 0.6027816120205739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6027816120205739
+    },
+    {
+      "train_step": 3682,
+      "loss": 0.6640836632137264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6640836632137264
+    },
+    {
+      "train_step": 3683,
+      "loss": 0.7482917730230162,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7482917730230162
+    },
+    {
+      "train_step": 3684,
+      "loss": 0.6042328806158659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6042328806158659
+    },
+    {
+      "train_step": 3685,
+      "loss": 0.323057691870547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.323057691870547
+    },
+    {
+      "train_step": 3686,
+      "loss": 0.6892948775009832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6892948775009832
+    },
+    {
+      "train_step": 3687,
+      "loss": 0.5289705502810808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5289705502810808
+    },
+    {
+      "train_step": 3688,
+      "loss": 0.4448209020126244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4448209020126244
+    },
+    {
+      "train_step": 3689,
+      "loss": 0.669792332961459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.669792332961459
+    },
+    {
+      "train_step": 3690,
+      "loss": 0.5340389448500407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5340389448500407
+    },
+    {
+      "train_step": 3691,
+      "loss": 0.6661005867164479,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6661005867164479
+    },
+    {
+      "train_step": 3692,
+      "loss": 0.7492941767392834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7492941767392834
+    },
+    {
+      "train_step": 3693,
+      "loss": 0.5225673421274062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5225673421274062
+    },
+    {
+      "train_step": 3694,
+      "loss": 0.7192966464100274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7192966464100274
+    },
+    {
+      "train_step": 3695,
+      "loss": 0.542914448601654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.542914448601654
+    },
+    {
+      "train_step": 3696,
+      "loss": 0.5274380033596341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5274380033596341
+    },
+    {
+      "train_step": 3697,
+      "loss": 0.6079800165165804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6079800165165804
+    },
+    {
+      "train_step": 3698,
+      "loss": 0.5307104436808793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5307104436808793
+    },
+    {
+      "train_step": 3699,
+      "loss": 0.5858761388341639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5858761388341639
+    },
+    {
+      "train_step": 3700,
+      "loss": 0.466445650119814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.466445650119814
+    },
+    {
+      "train_step": 3701,
+      "loss": 0.5652123134610065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5652123134610065
+    },
+    {
+      "train_step": 3702,
+      "loss": 0.8262730946172699,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8262730946172699
+    },
+    {
+      "train_step": 3703,
+      "loss": 0.7095923614051011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7095923614051011
+    },
+    {
+      "train_step": 3704,
+      "loss": 0.4222717319689465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4222717319689465
+    },
+    {
+      "train_step": 3705,
+      "loss": 0.6103479616907759,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6103479616907759
+    },
+    {
+      "train_step": 3706,
+      "loss": 0.6989842148524449,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6989842148524449
+    },
+    {
+      "train_step": 3707,
+      "loss": 0.7541462855831688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7541462855831688
+    },
+    {
+      "train_step": 3708,
+      "loss": 0.517828593686226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.517828593686226
+    },
+    {
+      "train_step": 3709,
+      "loss": 0.5984010187492808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5984010187492808
+    },
+    {
+      "train_step": 3710,
+      "loss": 0.5933426100130287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5933426100130287
+    },
+    {
+      "train_step": 3711,
+      "loss": 0.732003140490675,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.732003140490675
+    },
+    {
+      "train_step": 3712,
+      "loss": 0.6176982397534905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6176982397534905
+    },
+    {
+      "train_step": 3713,
+      "loss": 0.6745840927152652,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6745840927152652
+    },
+    {
+      "train_step": 3714,
+      "loss": 0.5079298845591685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5079298845591685
+    },
+    {
+      "train_step": 3715,
+      "loss": 0.45797810630311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45797810630311
+    },
+    {
+      "train_step": 3716,
+      "loss": 0.6701624801397135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6701624801397135
+    },
+    {
+      "train_step": 3717,
+      "loss": 0.6450943870050313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6450943870050313
+    },
+    {
+      "train_step": 3718,
+      "loss": 0.41505242836729067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41505242836729067
+    },
+    {
+      "train_step": 3719,
+      "loss": 0.5093341632030166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5093341632030166
+    },
+    {
+      "train_step": 3720,
+      "loss": 0.4887607586949386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4887607586949386
+    },
+    {
+      "train_step": 3721,
+      "loss": 0.4267271266249629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4267271266249629
+    },
+    {
+      "train_step": 3722,
+      "loss": 0.5773771347285538,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5773771347285538
+    },
+    {
+      "train_step": 3723,
+      "loss": 0.9207775242700268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9207775242700268
+    },
+    {
+      "train_step": 3724,
+      "loss": 0.6205690703552235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6205690703552235
+    },
+    {
+      "train_step": 3725,
+      "loss": 0.4134818793507345,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4134818793507345
+    },
+    {
+      "train_step": 3726,
+      "loss": 0.4991170215429231,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4991170215429231
+    },
+    {
+      "train_step": 3727,
+      "loss": 0.43360742658233903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43360742658233903
+    },
+    {
+      "train_step": 3728,
+      "loss": 0.48485704341358704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48485704341358704
+    },
+    {
+      "train_step": 3729,
+      "loss": 0.7894801562584564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7894801562584564
+    },
+    {
+      "train_step": 3730,
+      "loss": 0.6653744160070196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6653744160070196
+    },
+    {
+      "train_step": 3731,
+      "loss": 0.46021328217233876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46021328217233876
+    },
+    {
+      "train_step": 3732,
+      "loss": 0.6582636161144262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6582636161144262
+    },
+    {
+      "train_step": 3733,
+      "loss": 0.4517789814436406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4517789814436406
+    },
+    {
+      "train_step": 3734,
+      "loss": 0.3819094583190678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3819094583190678
+    },
+    {
+      "train_step": 3735,
+      "loss": 0.48384804142144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48384804142144
+    },
+    {
+      "train_step": 3736,
+      "loss": 0.5220919583256585,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5220919583256585
+    },
+    {
+      "train_step": 3737,
+      "loss": 0.3503529201316094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3503529201316094
+    },
+    {
+      "train_step": 3738,
+      "loss": 0.5565264052443784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5565264052443784
+    },
+    {
+      "train_step": 3739,
+      "loss": 0.591665538893379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.591665538893379
+    },
+    {
+      "train_step": 3740,
+      "loss": 0.3838939453208834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3838939453208834
+    },
+    {
+      "train_step": 3741,
+      "loss": 0.6143570259467296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6143570259467296
+    },
+    {
+      "train_step": 3742,
+      "loss": 0.64064602655284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.64064602655284
+    },
+    {
+      "train_step": 3743,
+      "loss": 0.6236844918718862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6236844918718862
+    },
+    {
+      "train_step": 3744,
+      "loss": 0.5723589228157356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5723589228157356
+    },
+    {
+      "train_step": 3745,
+      "loss": 0.31376877946644816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31376877946644816
+    },
+    {
+      "train_step": 3746,
+      "loss": 0.4948694399413788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4948694399413788
+    },
+    {
+      "train_step": 3747,
+      "loss": 0.6011478286648347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6011478286648347
+    },
+    {
+      "train_step": 3748,
+      "loss": 0.44203402660063057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44203402660063057
+    },
+    {
+      "train_step": 3749,
+      "loss": 0.5828664318191512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5828664318191512
+    },
+    {
+      "train_step": 3750,
+      "loss": 0.6833116708192637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6833116708192637
+    },
+    {
+      "train_step": 3751,
+      "loss": 0.4842967357327905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4842967357327905
+    },
+    {
+      "train_step": 3752,
+      "loss": 0.5320785139974202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5320785139974202
+    },
+    {
+      "train_step": 3753,
+      "loss": 0.5128464342674754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5128464342674754
+    },
+    {
+      "train_step": 3754,
+      "loss": 0.4400086151784812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4400086151784812
+    },
+    {
+      "train_step": 3755,
+      "loss": 0.7285175653406484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7285175653406484
+    },
+    {
+      "train_step": 3756,
+      "loss": 0.6951545481768574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6951545481768574
+    },
+    {
+      "train_step": 3757,
+      "loss": 0.49992715431212575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49992715431212575
+    },
+    {
+      "train_step": 3758,
+      "loss": 0.4338005469561517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4338005469561517
+    },
+    {
+      "train_step": 3759,
+      "loss": 0.5564183230367189,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5564183230367189
+    },
+    {
+      "train_step": 3760,
+      "loss": 0.760701540505335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.760701540505335
+    },
+    {
+      "train_step": 3761,
+      "loss": 0.4158526498073881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4158526498073881
+    },
+    {
+      "train_step": 3762,
+      "loss": 0.6056745413332204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6056745413332204
+    },
+    {
+      "train_step": 3763,
+      "loss": 0.5200956252455476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5200956252455476
+    },
+    {
+      "train_step": 3764,
+      "loss": 0.6591870517946712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6591870517946712
+    },
+    {
+      "train_step": 3765,
+      "loss": 0.7913437918018198,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7913437918018198
+    },
+    {
+      "train_step": 3766,
+      "loss": 0.4801122912755674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4801122912755674
+    },
+    {
+      "train_step": 3767,
+      "loss": 0.7708328127417158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7708328127417158
+    },
+    {
+      "train_step": 3768,
+      "loss": 0.6644469791261931,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6644469791261931
+    },
+    {
+      "train_step": 3769,
+      "loss": 0.5129681506612822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5129681506612822
+    },
+    {
+      "train_step": 3770,
+      "loss": 0.6089834194132264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6089834194132264
+    },
+    {
+      "train_step": 3771,
+      "loss": 0.7091287828695779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7091287828695779
+    },
+    {
+      "train_step": 3772,
+      "loss": 0.6052703590676031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6052703590676031
+    },
+    {
+      "train_step": 3773,
+      "loss": 0.5927576611854545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5927576611854545
+    },
+    {
+      "train_step": 3774,
+      "loss": 0.489643304817064,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.489643304817064
+    },
+    {
+      "train_step": 3775,
+      "loss": 0.6355653865432186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6355653865432186
+    },
+    {
+      "train_step": 3776,
+      "loss": 0.5317231391074214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5317231391074214
+    },
+    {
+      "train_step": 3777,
+      "loss": 0.5523645656072541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5523645656072541
+    },
+    {
+      "train_step": 3778,
+      "loss": 0.48397605627091594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48397605627091594
+    },
+    {
+      "train_step": 3779,
+      "loss": 0.4557472782590124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4557472782590124
+    },
+    {
+      "train_step": 3780,
+      "loss": 0.6603810289022565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6603810289022565
+    },
+    {
+      "train_step": 3781,
+      "loss": 0.5744173811283119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5744173811283119
+    },
+    {
+      "train_step": 3782,
+      "loss": 0.5010041351450463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5010041351450463
+    },
+    {
+      "train_step": 3783,
+      "loss": 0.4730530643065685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4730530643065685
+    },
+    {
+      "train_step": 3784,
+      "loss": 0.6316153679905073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6316153679905073
+    },
+    {
+      "train_step": 3785,
+      "loss": 0.5168335496424511,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5168335496424511
+    },
+    {
+      "train_step": 3786,
+      "loss": 0.41108050193590284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41108050193590284
+    },
+    {
+      "train_step": 3787,
+      "loss": 0.43796425689740187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43796425689740187
+    },
+    {
+      "train_step": 3788,
+      "loss": 0.49806789465311785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49806789465311785
+    },
+    {
+      "train_step": 3789,
+      "loss": 0.518322954933792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.518322954933792
+    },
+    {
+      "train_step": 3790,
+      "loss": 0.6245383185850312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6245383185850312
+    },
+    {
+      "train_step": 3791,
+      "loss": 0.6362639107215062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6362639107215062
+    },
+    {
+      "train_step": 3792,
+      "loss": 0.6028588904272206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6028588904272206
+    },
+    {
+      "train_step": 3793,
+      "loss": 0.46958225701428497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46958225701428497
+    },
+    {
+      "train_step": 3794,
+      "loss": 0.4985053336820927,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4985053336820927
+    },
+    {
+      "train_step": 3795,
+      "loss": 0.4416808296032769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4416808296032769
+    },
+    {
+      "train_step": 3796,
+      "loss": 0.5020844627264731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5020844627264731
+    },
+    {
+      "train_step": 3797,
+      "loss": 0.5116907851878475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5116907851878475
+    },
+    {
+      "train_step": 3798,
+      "loss": 0.5944539788568199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5944539788568199
+    },
+    {
+      "train_step": 3799,
+      "loss": 0.5604319977340402,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5604319977340402
+    },
+    {
+      "train_step": 3800,
+      "loss": 0.7123462909117164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7123462909117164
+    },
+    {
+      "train_step": 3801,
+      "loss": 0.4246097855206747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4246097855206747
+    },
+    {
+      "train_step": 3802,
+      "loss": 0.47913074825923924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47913074825923924
+    },
+    {
+      "train_step": 3803,
+      "loss": 0.5769863822626584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5769863822626584
+    },
+    {
+      "train_step": 3804,
+      "loss": 0.5013955202692716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5013955202692716
+    },
+    {
+      "train_step": 3805,
+      "loss": 0.7744946888392076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7744946888392076
+    },
+    {
+      "train_step": 3806,
+      "loss": 0.5136849937628934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5136849937628934
+    },
+    {
+      "train_step": 3807,
+      "loss": 0.388657910040912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.388657910040912
+    },
+    {
+      "train_step": 3808,
+      "loss": 0.5438928547742545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5438928547742545
+    },
+    {
+      "train_step": 3809,
+      "loss": 0.5777517941590794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5777517941590794
+    },
+    {
+      "train_step": 3810,
+      "loss": 0.5422109169735746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5422109169735746
+    },
+    {
+      "train_step": 3811,
+      "loss": 0.6390242719152454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6390242719152454
+    },
+    {
+      "train_step": 3812,
+      "loss": 0.510960783764975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.510960783764975
+    },
+    {
+      "train_step": 3813,
+      "loss": 0.7285560385957603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7285560385957603
+    },
+    {
+      "train_step": 3814,
+      "loss": 0.3546015445971457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3546015445971457
+    },
+    {
+      "train_step": 3815,
+      "loss": 0.3597879391976077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3597879391976077
+    },
+    {
+      "train_step": 3816,
+      "loss": 0.7052981334591054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7052981334591054
+    },
+    {
+      "train_step": 3817,
+      "loss": 0.4990739566022882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4990739566022882
+    },
+    {
+      "train_step": 3818,
+      "loss": 0.6368283361718546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6368283361718546
+    },
+    {
+      "train_step": 3819,
+      "loss": 0.5763823886993344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5763823886993344
+    },
+    {
+      "train_step": 3820,
+      "loss": 0.5482577230500514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5482577230500514
+    },
+    {
+      "train_step": 3821,
+      "loss": 0.6128505758929593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6128505758929593
+    },
+    {
+      "train_step": 3822,
+      "loss": 0.4586587583582592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4586587583582592
+    },
+    {
+      "train_step": 3823,
+      "loss": 0.5552323830231031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5552323830231031
+    },
+    {
+      "train_step": 3824,
+      "loss": 0.6152225469656687,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6152225469656687
+    },
+    {
+      "train_step": 3825,
+      "loss": 0.47001440938426947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47001440938426947
+    },
+    {
+      "train_step": 3826,
+      "loss": 0.733164709806573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.733164709806573
+    },
+    {
+      "train_step": 3827,
+      "loss": 0.7170436681442296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7170436681442296
+    },
+    {
+      "train_step": 3828,
+      "loss": 0.41279439190767847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41279439190767847
+    },
+    {
+      "train_step": 3829,
+      "loss": 0.6556171443512128,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6556171443512128
+    },
+    {
+      "train_step": 3830,
+      "loss": 0.36529803636995134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36529803636995134
+    },
+    {
+      "train_step": 3831,
+      "loss": 0.48937744409530926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48937744409530926
+    },
+    {
+      "train_step": 3832,
+      "loss": 0.427005766849397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.427005766849397
+    },
+    {
+      "train_step": 3833,
+      "loss": 0.701298295406439,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.701298295406439
+    },
+    {
+      "train_step": 3834,
+      "loss": 0.6787882305285716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6787882305285716
+    },
+    {
+      "train_step": 3835,
+      "loss": 0.6100185825381739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6100185825381739
+    },
+    {
+      "train_step": 3836,
+      "loss": 0.4561076953967963,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4561076953967963
+    },
+    {
+      "train_step": 3837,
+      "loss": 0.6024807420996501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6024807420996501
+    },
+    {
+      "train_step": 3838,
+      "loss": 0.5570512933977507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5570512933977507
+    },
+    {
+      "train_step": 3839,
+      "loss": 0.5399060173885153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5399060173885153
+    },
+    {
+      "train_step": 3840,
+      "loss": 0.44535236168708414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44535236168708414
+    },
+    {
+      "train_step": 3841,
+      "loss": 0.5685091498626487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5685091498626487
+    },
+    {
+      "train_step": 3842,
+      "loss": 0.5145351226470084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5145351226470084
+    },
+    {
+      "train_step": 3843,
+      "loss": 0.6504051785448961,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6504051785448961
+    },
+    {
+      "train_step": 3844,
+      "loss": 0.6422642156320212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6422642156320212
+    },
+    {
+      "train_step": 3845,
+      "loss": 0.5499158400767564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5499158400767564
+    },
+    {
+      "train_step": 3846,
+      "loss": 0.48942789783911966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48942789783911966
+    },
+    {
+      "train_step": 3847,
+      "loss": 0.49066322473474344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49066322473474344
+    },
+    {
+      "train_step": 3848,
+      "loss": 0.3981476859930763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3981476859930763
+    },
+    {
+      "train_step": 3849,
+      "loss": 0.6815232377607559,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6815232377607559
+    },
+    {
+      "train_step": 3850,
+      "loss": 0.6080076241445337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6080076241445337
+    },
+    {
+      "train_step": 3851,
+      "loss": 0.46258315684728124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46258315684728124
+    },
+    {
+      "train_step": 3852,
+      "loss": 0.5531769677893807,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5531769677893807
+    },
+    {
+      "train_step": 3853,
+      "loss": 0.5252489365037892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5252489365037892
+    },
+    {
+      "train_step": 3854,
+      "loss": 0.723305586178677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.723305586178677
+    },
+    {
+      "train_step": 3855,
+      "loss": 0.528124043167332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.528124043167332
+    },
+    {
+      "train_step": 3856,
+      "loss": 0.5869058124716163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5869058124716163
+    },
+    {
+      "train_step": 3857,
+      "loss": 0.7384386942989175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7384386942989175
+    },
+    {
+      "train_step": 3858,
+      "loss": 0.391989322384631,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.391989322384631
+    },
+    {
+      "train_step": 3859,
+      "loss": 0.5568980800837312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5568980800837312
+    },
+    {
+      "train_step": 3860,
+      "loss": 0.4973083380501536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4973083380501536
+    },
+    {
+      "train_step": 3861,
+      "loss": 0.5852036414061684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5852036414061684
+    },
+    {
+      "train_step": 3862,
+      "loss": 0.5920778202885683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5920778202885683
+    },
+    {
+      "train_step": 3863,
+      "loss": 0.4274807538802796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4274807538802796
+    },
+    {
+      "train_step": 3864,
+      "loss": 0.529297725023649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.529297725023649
+    },
+    {
+      "train_step": 3865,
+      "loss": 0.5583693739131743,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5583693739131743
+    },
+    {
+      "train_step": 3866,
+      "loss": 0.5712801936411075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5712801936411075
+    },
+    {
+      "train_step": 3867,
+      "loss": 0.42063220734486206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42063220734486206
+    },
+    {
+      "train_step": 3868,
+      "loss": 0.45930040983550563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45930040983550563
+    },
+    {
+      "train_step": 3869,
+      "loss": 0.42753904736061277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42753904736061277
+    },
+    {
+      "train_step": 3870,
+      "loss": 0.8137993552648423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8137993552648423
+    },
+    {
+      "train_step": 3871,
+      "loss": 0.6553199714480276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6553199714480276
+    },
+    {
+      "train_step": 3872,
+      "loss": 0.46921017196835624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46921017196835624
+    },
+    {
+      "train_step": 3873,
+      "loss": 0.7038596343511172,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7038596343511172
+    },
+    {
+      "train_step": 3874,
+      "loss": 0.4972014045801058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4972014045801058
+    },
+    {
+      "train_step": 3875,
+      "loss": 0.5686427406096157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5686427406096157
+    },
+    {
+      "train_step": 3876,
+      "loss": 0.5225019467956681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5225019467956681
+    },
+    {
+      "train_step": 3877,
+      "loss": 0.5881375211891732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5881375211891732
+    },
+    {
+      "train_step": 3878,
+      "loss": 0.4120863181460106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4120863181460106
+    },
+    {
+      "train_step": 3879,
+      "loss": 0.6674417899614162,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6674417899614162
+    },
+    {
+      "train_step": 3880,
+      "loss": 0.7208479772279418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7208479772279418
+    },
+    {
+      "train_step": 3881,
+      "loss": 0.5216824916625583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5216824916625583
+    },
+    {
+      "train_step": 3882,
+      "loss": 0.5225093711437073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5225093711437073
+    },
+    {
+      "train_step": 3883,
+      "loss": 0.39131419317648886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39131419317648886
+    },
+    {
+      "train_step": 3884,
+      "loss": 0.4741645285901682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4741645285901682
+    },
+    {
+      "train_step": 3885,
+      "loss": 0.4028122976840029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4028122976840029
+    },
+    {
+      "train_step": 3886,
+      "loss": 0.6796096070445328,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6796096070445328
+    },
+    {
+      "train_step": 3887,
+      "loss": 0.5983511457491575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5983511457491575
+    },
+    {
+      "train_step": 3888,
+      "loss": 0.5880606473778147,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5880606473778147
+    },
+    {
+      "train_step": 3889,
+      "loss": 0.625971137041657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.625971137041657
+    },
+    {
+      "train_step": 3890,
+      "loss": 0.5358683391028002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5358683391028002
+    },
+    {
+      "train_step": 3891,
+      "loss": 0.3346170714593945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3346170714593945
+    },
+    {
+      "train_step": 3892,
+      "loss": 0.6766079914085813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6766079914085813
+    },
+    {
+      "train_step": 3893,
+      "loss": 0.5360922258176793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5360922258176793
+    },
+    {
+      "train_step": 3894,
+      "loss": 0.6873062307110763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6873062307110763
+    },
+    {
+      "train_step": 3895,
+      "loss": 0.5557666470488452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5557666470488452
+    },
+    {
+      "train_step": 3896,
+      "loss": 0.3479305983309686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3479305983309686
+    },
+    {
+      "train_step": 3897,
+      "loss": 0.3890762250161638,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3890762250161638
+    },
+    {
+      "train_step": 3898,
+      "loss": 0.5200750518859312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5200750518859312
+    },
+    {
+      "train_step": 3899,
+      "loss": 0.6013432559581701,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6013432559581701
+    },
+    {
+      "train_step": 3900,
+      "loss": 0.6378641018899869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6378641018899869
+    },
+    {
+      "train_step": 3901,
+      "loss": 0.5331812281119173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5331812281119173
+    },
+    {
+      "train_step": 3902,
+      "loss": 0.7117232061795284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7117232061795284
+    },
+    {
+      "train_step": 3903,
+      "loss": 0.5228510779330261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5228510779330261
+    },
+    {
+      "train_step": 3904,
+      "loss": 0.47665921109479614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47665921109479614
+    },
+    {
+      "train_step": 3905,
+      "loss": 0.5339760005515831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5339760005515831
+    },
+    {
+      "train_step": 3906,
+      "loss": 0.6611357813254827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6611357813254827
+    },
+    {
+      "train_step": 3907,
+      "loss": 0.5736190663971962,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5736190663971962
+    },
+    {
+      "train_step": 3908,
+      "loss": 0.6938002116396167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6938002116396167
+    },
+    {
+      "train_step": 3909,
+      "loss": 0.5725776900987107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5725776900987107
+    },
+    {
+      "train_step": 3910,
+      "loss": 0.5139733361721773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5139733361721773
+    },
+    {
+      "train_step": 3911,
+      "loss": 0.5757785378196751,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5757785378196751
+    },
+    {
+      "train_step": 3912,
+      "loss": 0.5344569469192453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5344569469192453
+    },
+    {
+      "train_step": 3913,
+      "loss": 0.6327814649500608,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6327814649500608
+    },
+    {
+      "train_step": 3914,
+      "loss": 0.47312121979755867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47312121979755867
+    },
+    {
+      "train_step": 3915,
+      "loss": 0.5387328674748156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5387328674748156
+    },
+    {
+      "train_step": 3916,
+      "loss": 0.513009538786865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.513009538786865
+    },
+    {
+      "train_step": 3917,
+      "loss": 0.7517754687776437,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7517754687776437
+    },
+    {
+      "train_step": 3918,
+      "loss": 0.48992100892519,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48992100892519
+    },
+    {
+      "train_step": 3919,
+      "loss": 0.6302657420847904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6302657420847904
+    },
+    {
+      "train_step": 3920,
+      "loss": 0.5768613835477198,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5768613835477198
+    },
+    {
+      "train_step": 3921,
+      "loss": 0.5839473605270283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5839473605270283
+    },
+    {
+      "train_step": 3922,
+      "loss": 0.6227489921341604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6227489921341604
+    },
+    {
+      "train_step": 3923,
+      "loss": 0.7850903896199212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7850903896199212
+    },
+    {
+      "train_step": 3924,
+      "loss": 0.48479913352583653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48479913352583653
+    },
+    {
+      "train_step": 3925,
+      "loss": 0.7164357654197598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7164357654197598
+    },
+    {
+      "train_step": 3926,
+      "loss": 0.5694366122844533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5694366122844533
+    },
+    {
+      "train_step": 3927,
+      "loss": 0.2784559718978865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2784559718978865
+    },
+    {
+      "train_step": 3928,
+      "loss": 0.5375154252442793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5375154252442793
+    },
+    {
+      "train_step": 3929,
+      "loss": 0.5440379185972893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5440379185972893
+    },
+    {
+      "train_step": 3930,
+      "loss": 0.658541696045455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.658541696045455
+    },
+    {
+      "train_step": 3931,
+      "loss": 0.34217691750661433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34217691750661433
+    },
+    {
+      "train_step": 3932,
+      "loss": 0.5179038746061393,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5179038746061393
+    },
+    {
+      "train_step": 3933,
+      "loss": 0.7611929935745698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7611929935745698
+    },
+    {
+      "train_step": 3934,
+      "loss": 0.6634040359224642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6634040359224642
+    },
+    {
+      "train_step": 3935,
+      "loss": 0.4910345052318399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4910345052318399
+    },
+    {
+      "train_step": 3936,
+      "loss": 0.568235821719076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.568235821719076
+    },
+    {
+      "train_step": 3937,
+      "loss": 0.5633299426264936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5633299426264936
+    },
+    {
+      "train_step": 3938,
+      "loss": 0.4903761548412586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4903761548412586
+    },
+    {
+      "train_step": 3939,
+      "loss": 0.6048588289868126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6048588289868126
+    },
+    {
+      "train_step": 3940,
+      "loss": 0.5808315730144349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5808315730144349
+    },
+    {
+      "train_step": 3941,
+      "loss": 0.7114007416717882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7114007416717882
+    },
+    {
+      "train_step": 3942,
+      "loss": 0.4857010623573296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4857010623573296
+    },
+    {
+      "train_step": 3943,
+      "loss": 0.4649779182602434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4649779182602434
+    },
+    {
+      "train_step": 3944,
+      "loss": 0.5945857369449434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5945857369449434
+    },
+    {
+      "train_step": 3945,
+      "loss": 0.5092870826741941,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5092870826741941
+    },
+    {
+      "train_step": 3946,
+      "loss": 0.6692259706667394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6692259706667394
+    },
+    {
+      "train_step": 3947,
+      "loss": 0.626375896522511,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.626375896522511
+    },
+    {
+      "train_step": 3948,
+      "loss": 0.6518430613496285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6518430613496285
+    },
+    {
+      "train_step": 3949,
+      "loss": 0.770104091232053,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.770104091232053
+    },
+    {
+      "train_step": 3950,
+      "loss": 0.5902674492309611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5902674492309611
+    },
+    {
+      "train_step": 3951,
+      "loss": 0.4724465432097739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4724465432097739
+    },
+    {
+      "train_step": 3952,
+      "loss": 0.24262039107988928,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.24262039107988928
+    },
+    {
+      "train_step": 3953,
+      "loss": 0.4547700414139344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4547700414139344
+    },
+    {
+      "train_step": 3954,
+      "loss": 0.6179731776719133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6179731776719133
+    },
+    {
+      "train_step": 3955,
+      "loss": 0.6324049464384717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6324049464384717
+    },
+    {
+      "train_step": 3956,
+      "loss": 0.7491465848108565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7491465848108565
+    },
+    {
+      "train_step": 3957,
+      "loss": 0.7359051265396674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7359051265396674
+    },
+    {
+      "train_step": 3958,
+      "loss": 0.6527736628137867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6527736628137867
+    },
+    {
+      "train_step": 3959,
+      "loss": 0.7616376665402056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7616376665402056
+    },
+    {
+      "train_step": 3960,
+      "loss": 0.5370299067927307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5370299067927307
+    },
+    {
+      "train_step": 3961,
+      "loss": 0.5892579862479571,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5892579862479571
+    },
+    {
+      "train_step": 3962,
+      "loss": 0.45184741359630193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45184741359630193
+    },
+    {
+      "train_step": 3963,
+      "loss": 0.6456338326398193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6456338326398193
+    },
+    {
+      "train_step": 3964,
+      "loss": 0.6264265236031881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6264265236031881
+    },
+    {
+      "train_step": 3965,
+      "loss": 0.47122785981847837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47122785981847837
+    },
+    {
+      "train_step": 3966,
+      "loss": 0.8110082536324666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8110082536324666
+    },
+    {
+      "train_step": 3967,
+      "loss": 0.5432980583223684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5432980583223684
+    },
+    {
+      "train_step": 3968,
+      "loss": 0.5239815181819545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5239815181819545
+    },
+    {
+      "train_step": 3969,
+      "loss": 0.4743520233061167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4743520233061167
+    },
+    {
+      "train_step": 3970,
+      "loss": 0.4504377971633481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4504377971633481
+    },
+    {
+      "train_step": 3971,
+      "loss": 0.4901175787310064,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4901175787310064
+    },
+    {
+      "train_step": 3972,
+      "loss": 0.6143012712563519,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6143012712563519
+    },
+    {
+      "train_step": 3973,
+      "loss": 0.40359259542783577,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40359259542783577
+    },
+    {
+      "train_step": 3974,
+      "loss": 0.4149057768591975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4149057768591975
+    },
+    {
+      "train_step": 3975,
+      "loss": 0.34039080647943587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34039080647943587
+    },
+    {
+      "train_step": 3976,
+      "loss": 0.5489815702882626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5489815702882626
+    },
+    {
+      "train_step": 3977,
+      "loss": 0.44513928644522627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44513928644522627
+    },
+    {
+      "train_step": 3978,
+      "loss": 0.5840509897586116,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5840509897586116
+    },
+    {
+      "train_step": 3979,
+      "loss": 0.6018725514273293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6018725514273293
+    },
+    {
+      "train_step": 3980,
+      "loss": 0.42038385784910454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42038385784910454
+    },
+    {
+      "train_step": 3981,
+      "loss": 0.6409528530532884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6409528530532884
+    },
+    {
+      "train_step": 3982,
+      "loss": 0.4721652092846963,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4721652092846963
+    },
+    {
+      "train_step": 3983,
+      "loss": 0.5085396322737211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5085396322737211
+    },
+    {
+      "train_step": 3984,
+      "loss": 0.6756284645138958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6756284645138958
+    },
+    {
+      "train_step": 3985,
+      "loss": 0.6373477460197662,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6373477460197662
+    },
+    {
+      "train_step": 3986,
+      "loss": 0.5309992577616689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5309992577616689
+    },
+    {
+      "train_step": 3987,
+      "loss": 0.5349367897748318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5349367897748318
+    },
+    {
+      "train_step": 3988,
+      "loss": 0.636181755240832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.636181755240832
+    },
+    {
+      "train_step": 3989,
+      "loss": 0.565247676959004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.565247676959004
+    },
+    {
+      "train_step": 3990,
+      "loss": 0.6004253817721705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6004253817721705
+    },
+    {
+      "train_step": 3991,
+      "loss": 0.6056710134080416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6056710134080416
+    },
+    {
+      "train_step": 3992,
+      "loss": 0.7902334630504857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7902334630504857
+    },
+    {
+      "train_step": 3993,
+      "loss": 0.5057398437674554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5057398437674554
+    },
+    {
+      "train_step": 3994,
+      "loss": 0.566551295843214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.566551295843214
+    },
+    {
+      "train_step": 3995,
+      "loss": 0.35442065041043164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35442065041043164
+    },
+    {
+      "train_step": 3996,
+      "loss": 0.6700458051526563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6700458051526563
+    },
+    {
+      "train_step": 3997,
+      "loss": 0.5223840946738378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5223840946738378
+    },
+    {
+      "train_step": 3998,
+      "loss": 0.5607965078322431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5607965078322431
+    },
+    {
+      "train_step": 3999,
+      "loss": 0.4934626932779518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4934626932779518
+    },
+    {
+      "train_step": 4000,
+      "loss": 0.7141705450753015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7141705450753015
+    },
+    {
+      "train_step": 4001,
+      "loss": 0.6740870788733584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6740870788733584
+    },
+    {
+      "train_step": 4002,
+      "loss": 0.5551362967362845,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5551362967362845
+    },
+    {
+      "train_step": 4003,
+      "loss": 0.648101735911388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.648101735911388
+    },
+    {
+      "train_step": 4004,
+      "loss": 0.44991897351472626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44991897351472626
+    },
+    {
+      "train_step": 4005,
+      "loss": 0.535609377146514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.535609377146514
+    },
+    {
+      "train_step": 4006,
+      "loss": 0.7228776409177133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7228776409177133
+    },
+    {
+      "train_step": 4007,
+      "loss": 0.6469697369266644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6469697369266644
+    },
+    {
+      "train_step": 4008,
+      "loss": 0.6235543640528934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6235543640528934
+    },
+    {
+      "train_step": 4009,
+      "loss": 0.75762752423851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.75762752423851
+    },
+    {
+      "train_step": 4010,
+      "loss": 0.5003501189067525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5003501189067525
+    },
+    {
+      "train_step": 4011,
+      "loss": 0.4842617077450545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4842617077450545
+    },
+    {
+      "train_step": 4012,
+      "loss": 0.5686253286509939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5686253286509939
+    },
+    {
+      "train_step": 4013,
+      "loss": 0.3404994074827839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3404994074827839
+    },
+    {
+      "train_step": 4014,
+      "loss": 0.43959334858349797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43959334858349797
+    },
+    {
+      "train_step": 4015,
+      "loss": 0.5261477718054035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5261477718054035
+    },
+    {
+      "train_step": 4016,
+      "loss": 0.6398909846371358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6398909846371358
+    },
+    {
+      "train_step": 4017,
+      "loss": 0.37773347504744387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37773347504744387
+    },
+    {
+      "train_step": 4018,
+      "loss": 0.629283263337945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.629283263337945
+    },
+    {
+      "train_step": 4019,
+      "loss": 0.6106184107901648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6106184107901648
+    },
+    {
+      "train_step": 4020,
+      "loss": 0.47500285461638375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47500285461638375
+    },
+    {
+      "train_step": 4021,
+      "loss": 0.5409206766220271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5409206766220271
+    },
+    {
+      "train_step": 4022,
+      "loss": 0.4306692738639297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4306692738639297
+    },
+    {
+      "train_step": 4023,
+      "loss": 0.3053669988762229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3053669988762229
+    },
+    {
+      "train_step": 4024,
+      "loss": 0.7346123777678787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7346123777678787
+    },
+    {
+      "train_step": 4025,
+      "loss": 0.40087950530023614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40087950530023614
+    },
+    {
+      "train_step": 4026,
+      "loss": 0.5553136693979744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5553136693979744
+    },
+    {
+      "train_step": 4027,
+      "loss": 0.3819602446058652,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3819602446058652
+    },
+    {
+      "train_step": 4028,
+      "loss": 0.6191871848648335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6191871848648335
+    },
+    {
+      "train_step": 4029,
+      "loss": 0.679517835753796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.679517835753796
+    },
+    {
+      "train_step": 4030,
+      "loss": 0.678213652027767,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.678213652027767
+    },
+    {
+      "train_step": 4031,
+      "loss": 0.572163454983823,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.572163454983823
+    },
+    {
+      "train_step": 4032,
+      "loss": 0.8419615713042959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8419615713042959
+    },
+    {
+      "train_step": 4033,
+      "loss": 0.4642548004748454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4642548004748454
+    },
+    {
+      "train_step": 4034,
+      "loss": 0.5929114105816118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5929114105816118
+    },
+    {
+      "train_step": 4035,
+      "loss": 0.598233008484592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.598233008484592
+    },
+    {
+      "train_step": 4036,
+      "loss": 0.5012073035346601,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5012073035346601
+    },
+    {
+      "train_step": 4037,
+      "loss": 0.7056118009752618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7056118009752618
+    },
+    {
+      "train_step": 4038,
+      "loss": 0.5266960936665761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5266960936665761
+    },
+    {
+      "train_step": 4039,
+      "loss": 0.5638518512914715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5638518512914715
+    },
+    {
+      "train_step": 4040,
+      "loss": 0.4924066750694325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4924066750694325
+    },
+    {
+      "train_step": 4041,
+      "loss": 0.6439520615073928,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6439520615073928
+    },
+    {
+      "train_step": 4042,
+      "loss": 0.5362388876973784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5362388876973784
+    },
+    {
+      "train_step": 4043,
+      "loss": 0.5026549533356577,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5026549533356577
+    },
+    {
+      "train_step": 4044,
+      "loss": 0.6348281483447876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6348281483447876
+    },
+    {
+      "train_step": 4045,
+      "loss": 0.49283839364135995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49283839364135995
+    },
+    {
+      "train_step": 4046,
+      "loss": 0.6293158234684346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6293158234684346
+    },
+    {
+      "train_step": 4047,
+      "loss": 0.5909807730189388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5909807730189388
+    },
+    {
+      "train_step": 4048,
+      "loss": 0.6758459249059009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6758459249059009
+    },
+    {
+      "train_step": 4049,
+      "loss": 0.48633952388466173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48633952388466173
+    },
+    {
+      "train_step": 4050,
+      "loss": 0.4198208204180481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4198208204180481
+    },
+    {
+      "train_step": 4051,
+      "loss": 0.6771918793714107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6771918793714107
+    },
+    {
+      "train_step": 4052,
+      "loss": 0.6551849491303285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6551849491303285
+    },
+    {
+      "train_step": 4053,
+      "loss": 0.7933576195853852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7933576195853852
+    },
+    {
+      "train_step": 4054,
+      "loss": 0.867642030639834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.867642030639834
+    },
+    {
+      "train_step": 4055,
+      "loss": 0.5156835200565468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5156835200565468
+    },
+    {
+      "train_step": 4056,
+      "loss": 0.6516734276491792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6516734276491792
+    },
+    {
+      "train_step": 4057,
+      "loss": 0.7591033570588761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7591033570588761
+    },
+    {
+      "train_step": 4058,
+      "loss": 0.686740871940497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.686740871940497
+    },
+    {
+      "train_step": 4059,
+      "loss": 0.593260180588091,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.593260180588091
+    },
+    {
+      "train_step": 4060,
+      "loss": 0.6439177102704279,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6439177102704279
+    },
+    {
+      "train_step": 4061,
+      "loss": 0.5271783022014684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5271783022014684
+    },
+    {
+      "train_step": 4062,
+      "loss": 0.5239443453136062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5239443453136062
+    },
+    {
+      "train_step": 4063,
+      "loss": 0.6206077588460635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6206077588460635
+    },
+    {
+      "train_step": 4064,
+      "loss": 0.6415242466615049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6415242466615049
+    },
+    {
+      "train_step": 4065,
+      "loss": 0.5016627213281966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5016627213281966
+    },
+    {
+      "train_step": 4066,
+      "loss": 0.5886040912282456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5886040912282456
+    },
+    {
+      "train_step": 4067,
+      "loss": 0.7490405990990159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7490405990990159
+    },
+    {
+      "train_step": 4068,
+      "loss": 0.7070121727789882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7070121727789882
+    },
+    {
+      "train_step": 4069,
+      "loss": 0.5381063210869387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5381063210869387
+    },
+    {
+      "train_step": 4070,
+      "loss": 0.5343325437418984,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5343325437418984
+    },
+    {
+      "train_step": 4071,
+      "loss": 0.6828766386909196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6828766386909196
+    },
+    {
+      "train_step": 4072,
+      "loss": 0.6422707449598823,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6422707449598823
+    },
+    {
+      "train_step": 4073,
+      "loss": 0.7232099642598324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7232099642598324
+    },
+    {
+      "train_step": 4074,
+      "loss": 0.4488378966537957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4488378966537957
+    },
+    {
+      "train_step": 4075,
+      "loss": 0.7991584453719531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7991584453719531
+    },
+    {
+      "train_step": 4076,
+      "loss": 0.5603838948787484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5603838948787484
+    },
+    {
+      "train_step": 4077,
+      "loss": 0.6164333441713075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6164333441713075
+    },
+    {
+      "train_step": 4078,
+      "loss": 0.6456298476772021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6456298476772021
+    },
+    {
+      "train_step": 4079,
+      "loss": 0.6621523914940001,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6621523914940001
+    },
+    {
+      "train_step": 4080,
+      "loss": 0.5130020444903204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5130020444903204
+    },
+    {
+      "train_step": 4081,
+      "loss": 0.6436939743111244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6436939743111244
+    },
+    {
+      "train_step": 4082,
+      "loss": 0.45490627757371194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45490627757371194
+    },
+    {
+      "train_step": 4083,
+      "loss": 0.6388614107297331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6388614107297331
+    },
+    {
+      "train_step": 4084,
+      "loss": 0.43488255028702755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43488255028702755
+    },
+    {
+      "train_step": 4085,
+      "loss": 0.4469301826202642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4469301826202642
+    },
+    {
+      "train_step": 4086,
+      "loss": 0.5339273018190343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5339273018190343
+    },
+    {
+      "train_step": 4087,
+      "loss": 0.546937310376487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.546937310376487
+    },
+    {
+      "train_step": 4088,
+      "loss": 0.42918066714310327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42918066714310327
+    },
+    {
+      "train_step": 4089,
+      "loss": 0.7671246132647387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7671246132647387
+    },
+    {
+      "train_step": 4090,
+      "loss": 0.6367301298553569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6367301298553569
+    },
+    {
+      "train_step": 4091,
+      "loss": 0.5581178428749229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5581178428749229
+    },
+    {
+      "train_step": 4092,
+      "loss": 0.5874667735080876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5874667735080876
+    },
+    {
+      "train_step": 4093,
+      "loss": 0.742306635821108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.742306635821108
+    },
+    {
+      "train_step": 4094,
+      "loss": 0.5644390350531725,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5644390350531725
+    },
+    {
+      "train_step": 4095,
+      "loss": 0.4352929391737078,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4352929391737078
+    },
+    {
+      "train_step": 4096,
+      "loss": 0.5830551265800952,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5830551265800952
+    },
+    {
+      "train_step": 4097,
+      "loss": 0.7630653527719142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7630653527719142
+    },
+    {
+      "train_step": 4098,
+      "loss": 0.48205505726225834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48205505726225834
+    },
+    {
+      "train_step": 4099,
+      "loss": 0.4863981494416106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4863981494416106
+    },
+    {
+      "train_step": 4100,
+      "loss": 0.6847323945431327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6847323945431327
+    },
+    {
+      "train_step": 4101,
+      "loss": 0.5280245296322912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5280245296322912
+    },
+    {
+      "train_step": 4102,
+      "loss": 0.42641827962715745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42641827962715745
+    },
+    {
+      "train_step": 4103,
+      "loss": 0.6304898507635279,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6304898507635279
+    },
+    {
+      "train_step": 4104,
+      "loss": 0.5621602562101539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5621602562101539
+    },
+    {
+      "train_step": 4105,
+      "loss": 0.6244091055495762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6244091055495762
+    },
+    {
+      "train_step": 4106,
+      "loss": 0.6235997712196346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6235997712196346
+    },
+    {
+      "train_step": 4107,
+      "loss": 0.5251074798064419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5251074798064419
+    },
+    {
+      "train_step": 4108,
+      "loss": 0.6101407317956319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6101407317956319
+    },
+    {
+      "train_step": 4109,
+      "loss": 0.6049769413325682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6049769413325682
+    },
+    {
+      "train_step": 4110,
+      "loss": 0.6411368652493397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6411368652493397
+    },
+    {
+      "train_step": 4111,
+      "loss": 0.7230122977162518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7230122977162518
+    },
+    {
+      "train_step": 4112,
+      "loss": 0.6933413967820111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6933413967820111
+    },
+    {
+      "train_step": 4113,
+      "loss": 0.385553918592995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.385553918592995
+    },
+    {
+      "train_step": 4114,
+      "loss": 0.6242890449054455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6242890449054455
+    },
+    {
+      "train_step": 4115,
+      "loss": 0.42982632067153204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42982632067153204
+    },
+    {
+      "train_step": 4116,
+      "loss": 0.8715931755910105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8715931755910105
+    },
+    {
+      "train_step": 4117,
+      "loss": 0.5863678771635867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5863678771635867
+    },
+    {
+      "train_step": 4118,
+      "loss": 0.6877262437453634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6877262437453634
+    },
+    {
+      "train_step": 4119,
+      "loss": 0.5847006413784068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5847006413784068
+    },
+    {
+      "train_step": 4120,
+      "loss": 0.7072495223865171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7072495223865171
+    },
+    {
+      "train_step": 4121,
+      "loss": 0.5300741976062394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5300741976062394
+    },
+    {
+      "train_step": 4122,
+      "loss": 0.573194408303787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.573194408303787
+    },
+    {
+      "train_step": 4123,
+      "loss": 0.3769119320867569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3769119320867569
+    },
+    {
+      "train_step": 4124,
+      "loss": 0.55334501950312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.55334501950312
+    },
+    {
+      "train_step": 4125,
+      "loss": 0.4692805301960524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4692805301960524
+    },
+    {
+      "train_step": 4126,
+      "loss": 0.6402361398102722,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6402361398102722
+    },
+    {
+      "train_step": 4127,
+      "loss": 0.5139236643010643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5139236643010643
+    },
+    {
+      "train_step": 4128,
+      "loss": 0.6755142975514385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6755142975514385
+    },
+    {
+      "train_step": 4129,
+      "loss": 0.6530870523728469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6530870523728469
+    },
+    {
+      "train_step": 4130,
+      "loss": 0.5629014630443495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5629014630443495
+    },
+    {
+      "train_step": 4131,
+      "loss": 0.3951993145780981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3951993145780981
+    },
+    {
+      "train_step": 4132,
+      "loss": 0.41394116745052645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41394116745052645
+    },
+    {
+      "train_step": 4133,
+      "loss": 0.6552199120943253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6552199120943253
+    },
+    {
+      "train_step": 4134,
+      "loss": 0.6756173275961919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6756173275961919
+    },
+    {
+      "train_step": 4135,
+      "loss": 0.5778305806267169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5778305806267169
+    },
+    {
+      "train_step": 4136,
+      "loss": 0.7584068514986838,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7584068514986838
+    },
+    {
+      "train_step": 4137,
+      "loss": 0.3847761648604553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3847761648604553
+    },
+    {
+      "train_step": 4138,
+      "loss": 0.6697639483143794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6697639483143794
+    },
+    {
+      "train_step": 4139,
+      "loss": 0.7506596956320047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7506596956320047
+    },
+    {
+      "train_step": 4140,
+      "loss": 0.7976757201614827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7976757201614827
+    },
+    {
+      "train_step": 4141,
+      "loss": 0.5119623498307185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5119623498307185
+    },
+    {
+      "train_step": 4142,
+      "loss": 0.6581289823643879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6581289823643879
+    },
+    {
+      "train_step": 4143,
+      "loss": 0.5853470651010665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5853470651010665
+    },
+    {
+      "train_step": 4144,
+      "loss": 0.589294961894933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.589294961894933
+    },
+    {
+      "train_step": 4145,
+      "loss": 0.6043594549486103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6043594549486103
+    },
+    {
+      "train_step": 4146,
+      "loss": 0.42393045452794836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42393045452794836
+    },
+    {
+      "train_step": 4147,
+      "loss": 0.6220507497185045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6220507497185045
+    },
+    {
+      "train_step": 4148,
+      "loss": 0.4787288825112771,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4787288825112771
+    },
+    {
+      "train_step": 4149,
+      "loss": 0.2849381565130009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2849381565130009
+    },
+    {
+      "train_step": 4150,
+      "loss": 0.6984539999019246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6984539999019246
+    },
+    {
+      "train_step": 4151,
+      "loss": 0.5186269722375264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5186269722375264
+    },
+    {
+      "train_step": 4152,
+      "loss": 0.7782708065919596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7782708065919596
+    },
+    {
+      "train_step": 4153,
+      "loss": 0.4877351348265754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4877351348265754
+    },
+    {
+      "train_step": 4154,
+      "loss": 0.5381922224362898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5381922224362898
+    },
+    {
+      "train_step": 4155,
+      "loss": 0.5900992732755475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5900992732755475
+    },
+    {
+      "train_step": 4156,
+      "loss": 0.6027888642900752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6027888642900752
+    },
+    {
+      "train_step": 4157,
+      "loss": 0.3572444636557854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3572444636557854
+    },
+    {
+      "train_step": 4158,
+      "loss": 0.5999392831166547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5999392831166547
+    },
+    {
+      "train_step": 4159,
+      "loss": 0.7178120463584542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7178120463584542
+    },
+    {
+      "train_step": 4160,
+      "loss": 0.5658450983085261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5658450983085261
+    },
+    {
+      "train_step": 4161,
+      "loss": 0.75214537745165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.75214537745165
+    },
+    {
+      "train_step": 4162,
+      "loss": 0.6307889777095081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6307889777095081
+    },
+    {
+      "train_step": 4163,
+      "loss": 0.5630766872788913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5630766872788913
+    },
+    {
+      "train_step": 4164,
+      "loss": 0.6658487229373731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6658487229373731
+    },
+    {
+      "train_step": 4165,
+      "loss": 0.47540872039098697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47540872039098697
+    },
+    {
+      "train_step": 4166,
+      "loss": 0.6213594394199822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6213594394199822
+    },
+    {
+      "train_step": 4167,
+      "loss": 0.6497956248726373,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6497956248726373
+    },
+    {
+      "train_step": 4168,
+      "loss": 0.5388195015515516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5388195015515516
+    },
+    {
+      "train_step": 4169,
+      "loss": 0.7650473238755762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7650473238755762
+    },
+    {
+      "train_step": 4170,
+      "loss": 0.6926140988079098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6926140988079098
+    },
+    {
+      "train_step": 4171,
+      "loss": 0.8441719081373958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8441719081373958
+    },
+    {
+      "train_step": 4172,
+      "loss": 0.5485016869518056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5485016869518056
+    },
+    {
+      "train_step": 4173,
+      "loss": 0.661176105614713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.661176105614713
+    },
+    {
+      "train_step": 4174,
+      "loss": 0.671557469817967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.671557469817967
+    },
+    {
+      "train_step": 4175,
+      "loss": 0.6134706211573049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6134706211573049
+    },
+    {
+      "train_step": 4176,
+      "loss": 0.38680895674400745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38680895674400745
+    },
+    {
+      "train_step": 4177,
+      "loss": 0.554100167112729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.554100167112729
+    },
+    {
+      "train_step": 4178,
+      "loss": 0.4989858962469721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4989858962469721
+    },
+    {
+      "train_step": 4179,
+      "loss": 0.5330254479849016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5330254479849016
+    },
+    {
+      "train_step": 4180,
+      "loss": 0.4264989356566282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4264989356566282
+    },
+    {
+      "train_step": 4181,
+      "loss": 0.6470679445334254,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6470679445334254
+    },
+    {
+      "train_step": 4182,
+      "loss": 0.5646679239933161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5646679239933161
+    },
+    {
+      "train_step": 4183,
+      "loss": 0.4678166979677662,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4678166979677662
+    },
+    {
+      "train_step": 4184,
+      "loss": 0.7735350684024851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7735350684024851
+    },
+    {
+      "train_step": 4185,
+      "loss": 0.5261787727929947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5261787727929947
+    },
+    {
+      "train_step": 4186,
+      "loss": 0.5299127620380518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5299127620380518
+    },
+    {
+      "train_step": 4187,
+      "loss": 0.4965119453523422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4965119453523422
+    },
+    {
+      "train_step": 4188,
+      "loss": 0.44727092286457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44727092286457
+    },
+    {
+      "train_step": 4189,
+      "loss": 0.5921926634222798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5921926634222798
+    },
+    {
+      "train_step": 4190,
+      "loss": 0.3851190953994656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3851190953994656
+    },
+    {
+      "train_step": 4191,
+      "loss": 0.6785119672582762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6785119672582762
+    },
+    {
+      "train_step": 4192,
+      "loss": 0.5637401334832176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5637401334832176
+    },
+    {
+      "train_step": 4193,
+      "loss": 0.6823367551088716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6823367551088716
+    },
+    {
+      "train_step": 4194,
+      "loss": 0.5088400275556195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5088400275556195
+    },
+    {
+      "train_step": 4195,
+      "loss": 0.4407681414397983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4407681414397983
+    },
+    {
+      "train_step": 4196,
+      "loss": 0.5685098303008436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5685098303008436
+    },
+    {
+      "train_step": 4197,
+      "loss": 0.5242603227679797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5242603227679797
+    },
+    {
+      "train_step": 4198,
+      "loss": 0.6576551006929736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6576551006929736
+    },
+    {
+      "train_step": 4199,
+      "loss": 0.5599304854857837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5599304854857837
+    },
+    {
+      "train_step": 4200,
+      "loss": 0.5739166423300206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5739166423300206
+    },
+    {
+      "train_step": 4201,
+      "loss": 0.5840094335826398,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5840094335826398
+    },
+    {
+      "train_step": 4202,
+      "loss": 0.6174485034545222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6174485034545222
+    },
+    {
+      "train_step": 4203,
+      "loss": 0.6075104953330394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6075104953330394
+    },
+    {
+      "train_step": 4204,
+      "loss": 0.4647843319300849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4647843319300849
+    },
+    {
+      "train_step": 4205,
+      "loss": 0.5133949725793883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5133949725793883
+    },
+    {
+      "train_step": 4206,
+      "loss": 0.4940751356745445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4940751356745445
+    },
+    {
+      "train_step": 4207,
+      "loss": 0.606998426955014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.606998426955014
+    },
+    {
+      "train_step": 4208,
+      "loss": 0.35025917725740235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35025917725740235
+    },
+    {
+      "train_step": 4209,
+      "loss": 0.5196211725489283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5196211725489283
+    },
+    {
+      "train_step": 4210,
+      "loss": 0.7080982831101359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7080982831101359
+    },
+    {
+      "train_step": 4211,
+      "loss": 0.4382473743029657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4382473743029657
+    },
+    {
+      "train_step": 4212,
+      "loss": 0.637570129701486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.637570129701486
+    },
+    {
+      "train_step": 4213,
+      "loss": 0.5616668970876932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5616668970876932
+    },
+    {
+      "train_step": 4214,
+      "loss": 0.6201502053341086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6201502053341086
+    },
+    {
+      "train_step": 4215,
+      "loss": 0.5255662516491537,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5255662516491537
+    },
+    {
+      "train_step": 4216,
+      "loss": 0.5242794701605045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5242794701605045
+    },
+    {
+      "train_step": 4217,
+      "loss": 0.6518720422028937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6518720422028937
+    },
+    {
+      "train_step": 4218,
+      "loss": 0.5464820679833637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5464820679833637
+    },
+    {
+      "train_step": 4219,
+      "loss": 0.5427369301799464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5427369301799464
+    },
+    {
+      "train_step": 4220,
+      "loss": 0.4342261155424083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4342261155424083
+    },
+    {
+      "train_step": 4221,
+      "loss": 0.49439347032665126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49439347032665126
+    },
+    {
+      "train_step": 4222,
+      "loss": 0.5413120401616869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5413120401616869
+    },
+    {
+      "train_step": 4223,
+      "loss": 0.517326281104616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.517326281104616
+    },
+    {
+      "train_step": 4224,
+      "loss": 0.6957412295211562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6957412295211562
+    },
+    {
+      "train_step": 4225,
+      "loss": 0.4038004449648532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4038004449648532
+    },
+    {
+      "train_step": 4226,
+      "loss": 0.6439568089583912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6439568089583912
+    },
+    {
+      "train_step": 4227,
+      "loss": 0.5943919248651819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5943919248651819
+    },
+    {
+      "train_step": 4228,
+      "loss": 0.6053715525004115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6053715525004115
+    },
+    {
+      "train_step": 4229,
+      "loss": 0.649667124178448,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.649667124178448
+    },
+    {
+      "train_step": 4230,
+      "loss": 0.47142628614411547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47142628614411547
+    },
+    {
+      "train_step": 4231,
+      "loss": 0.6469301800884618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6469301800884618
+    },
+    {
+      "train_step": 4232,
+      "loss": 0.4842873212192869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4842873212192869
+    },
+    {
+      "train_step": 4233,
+      "loss": 0.6197228442887365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6197228442887365
+    },
+    {
+      "train_step": 4234,
+      "loss": 0.5671287800510375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5671287800510375
+    },
+    {
+      "train_step": 4235,
+      "loss": 0.5755196806460247,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5755196806460247
+    },
+    {
+      "train_step": 4236,
+      "loss": 0.6554196162235172,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6554196162235172
+    },
+    {
+      "train_step": 4237,
+      "loss": 0.4062508738257827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4062508738257827
+    },
+    {
+      "train_step": 4238,
+      "loss": 0.4679756360783893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4679756360783893
+    },
+    {
+      "train_step": 4239,
+      "loss": 0.4384252157814566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4384252157814566
+    },
+    {
+      "train_step": 4240,
+      "loss": 0.5532335978037931,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5532335978037931
+    },
+    {
+      "train_step": 4241,
+      "loss": 0.6236848374011104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6236848374011104
+    },
+    {
+      "train_step": 4242,
+      "loss": 0.5580160959603185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5580160959603185
+    },
+    {
+      "train_step": 4243,
+      "loss": 0.5755821718640964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5755821718640964
+    },
+    {
+      "train_step": 4244,
+      "loss": 0.7893979009537545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7893979009537545
+    },
+    {
+      "train_step": 4245,
+      "loss": 0.6518296798616099,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6518296798616099
+    },
+    {
+      "train_step": 4246,
+      "loss": 0.4171997873579281,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4171997873579281
+    },
+    {
+      "train_step": 4247,
+      "loss": 0.4802695229994845,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4802695229994845
+    },
+    {
+      "train_step": 4248,
+      "loss": 0.45092542464894725,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45092542464894725
+    },
+    {
+      "train_step": 4249,
+      "loss": 0.5582737932878785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5582737932878785
+    },
+    {
+      "train_step": 4250,
+      "loss": 0.5586278206240165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5586278206240165
+    },
+    {
+      "train_step": 4251,
+      "loss": 0.45037513973363363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45037513973363363
+    },
+    {
+      "train_step": 4252,
+      "loss": 0.6987728904671836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6987728904671836
+    },
+    {
+      "train_step": 4253,
+      "loss": 0.6143274210254859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6143274210254859
+    },
+    {
+      "train_step": 4254,
+      "loss": 0.5804172135399656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5804172135399656
+    },
+    {
+      "train_step": 4255,
+      "loss": 0.5944734651162525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5944734651162525
+    },
+    {
+      "train_step": 4256,
+      "loss": 0.5607447932089747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5607447932089747
+    },
+    {
+      "train_step": 4257,
+      "loss": 0.5087845543773947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5087845543773947
+    },
+    {
+      "train_step": 4258,
+      "loss": 0.5711337911444271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5711337911444271
+    },
+    {
+      "train_step": 4259,
+      "loss": 0.551757690110219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.551757690110219
+    },
+    {
+      "train_step": 4260,
+      "loss": 0.6929340923312497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6929340923312497
+    },
+    {
+      "train_step": 4261,
+      "loss": 0.6858172176979321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6858172176979321
+    },
+    {
+      "train_step": 4262,
+      "loss": 0.6659099212014845,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6659099212014845
+    },
+    {
+      "train_step": 4263,
+      "loss": 0.671538997374768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.671538997374768
+    },
+    {
+      "train_step": 4264,
+      "loss": 0.5544504772303698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5544504772303698
+    },
+    {
+      "train_step": 4265,
+      "loss": 0.6445252959177418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6445252959177418
+    },
+    {
+      "train_step": 4266,
+      "loss": 0.4440206906741071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4440206906741071
+    },
+    {
+      "train_step": 4267,
+      "loss": 0.6145238885124416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6145238885124416
+    },
+    {
+      "train_step": 4268,
+      "loss": 0.677255942540556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.677255942540556
+    },
+    {
+      "train_step": 4269,
+      "loss": 0.35563586421877313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35563586421877313
+    },
+    {
+      "train_step": 4270,
+      "loss": 0.5362875666636113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5362875666636113
+    },
+    {
+      "train_step": 4271,
+      "loss": 0.7554371298001478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7554371298001478
+    },
+    {
+      "train_step": 4272,
+      "loss": 0.5531964351574553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5531964351574553
+    },
+    {
+      "train_step": 4273,
+      "loss": 0.7532707080061485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7532707080061485
+    },
+    {
+      "train_step": 4274,
+      "loss": 0.42699276163519334,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42699276163519334
+    },
+    {
+      "train_step": 4275,
+      "loss": 0.5100247878094621,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5100247878094621
+    },
+    {
+      "train_step": 4276,
+      "loss": 0.578173615660019,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.578173615660019
+    },
+    {
+      "train_step": 4277,
+      "loss": 0.6302088399439681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6302088399439681
+    },
+    {
+      "train_step": 4278,
+      "loss": 0.6012919927541335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6012919927541335
+    },
+    {
+      "train_step": 4279,
+      "loss": 0.4791749490646373,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4791749490646373
+    },
+    {
+      "train_step": 4280,
+      "loss": 0.24241665760721443,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.24241665760721443
+    },
+    {
+      "train_step": 4281,
+      "loss": 0.5860601521510429,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5860601521510429
+    },
+    {
+      "train_step": 4282,
+      "loss": 0.5302424120146388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5302424120146388
+    },
+    {
+      "train_step": 4283,
+      "loss": 0.5749693119123179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5749693119123179
+    },
+    {
+      "train_step": 4284,
+      "loss": 0.5876576190054323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5876576190054323
+    },
+    {
+      "train_step": 4285,
+      "loss": 0.44095366732773367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44095366732773367
+    },
+    {
+      "train_step": 4286,
+      "loss": 0.6455664837410295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6455664837410295
+    },
+    {
+      "train_step": 4287,
+      "loss": 0.5552017549919026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5552017549919026
+    },
+    {
+      "train_step": 4288,
+      "loss": 0.598120835153513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.598120835153513
+    },
+    {
+      "train_step": 4289,
+      "loss": 0.5601120580362349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5601120580362349
+    },
+    {
+      "train_step": 4290,
+      "loss": 0.5416552964578478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5416552964578478
+    },
+    {
+      "train_step": 4291,
+      "loss": 0.678333346644762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.678333346644762
+    },
+    {
+      "train_step": 4292,
+      "loss": 0.47500800722099656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47500800722099656
+    },
+    {
+      "train_step": 4293,
+      "loss": 0.5303067065361862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5303067065361862
+    },
+    {
+      "train_step": 4294,
+      "loss": 0.4547364571345981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4547364571345981
+    },
+    {
+      "train_step": 4295,
+      "loss": 0.4795842226937159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4795842226937159
+    },
+    {
+      "train_step": 4296,
+      "loss": 0.5662869747043717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5662869747043717
+    },
+    {
+      "train_step": 4297,
+      "loss": 0.8855422796032011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8855422796032011
+    },
+    {
+      "train_step": 4298,
+      "loss": 0.7243532768658529,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7243532768658529
+    },
+    {
+      "train_step": 4299,
+      "loss": 0.5575734288614863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5575734288614863
+    },
+    {
+      "train_step": 4300,
+      "loss": 0.6183082478399339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6183082478399339
+    },
+    {
+      "train_step": 4301,
+      "loss": 0.5535599182269,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5535599182269
+    },
+    {
+      "train_step": 4302,
+      "loss": 0.7297921760944872,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7297921760944872
+    },
+    {
+      "train_step": 4303,
+      "loss": 0.4085502885203104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4085502885203104
+    },
+    {
+      "train_step": 4304,
+      "loss": 0.7396279354007994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7396279354007994
+    },
+    {
+      "train_step": 4305,
+      "loss": 0.4621786176849839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4621786176849839
+    },
+    {
+      "train_step": 4306,
+      "loss": 0.4704150717889177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4704150717889177
+    },
+    {
+      "train_step": 4307,
+      "loss": 0.5169832816082337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5169832816082337
+    },
+    {
+      "train_step": 4308,
+      "loss": 0.4793197816323973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4793197816323973
+    },
+    {
+      "train_step": 4309,
+      "loss": 0.5950479760238689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5950479760238689
+    },
+    {
+      "train_step": 4310,
+      "loss": 0.4786463587021042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4786463587021042
+    },
+    {
+      "train_step": 4311,
+      "loss": 0.6321240125213705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6321240125213705
+    },
+    {
+      "train_step": 4312,
+      "loss": 0.4766111476095845,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4766111476095845
+    },
+    {
+      "train_step": 4313,
+      "loss": 0.4619976759325485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4619976759325485
+    },
+    {
+      "train_step": 4314,
+      "loss": 0.633616047835266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.633616047835266
+    },
+    {
+      "train_step": 4315,
+      "loss": 0.7636845857475143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7636845857475143
+    },
+    {
+      "train_step": 4316,
+      "loss": 0.6224275860713202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6224275860713202
+    },
+    {
+      "train_step": 4317,
+      "loss": 0.7224226212417801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7224226212417801
+    },
+    {
+      "train_step": 4318,
+      "loss": 0.27966569003696745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.27966569003696745
+    },
+    {
+      "train_step": 4319,
+      "loss": 0.4788639605292626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4788639605292626
+    },
+    {
+      "train_step": 4320,
+      "loss": 0.5391719296697914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5391719296697914
+    },
+    {
+      "train_step": 4321,
+      "loss": 0.5204161550335029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5204161550335029
+    },
+    {
+      "train_step": 4322,
+      "loss": 0.5404034957054378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5404034957054378
+    },
+    {
+      "train_step": 4323,
+      "loss": 0.720787268498561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.720787268498561
+    },
+    {
+      "train_step": 4324,
+      "loss": 0.6270612430043065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6270612430043065
+    },
+    {
+      "train_step": 4325,
+      "loss": 0.43012047100280143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43012047100280143
+    },
+    {
+      "train_step": 4326,
+      "loss": 0.503774205419639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.503774205419639
+    },
+    {
+      "train_step": 4327,
+      "loss": 0.638225386854074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.638225386854074
+    },
+    {
+      "train_step": 4328,
+      "loss": 0.699666815661415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.699666815661415
+    },
+    {
+      "train_step": 4329,
+      "loss": 0.46725619762982695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46725619762982695
+    },
+    {
+      "train_step": 4330,
+      "loss": 0.6585658899465985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6585658899465985
+    },
+    {
+      "train_step": 4331,
+      "loss": 0.6539319480892276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6539319480892276
+    },
+    {
+      "train_step": 4332,
+      "loss": 0.6429753782584382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6429753782584382
+    },
+    {
+      "train_step": 4333,
+      "loss": 0.660368029833821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.660368029833821
+    },
+    {
+      "train_step": 4334,
+      "loss": 0.5331012638882615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5331012638882615
+    },
+    {
+      "train_step": 4335,
+      "loss": 0.7423415979399746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7423415979399746
+    },
+    {
+      "train_step": 4336,
+      "loss": 0.33731177381885935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33731177381885935
+    },
+    {
+      "train_step": 4337,
+      "loss": 0.5353060285181229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5353060285181229
+    },
+    {
+      "train_step": 4338,
+      "loss": 0.5411668922467526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5411668922467526
+    },
+    {
+      "train_step": 4339,
+      "loss": 0.5309766124115121,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5309766124115121
+    },
+    {
+      "train_step": 4340,
+      "loss": 0.5504663100884561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5504663100884561
+    },
+    {
+      "train_step": 4341,
+      "loss": 0.5063234921150799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5063234921150799
+    },
+    {
+      "train_step": 4342,
+      "loss": 0.5508440413130062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5508440413130062
+    },
+    {
+      "train_step": 4343,
+      "loss": 0.6506052666574949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6506052666574949
+    },
+    {
+      "train_step": 4344,
+      "loss": 0.6998751496096737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6998751496096737
+    },
+    {
+      "train_step": 4345,
+      "loss": 0.45327225436393037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45327225436393037
+    },
+    {
+      "train_step": 4346,
+      "loss": 0.47270631330645013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47270631330645013
+    },
+    {
+      "train_step": 4347,
+      "loss": 0.7809767365686312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7809767365686312
+    },
+    {
+      "train_step": 4348,
+      "loss": 0.7575913513909922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7575913513909922
+    },
+    {
+      "train_step": 4349,
+      "loss": 0.43979249744469834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43979249744469834
+    },
+    {
+      "train_step": 4350,
+      "loss": 0.6282361841745144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6282361841745144
+    },
+    {
+      "train_step": 4351,
+      "loss": 0.46587805956561096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46587805956561096
+    },
+    {
+      "train_step": 4352,
+      "loss": 0.4198649530178733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4198649530178733
+    },
+    {
+      "train_step": 4353,
+      "loss": 0.671753840021639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.671753840021639
+    },
+    {
+      "train_step": 4354,
+      "loss": 0.4691557113062967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4691557113062967
+    },
+    {
+      "train_step": 4355,
+      "loss": 0.5865191688445486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5865191688445486
+    },
+    {
+      "train_step": 4356,
+      "loss": 0.42436638717334374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42436638717334374
+    },
+    {
+      "train_step": 4357,
+      "loss": 0.6730177291960752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6730177291960752
+    },
+    {
+      "train_step": 4358,
+      "loss": 0.6399425116275772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6399425116275772
+    },
+    {
+      "train_step": 4359,
+      "loss": 0.5089768203794015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5089768203794015
+    },
+    {
+      "train_step": 4360,
+      "loss": 0.3991935028072314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3991935028072314
+    },
+    {
+      "train_step": 4361,
+      "loss": 0.5605898718885848,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5605898718885848
+    },
+    {
+      "train_step": 4362,
+      "loss": 0.592803215883937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.592803215883937
+    },
+    {
+      "train_step": 4363,
+      "loss": 0.6805203022528996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6805203022528996
+    },
+    {
+      "train_step": 4364,
+      "loss": 0.5941823170913049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5941823170913049
+    },
+    {
+      "train_step": 4365,
+      "loss": 0.6040186814524562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6040186814524562
+    },
+    {
+      "train_step": 4366,
+      "loss": 0.5964548014523157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5964548014523157
+    },
+    {
+      "train_step": 4367,
+      "loss": 0.48635341327993065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48635341327993065
+    },
+    {
+      "train_step": 4368,
+      "loss": 0.6878929882404303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6878929882404303
+    },
+    {
+      "train_step": 4369,
+      "loss": 0.5603767648176006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5603767648176006
+    },
+    {
+      "train_step": 4370,
+      "loss": 0.6751562999822738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6751562999822738
+    },
+    {
+      "train_step": 4371,
+      "loss": 0.6712786282840594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6712786282840594
+    },
+    {
+      "train_step": 4372,
+      "loss": 0.5280345549298678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5280345549298678
+    },
+    {
+      "train_step": 4373,
+      "loss": 0.4054954536440584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4054954536440584
+    },
+    {
+      "train_step": 4374,
+      "loss": 0.7374576764192252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7374576764192252
+    },
+    {
+      "train_step": 4375,
+      "loss": 0.5231883336537143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5231883336537143
+    },
+    {
+      "train_step": 4376,
+      "loss": 0.5439551639462169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5439551639462169
+    },
+    {
+      "train_step": 4377,
+      "loss": 0.3937010576125937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3937010576125937
+    },
+    {
+      "train_step": 4378,
+      "loss": 0.7486004416306619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7486004416306619
+    },
+    {
+      "train_step": 4379,
+      "loss": 0.6250762519194832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6250762519194832
+    },
+    {
+      "train_step": 4380,
+      "loss": 0.5834040121246827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5834040121246827
+    },
+    {
+      "train_step": 4381,
+      "loss": 0.5688821718118986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5688821718118986
+    },
+    {
+      "train_step": 4382,
+      "loss": 0.6159041004803777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6159041004803777
+    },
+    {
+      "train_step": 4383,
+      "loss": 0.7516047049216948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7516047049216948
+    },
+    {
+      "train_step": 4384,
+      "loss": 0.364447407852643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.364447407852643
+    },
+    {
+      "train_step": 4385,
+      "loss": 0.5877950060364205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5877950060364205
+    },
+    {
+      "train_step": 4386,
+      "loss": 0.5477814926863984,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5477814926863984
+    },
+    {
+      "train_step": 4387,
+      "loss": 0.5347861680539266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5347861680539266
+    },
+    {
+      "train_step": 4388,
+      "loss": 0.6583592614034737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6583592614034737
+    },
+    {
+      "train_step": 4389,
+      "loss": 0.5334927042993366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5334927042993366
+    },
+    {
+      "train_step": 4390,
+      "loss": 0.685326285992076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.685326285992076
+    },
+    {
+      "train_step": 4391,
+      "loss": 0.8560754180102756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8560754180102756
+    },
+    {
+      "train_step": 4392,
+      "loss": 0.5536876961996432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5536876961996432
+    },
+    {
+      "train_step": 4393,
+      "loss": 0.578571217716111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.578571217716111
+    },
+    {
+      "train_step": 4394,
+      "loss": 0.7864721254655579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7864721254655579
+    },
+    {
+      "train_step": 4395,
+      "loss": 0.5754612840001888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5754612840001888
+    },
+    {
+      "train_step": 4396,
+      "loss": 0.5972107489061496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5972107489061496
+    },
+    {
+      "train_step": 4397,
+      "loss": 0.470841988512363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.470841988512363
+    },
+    {
+      "train_step": 4398,
+      "loss": 0.5593604934679884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5593604934679884
+    },
+    {
+      "train_step": 4399,
+      "loss": 0.6342468174278362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6342468174278362
+    },
+    {
+      "train_step": 4400,
+      "loss": 0.289345265052993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.289345265052993
+    },
+    {
+      "train_step": 4401,
+      "loss": 0.6862803946781303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6862803946781303
+    },
+    {
+      "train_step": 4402,
+      "loss": 0.6898644885849081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6898644885849081
+    },
+    {
+      "train_step": 4403,
+      "loss": 0.7435889735826122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7435889735826122
+    },
+    {
+      "train_step": 4404,
+      "loss": 0.567566656842937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.567566656842937
+    },
+    {
+      "train_step": 4405,
+      "loss": 0.34545184582815713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34545184582815713
+    },
+    {
+      "train_step": 4406,
+      "loss": 0.739557547961609,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.739557547961609
+    },
+    {
+      "train_step": 4407,
+      "loss": 0.5964065134683293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5964065134683293
+    },
+    {
+      "train_step": 4408,
+      "loss": 0.5520282340580042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5520282340580042
+    },
+    {
+      "train_step": 4409,
+      "loss": 0.6453108078521502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6453108078521502
+    },
+    {
+      "train_step": 4410,
+      "loss": 0.6524178818271844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6524178818271844
+    },
+    {
+      "train_step": 4411,
+      "loss": 0.4612526470830068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4612526470830068
+    },
+    {
+      "train_step": 4412,
+      "loss": 0.6827645314866392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6827645314866392
+    },
+    {
+      "train_step": 4413,
+      "loss": 0.4258207139981475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4258207139981475
+    },
+    {
+      "train_step": 4414,
+      "loss": 0.5608222902420007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5608222902420007
+    },
+    {
+      "train_step": 4415,
+      "loss": 0.7293370529157477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7293370529157477
+    },
+    {
+      "train_step": 4416,
+      "loss": 0.5447781790106095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5447781790106095
+    },
+    {
+      "train_step": 4417,
+      "loss": 0.41216470669249156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41216470669249156
+    },
+    {
+      "train_step": 4418,
+      "loss": 0.7920974475894678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7920974475894678
+    },
+    {
+      "train_step": 4419,
+      "loss": 0.45905313123784075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45905313123784075
+    },
+    {
+      "train_step": 4420,
+      "loss": 0.7064176334925535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7064176334925535
+    },
+    {
+      "train_step": 4421,
+      "loss": 0.5601908885596023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5601908885596023
+    },
+    {
+      "train_step": 4422,
+      "loss": 0.4205236447880471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4205236447880471
+    },
+    {
+      "train_step": 4423,
+      "loss": 0.5208694754036035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5208694754036035
+    },
+    {
+      "train_step": 4424,
+      "loss": 0.6796225083871227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6796225083871227
+    },
+    {
+      "train_step": 4425,
+      "loss": 0.4039992991153448,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4039992991153448
+    },
+    {
+      "train_step": 4426,
+      "loss": 0.5418051984469507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5418051984469507
+    },
+    {
+      "train_step": 4427,
+      "loss": 0.5818537405331377,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5818537405331377
+    },
+    {
+      "train_step": 4428,
+      "loss": 0.5935915505050393,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5935915505050393
+    },
+    {
+      "train_step": 4429,
+      "loss": 0.7134450051045351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7134450051045351
+    },
+    {
+      "train_step": 4430,
+      "loss": 0.6707118579604799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6707118579604799
+    },
+    {
+      "train_step": 4431,
+      "loss": 0.49932145180583937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49932145180583937
+    },
+    {
+      "train_step": 4432,
+      "loss": 0.596230634485131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.596230634485131
+    },
+    {
+      "train_step": 4433,
+      "loss": 0.5469689114677638,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5469689114677638
+    },
+    {
+      "train_step": 4434,
+      "loss": 0.5155895571002858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5155895571002858
+    },
+    {
+      "train_step": 4435,
+      "loss": 0.6517968660324853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6517968660324853
+    },
+    {
+      "train_step": 4436,
+      "loss": 0.4972885131640809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4972885131640809
+    },
+    {
+      "train_step": 4437,
+      "loss": 0.7865988053185701,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7865988053185701
+    },
+    {
+      "train_step": 4438,
+      "loss": 0.469798209977321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.469798209977321
+    },
+    {
+      "train_step": 4439,
+      "loss": 0.47542196738596754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47542196738596754
+    },
+    {
+      "train_step": 4440,
+      "loss": 0.5821344882436433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5821344882436433
+    },
+    {
+      "train_step": 4441,
+      "loss": 0.46216487500064013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46216487500064013
+    },
+    {
+      "train_step": 4442,
+      "loss": 0.6372706103507768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6372706103507768
+    },
+    {
+      "train_step": 4443,
+      "loss": 0.6465397652935826,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6465397652935826
+    },
+    {
+      "train_step": 4444,
+      "loss": 0.6719960929935633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6719960929935633
+    },
+    {
+      "train_step": 4445,
+      "loss": 0.49887081875545275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49887081875545275
+    },
+    {
+      "train_step": 4446,
+      "loss": 0.6100344919560569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6100344919560569
+    },
+    {
+      "train_step": 4447,
+      "loss": 0.7096069927150608,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7096069927150608
+    },
+    {
+      "train_step": 4448,
+      "loss": 0.7070843567655667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7070843567655667
+    },
+    {
+      "train_step": 4449,
+      "loss": 0.6785418689498132,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6785418689498132
+    },
+    {
+      "train_step": 4450,
+      "loss": 0.5200932430402262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5200932430402262
+    },
+    {
+      "train_step": 4451,
+      "loss": 0.5453313785705072,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5453313785705072
+    },
+    {
+      "train_step": 4452,
+      "loss": 0.5283136575845306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5283136575845306
+    },
+    {
+      "train_step": 4453,
+      "loss": 0.5211238657501016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5211238657501016
+    },
+    {
+      "train_step": 4454,
+      "loss": 0.5030586965083338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5030586965083338
+    },
+    {
+      "train_step": 4455,
+      "loss": 0.684611609961184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.684611609961184
+    },
+    {
+      "train_step": 4456,
+      "loss": 0.7828168957887418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7828168957887418
+    },
+    {
+      "train_step": 4457,
+      "loss": 0.5529592272948676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5529592272948676
+    },
+    {
+      "train_step": 4458,
+      "loss": 0.545192620378125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.545192620378125
+    },
+    {
+      "train_step": 4459,
+      "loss": 0.54756401444194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.54756401444194
+    },
+    {
+      "train_step": 4460,
+      "loss": 0.6829538046013053,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6829538046013053
+    },
+    {
+      "train_step": 4461,
+      "loss": 0.4902611757651948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4902611757651948
+    },
+    {
+      "train_step": 4462,
+      "loss": 0.5781425365314251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5781425365314251
+    },
+    {
+      "train_step": 4463,
+      "loss": 0.6082721780916843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6082721780916843
+    },
+    {
+      "train_step": 4464,
+      "loss": 0.4992248648021066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4992248648021066
+    },
+    {
+      "train_step": 4465,
+      "loss": 0.631083672537444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.631083672537444
+    },
+    {
+      "train_step": 4466,
+      "loss": 0.5406613638138889,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5406613638138889
+    },
+    {
+      "train_step": 4467,
+      "loss": 0.5990612102371535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5990612102371535
+    },
+    {
+      "train_step": 4468,
+      "loss": 0.587123908872923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.587123908872923
+    },
+    {
+      "train_step": 4469,
+      "loss": 0.5748938819467095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5748938819467095
+    },
+    {
+      "train_step": 4470,
+      "loss": 0.6212603134735882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6212603134735882
+    },
+    {
+      "train_step": 4471,
+      "loss": 0.639944330417462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.639944330417462
+    },
+    {
+      "train_step": 4472,
+      "loss": 0.7410745766673268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7410745766673268
+    },
+    {
+      "train_step": 4473,
+      "loss": 0.373571912445324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.373571912445324
+    },
+    {
+      "train_step": 4474,
+      "loss": 0.6235645868946658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6235645868946658
+    },
+    {
+      "train_step": 4475,
+      "loss": 0.7441573687002581,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7441573687002581
+    },
+    {
+      "train_step": 4476,
+      "loss": 0.6422386597544562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6422386597544562
+    },
+    {
+      "train_step": 4477,
+      "loss": 0.49183598137667517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49183598137667517
+    },
+    {
+      "train_step": 4478,
+      "loss": 0.5085327774762175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5085327774762175
+    },
+    {
+      "train_step": 4479,
+      "loss": 0.6403424041543284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6403424041543284
+    },
+    {
+      "train_step": 4480,
+      "loss": 0.5201896089757666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5201896089757666
+    },
+    {
+      "train_step": 4481,
+      "loss": 0.5057510192827905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5057510192827905
+    },
+    {
+      "train_step": 4482,
+      "loss": 0.7530025471884011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7530025471884011
+    },
+    {
+      "train_step": 4483,
+      "loss": 0.610166073229777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.610166073229777
+    },
+    {
+      "train_step": 4484,
+      "loss": 0.497800535911588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.497800535911588
+    },
+    {
+      "train_step": 4485,
+      "loss": 0.420937220592931,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.420937220592931
+    },
+    {
+      "train_step": 4486,
+      "loss": 0.6664277483644127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6664277483644127
+    },
+    {
+      "train_step": 4487,
+      "loss": 0.6569005790540073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6569005790540073
+    },
+    {
+      "train_step": 4488,
+      "loss": 0.47725915975544103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47725915975544103
+    },
+    {
+      "train_step": 4489,
+      "loss": 0.6502674091527526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6502674091527526
+    },
+    {
+      "train_step": 4490,
+      "loss": 0.5743217471015147,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5743217471015147
+    },
+    {
+      "train_step": 4491,
+      "loss": 0.6739982218561731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6739982218561731
+    },
+    {
+      "train_step": 4492,
+      "loss": 0.7484961108457406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7484961108457406
+    },
+    {
+      "train_step": 4493,
+      "loss": 0.437973150231626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.437973150231626
+    },
+    {
+      "train_step": 4494,
+      "loss": 0.8227036941403928,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8227036941403928
+    },
+    {
+      "train_step": 4495,
+      "loss": 1.038830958087248,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 1.038830958087248
+    },
+    {
+      "train_step": 4496,
+      "loss": 0.6050808143956392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6050808143956392
+    },
+    {
+      "train_step": 4497,
+      "loss": 0.4709832654487218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4709832654487218
+    },
+    {
+      "train_step": 4498,
+      "loss": 0.5716130944702744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5716130944702744
+    },
+    {
+      "train_step": 4499,
+      "loss": 0.7840793200202049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7840793200202049
+    },
+    {
+      "train_step": 4500,
+      "loss": 0.5622068913401903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5622068913401903
+    },
+    {
+      "train_step": 4501,
+      "loss": 0.41844137728574327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41844137728574327
+    },
+    {
+      "train_step": 4502,
+      "loss": 0.4886229423133237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4886229423133237
+    },
+    {
+      "train_step": 4503,
+      "loss": 0.7090787925238844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7090787925238844
+    },
+    {
+      "train_step": 4504,
+      "loss": 0.66241960491048,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66241960491048
+    },
+    {
+      "train_step": 4505,
+      "loss": 0.6059128604615894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6059128604615894
+    },
+    {
+      "train_step": 4506,
+      "loss": 0.5876455284196735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5876455284196735
+    },
+    {
+      "train_step": 4507,
+      "loss": 0.6101021891135339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6101021891135339
+    },
+    {
+      "train_step": 4508,
+      "loss": 0.6952387559082258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6952387559082258
+    },
+    {
+      "train_step": 4509,
+      "loss": 0.5238619751369015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5238619751369015
+    },
+    {
+      "train_step": 4510,
+      "loss": 0.7346240003560716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7346240003560716
+    },
+    {
+      "train_step": 4511,
+      "loss": 0.546143392442146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.546143392442146
+    },
+    {
+      "train_step": 4512,
+      "loss": 0.7670720564894246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7670720564894246
+    },
+    {
+      "train_step": 4513,
+      "loss": 0.5822305491210802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5822305491210802
+    },
+    {
+      "train_step": 4514,
+      "loss": 0.46490979495381723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46490979495381723
+    },
+    {
+      "train_step": 4515,
+      "loss": 0.7212580135879898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7212580135879898
+    },
+    {
+      "train_step": 4516,
+      "loss": 0.5274618140377619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5274618140377619
+    },
+    {
+      "train_step": 4517,
+      "loss": 0.54163000894961,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.54163000894961
+    },
+    {
+      "train_step": 4518,
+      "loss": 0.5541337049961145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5541337049961145
+    },
+    {
+      "train_step": 4519,
+      "loss": 0.5004986739760331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5004986739760331
+    },
+    {
+      "train_step": 4520,
+      "loss": 0.5478429698865117,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5478429698865117
+    },
+    {
+      "train_step": 4521,
+      "loss": 0.5701476885786032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5701476885786032
+    },
+    {
+      "train_step": 4522,
+      "loss": 0.5202611088748205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5202611088748205
+    },
+    {
+      "train_step": 4523,
+      "loss": 0.6412083327521413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6412083327521413
+    },
+    {
+      "train_step": 4524,
+      "loss": 0.5215401731839913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5215401731839913
+    },
+    {
+      "train_step": 4525,
+      "loss": 0.538178757880125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.538178757880125
+    },
+    {
+      "train_step": 4526,
+      "loss": 0.48953448976479175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48953448976479175
+    },
+    {
+      "train_step": 4527,
+      "loss": 0.6498446513293233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6498446513293233
+    },
+    {
+      "train_step": 4528,
+      "loss": 0.5423269516598167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5423269516598167
+    },
+    {
+      "train_step": 4529,
+      "loss": 0.4746771806923262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4746771806923262
+    },
+    {
+      "train_step": 4530,
+      "loss": 0.5904953775991418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5904953775991418
+    },
+    {
+      "train_step": 4531,
+      "loss": 0.6589625740422915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6589625740422915
+    },
+    {
+      "train_step": 4532,
+      "loss": 0.4644427279735388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4644427279735388
+    },
+    {
+      "train_step": 4533,
+      "loss": 0.6718438717566249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6718438717566249
+    },
+    {
+      "train_step": 4534,
+      "loss": 0.609522267312085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.609522267312085
+    },
+    {
+      "train_step": 4535,
+      "loss": 0.5098447922997994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5098447922997994
+    },
+    {
+      "train_step": 4536,
+      "loss": 0.8216280473540601,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8216280473540601
+    },
+    {
+      "train_step": 4537,
+      "loss": 0.35923512500099797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35923512500099797
+    },
+    {
+      "train_step": 4538,
+      "loss": 0.4744600752623349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4744600752623349
+    },
+    {
+      "train_step": 4539,
+      "loss": 0.563909854416204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.563909854416204
+    },
+    {
+      "train_step": 4540,
+      "loss": 0.6753500252393583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6753500252393583
+    },
+    {
+      "train_step": 4541,
+      "loss": 0.5804392581813457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5804392581813457
+    },
+    {
+      "train_step": 4542,
+      "loss": 0.5059871635075789,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5059871635075789
+    },
+    {
+      "train_step": 4543,
+      "loss": 0.34829656789831515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34829656789831515
+    },
+    {
+      "train_step": 4544,
+      "loss": 0.7457840387204328,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7457840387204328
+    },
+    {
+      "train_step": 4545,
+      "loss": 0.5866569012564193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5866569012564193
+    },
+    {
+      "train_step": 4546,
+      "loss": 0.6547266122613736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6547266122613736
+    },
+    {
+      "train_step": 4547,
+      "loss": 0.6131822297135586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6131822297135586
+    },
+    {
+      "train_step": 4548,
+      "loss": 0.3845992071385957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3845992071385957
+    },
+    {
+      "train_step": 4549,
+      "loss": 0.7599953448706943,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7599953448706943
+    },
+    {
+      "train_step": 4550,
+      "loss": 0.4629694742166729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4629694742166729
+    },
+    {
+      "train_step": 4551,
+      "loss": 0.45729076579669214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45729076579669214
+    },
+    {
+      "train_step": 4552,
+      "loss": 0.8093336577558655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8093336577558655
+    },
+    {
+      "train_step": 4553,
+      "loss": 0.5744907861021472,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5744907861021472
+    },
+    {
+      "train_step": 4554,
+      "loss": 0.5136701015186714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5136701015186714
+    },
+    {
+      "train_step": 4555,
+      "loss": 0.5493413916714224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5493413916714224
+    },
+    {
+      "train_step": 4556,
+      "loss": 0.4426666073250671,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4426666073250671
+    },
+    {
+      "train_step": 4557,
+      "loss": 0.35094764677725876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35094764677725876
+    },
+    {
+      "train_step": 4558,
+      "loss": 0.393666230648857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.393666230648857
+    },
+    {
+      "train_step": 4559,
+      "loss": 0.5515165736182612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5515165736182612
+    },
+    {
+      "train_step": 4560,
+      "loss": 0.6199927799696432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6199927799696432
+    },
+    {
+      "train_step": 4561,
+      "loss": 0.3809989270017107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3809989270017107
+    },
+    {
+      "train_step": 4562,
+      "loss": 0.6726641321929268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6726641321929268
+    },
+    {
+      "train_step": 4563,
+      "loss": 0.6225654769721012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6225654769721012
+    },
+    {
+      "train_step": 4564,
+      "loss": 0.5064741645716532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5064741645716532
+    },
+    {
+      "train_step": 4565,
+      "loss": 0.3615051121834715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3615051121834715
+    },
+    {
+      "train_step": 4566,
+      "loss": 0.6080580964843908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6080580964843908
+    },
+    {
+      "train_step": 4567,
+      "loss": 0.5403856943749019,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5403856943749019
+    },
+    {
+      "train_step": 4568,
+      "loss": 0.7465812655003469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7465812655003469
+    },
+    {
+      "train_step": 4569,
+      "loss": 0.5877471658897188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5877471658897188
+    },
+    {
+      "train_step": 4570,
+      "loss": 0.518770115204352,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.518770115204352
+    },
+    {
+      "train_step": 4571,
+      "loss": 0.4089448793287951,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4089448793287951
+    },
+    {
+      "train_step": 4572,
+      "loss": 0.3592579763272122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3592579763272122
+    },
+    {
+      "train_step": 4573,
+      "loss": 0.2963070420575094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2963070420575094
+    },
+    {
+      "train_step": 4574,
+      "loss": 0.5036426406085991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5036426406085991
+    },
+    {
+      "train_step": 4575,
+      "loss": 0.6344077281189979,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6344077281189979
+    },
+    {
+      "train_step": 4576,
+      "loss": 0.49057012257088295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49057012257088295
+    },
+    {
+      "train_step": 4577,
+      "loss": 0.7115374611900415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7115374611900415
+    },
+    {
+      "train_step": 4578,
+      "loss": 0.5681739970594506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5681739970594506
+    },
+    {
+      "train_step": 4579,
+      "loss": 0.7155870369310173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7155870369310173
+    },
+    {
+      "train_step": 4580,
+      "loss": 0.7104777278155818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7104777278155818
+    },
+    {
+      "train_step": 4581,
+      "loss": 0.6278926638530442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6278926638530442
+    },
+    {
+      "train_step": 4582,
+      "loss": 0.6556915259878375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6556915259878375
+    },
+    {
+      "train_step": 4583,
+      "loss": 0.7698037527176796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7698037527176796
+    },
+    {
+      "train_step": 4584,
+      "loss": 0.9671652328597727,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9671652328597727
+    },
+    {
+      "train_step": 4585,
+      "loss": 0.4691636354433846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4691636354433846
+    },
+    {
+      "train_step": 4586,
+      "loss": 0.5172225642127095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5172225642127095
+    },
+    {
+      "train_step": 4587,
+      "loss": 0.6470645301394463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6470645301394463
+    },
+    {
+      "train_step": 4588,
+      "loss": 0.8733268440089959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8733268440089959
+    },
+    {
+      "train_step": 4589,
+      "loss": 0.41977872241332137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41977872241332137
+    },
+    {
+      "train_step": 4590,
+      "loss": 0.626954970182692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.626954970182692
+    },
+    {
+      "train_step": 4591,
+      "loss": 0.7415796245731124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7415796245731124
+    },
+    {
+      "train_step": 4592,
+      "loss": 0.5511638381505015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5511638381505015
+    },
+    {
+      "train_step": 4593,
+      "loss": 0.6561765617969431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6561765617969431
+    },
+    {
+      "train_step": 4594,
+      "loss": 0.5626571282976571,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5626571282976571
+    },
+    {
+      "train_step": 4595,
+      "loss": 0.5674473343460392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5674473343460392
+    },
+    {
+      "train_step": 4596,
+      "loss": 0.5069550805443367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5069550805443367
+    },
+    {
+      "train_step": 4597,
+      "loss": 0.44304420758124563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44304420758124563
+    },
+    {
+      "train_step": 4598,
+      "loss": 0.7589295597128285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7589295597128285
+    },
+    {
+      "train_step": 4599,
+      "loss": 0.5138936976734794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5138936976734794
+    },
+    {
+      "train_step": 4600,
+      "loss": 0.5244042619637971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5244042619637971
+    },
+    {
+      "train_step": 4601,
+      "loss": 0.6241348156406957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6241348156406957
+    },
+    {
+      "train_step": 4602,
+      "loss": 0.5381660610577153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5381660610577153
+    },
+    {
+      "train_step": 4603,
+      "loss": 0.5498035979790425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5498035979790425
+    },
+    {
+      "train_step": 4604,
+      "loss": 0.45342141601256963,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45342141601256963
+    },
+    {
+      "train_step": 4605,
+      "loss": 0.7706467747973086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7706467747973086
+    },
+    {
+      "train_step": 4606,
+      "loss": 0.6718277364592908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6718277364592908
+    },
+    {
+      "train_step": 4607,
+      "loss": 0.46413439474432167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46413439474432167
+    },
+    {
+      "train_step": 4608,
+      "loss": 0.6382023390600869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6382023390600869
+    },
+    {
+      "train_step": 4609,
+      "loss": 0.6843091592309193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6843091592309193
+    },
+    {
+      "train_step": 4610,
+      "loss": 0.7150963461787012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7150963461787012
+    },
+    {
+      "train_step": 4611,
+      "loss": 0.34856547458333553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34856547458333553
+    },
+    {
+      "train_step": 4612,
+      "loss": 0.6423695997560731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6423695997560731
+    },
+    {
+      "train_step": 4613,
+      "loss": 0.6647026219270495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6647026219270495
+    },
+    {
+      "train_step": 4614,
+      "loss": 0.474988296909072,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.474988296909072
+    },
+    {
+      "train_step": 4615,
+      "loss": 0.4666058519479322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4666058519479322
+    },
+    {
+      "train_step": 4616,
+      "loss": 0.5516419622867795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5516419622867795
+    },
+    {
+      "train_step": 4617,
+      "loss": 0.4794335426958912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4794335426958912
+    },
+    {
+      "train_step": 4618,
+      "loss": 0.6257801065159665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6257801065159665
+    },
+    {
+      "train_step": 4619,
+      "loss": 0.6039520624689949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6039520624689949
+    },
+    {
+      "train_step": 4620,
+      "loss": 0.47612324497535014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47612324497535014
+    },
+    {
+      "train_step": 4621,
+      "loss": 0.6758294939672784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6758294939672784
+    },
+    {
+      "train_step": 4622,
+      "loss": 0.4178771032121465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4178771032121465
+    },
+    {
+      "train_step": 4623,
+      "loss": 0.47525751297613666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47525751297613666
+    },
+    {
+      "train_step": 4624,
+      "loss": 0.647322459326912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.647322459326912
+    },
+    {
+      "train_step": 4625,
+      "loss": 0.39329274625112187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39329274625112187
+    },
+    {
+      "train_step": 4626,
+      "loss": 0.3372192981027384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3372192981027384
+    },
+    {
+      "train_step": 4627,
+      "loss": 0.7594602828203139,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7594602828203139
+    },
+    {
+      "train_step": 4628,
+      "loss": 0.5067580752101218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5067580752101218
+    },
+    {
+      "train_step": 4629,
+      "loss": 0.7568394814257394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7568394814257394
+    },
+    {
+      "train_step": 4630,
+      "loss": 0.44526756005655915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44526756005655915
+    },
+    {
+      "train_step": 4631,
+      "loss": 0.6943190850602184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6943190850602184
+    },
+    {
+      "train_step": 4632,
+      "loss": 0.5478739603195534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5478739603195534
+    },
+    {
+      "train_step": 4633,
+      "loss": 0.6506136490005994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6506136490005994
+    },
+    {
+      "train_step": 4634,
+      "loss": 0.54186475578818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.54186475578818
+    },
+    {
+      "train_step": 4635,
+      "loss": 0.3866592349305658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3866592349305658
+    },
+    {
+      "train_step": 4636,
+      "loss": 0.5123363254086611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5123363254086611
+    },
+    {
+      "train_step": 4637,
+      "loss": 0.5291446077119499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5291446077119499
+    },
+    {
+      "train_step": 4638,
+      "loss": 0.5590973326027647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5590973326027647
+    },
+    {
+      "train_step": 4639,
+      "loss": 0.5773895524839524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5773895524839524
+    },
+    {
+      "train_step": 4640,
+      "loss": 0.5000983768154479,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5000983768154479
+    },
+    {
+      "train_step": 4641,
+      "loss": 0.6192650955622219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6192650955622219
+    },
+    {
+      "train_step": 4642,
+      "loss": 0.5649970845463017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5649970845463017
+    },
+    {
+      "train_step": 4643,
+      "loss": 0.45550612793511364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45550612793511364
+    },
+    {
+      "train_step": 4644,
+      "loss": 0.5601766599657091,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5601766599657091
+    },
+    {
+      "train_step": 4645,
+      "loss": 0.5661926897015428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5661926897015428
+    },
+    {
+      "train_step": 4646,
+      "loss": 0.7202540092672541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7202540092672541
+    },
+    {
+      "train_step": 4647,
+      "loss": 0.3056589104671358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3056589104671358
+    },
+    {
+      "train_step": 4648,
+      "loss": 0.534147326280771,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.534147326280771
+    },
+    {
+      "train_step": 4649,
+      "loss": 0.6347461481186398,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6347461481186398
+    },
+    {
+      "train_step": 4650,
+      "loss": 0.5231320134029616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5231320134029616
+    },
+    {
+      "train_step": 4651,
+      "loss": 0.6702828719113922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6702828719113922
+    },
+    {
+      "train_step": 4652,
+      "loss": 0.4765911744716835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4765911744716835
+    },
+    {
+      "train_step": 4653,
+      "loss": 0.5526120140939289,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5526120140939289
+    },
+    {
+      "train_step": 4654,
+      "loss": 0.594547044419026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.594547044419026
+    },
+    {
+      "train_step": 4655,
+      "loss": 0.7550849712875558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7550849712875558
+    },
+    {
+      "train_step": 4656,
+      "loss": 0.8942276000144638,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8942276000144638
+    },
+    {
+      "train_step": 4657,
+      "loss": 0.5528824474377712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5528824474377712
+    },
+    {
+      "train_step": 4658,
+      "loss": 0.65156273919267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.65156273919267
+    },
+    {
+      "train_step": 4659,
+      "loss": 0.47802565180874856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47802565180874856
+    },
+    {
+      "train_step": 4660,
+      "loss": 0.6434687006749689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6434687006749689
+    },
+    {
+      "train_step": 4661,
+      "loss": 0.6679999741371343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6679999741371343
+    },
+    {
+      "train_step": 4662,
+      "loss": 0.7680104046377197,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7680104046377197
+    },
+    {
+      "train_step": 4663,
+      "loss": 0.6529088851565418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6529088851565418
+    },
+    {
+      "train_step": 4664,
+      "loss": 0.6267928392868424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6267928392868424
+    },
+    {
+      "train_step": 4665,
+      "loss": 0.7181812000153691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7181812000153691
+    },
+    {
+      "train_step": 4666,
+      "loss": 0.6188045517500631,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6188045517500631
+    },
+    {
+      "train_step": 4667,
+      "loss": 0.4950142422940238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4950142422940238
+    },
+    {
+      "train_step": 4668,
+      "loss": 0.582307866501786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.582307866501786
+    },
+    {
+      "train_step": 4669,
+      "loss": 0.4030076544840934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4030076544840934
+    },
+    {
+      "train_step": 4670,
+      "loss": 0.517554829624125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.517554829624125
+    },
+    {
+      "train_step": 4671,
+      "loss": 0.5792281473908625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5792281473908625
+    },
+    {
+      "train_step": 4672,
+      "loss": 0.5979325286688107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5979325286688107
+    },
+    {
+      "train_step": 4673,
+      "loss": 0.7059483470980717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7059483470980717
+    },
+    {
+      "train_step": 4674,
+      "loss": 0.6394200506715242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6394200506715242
+    },
+    {
+      "train_step": 4675,
+      "loss": 0.5157108539942777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5157108539942777
+    },
+    {
+      "train_step": 4676,
+      "loss": 0.41640122417697084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41640122417697084
+    },
+    {
+      "train_step": 4677,
+      "loss": 0.4074041387400045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4074041387400045
+    },
+    {
+      "train_step": 4678,
+      "loss": 0.6736539660008791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6736539660008791
+    },
+    {
+      "train_step": 4679,
+      "loss": 0.6877041287340989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6877041287340989
+    },
+    {
+      "train_step": 4680,
+      "loss": 0.6190371803697283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6190371803697283
+    },
+    {
+      "train_step": 4681,
+      "loss": 0.5205946231200556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5205946231200556
+    },
+    {
+      "train_step": 4682,
+      "loss": 0.36044134001206707,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36044134001206707
+    },
+    {
+      "train_step": 4683,
+      "loss": 0.5586793400645983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5586793400645983
+    },
+    {
+      "train_step": 4684,
+      "loss": 0.2957528851703216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2957528851703216
+    },
+    {
+      "train_step": 4685,
+      "loss": 0.6253269123156455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6253269123156455
+    },
+    {
+      "train_step": 4686,
+      "loss": 0.5125241877543613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5125241877543613
+    },
+    {
+      "train_step": 4687,
+      "loss": 0.7610879799374002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7610879799374002
+    },
+    {
+      "train_step": 4688,
+      "loss": 0.48062057119870816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48062057119870816
+    },
+    {
+      "train_step": 4689,
+      "loss": 0.6871287204422898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6871287204422898
+    },
+    {
+      "train_step": 4690,
+      "loss": 0.5194801478231125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5194801478231125
+    },
+    {
+      "train_step": 4691,
+      "loss": 0.43941025140799655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43941025140799655
+    },
+    {
+      "train_step": 4692,
+      "loss": 0.5595208331125863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5595208331125863
+    },
+    {
+      "train_step": 4693,
+      "loss": 0.48732734381879583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48732734381879583
+    },
+    {
+      "train_step": 4694,
+      "loss": 0.37228030535935586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37228030535935586
+    },
+    {
+      "train_step": 4695,
+      "loss": 0.719729752811683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.719729752811683
+    },
+    {
+      "train_step": 4696,
+      "loss": 0.45432674668531575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45432674668531575
+    },
+    {
+      "train_step": 4697,
+      "loss": 0.4463551012421674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4463551012421674
+    },
+    {
+      "train_step": 4698,
+      "loss": 0.7666112457218199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7666112457218199
+    },
+    {
+      "train_step": 4699,
+      "loss": 0.6702287066473671,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6702287066473671
+    },
+    {
+      "train_step": 4700,
+      "loss": 0.6930766064235258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6930766064235258
+    },
+    {
+      "train_step": 4701,
+      "loss": 0.470969250719873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.470969250719873
+    },
+    {
+      "train_step": 4702,
+      "loss": 0.5277828937635692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5277828937635692
+    },
+    {
+      "train_step": 4703,
+      "loss": 0.4500973174389374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4500973174389374
+    },
+    {
+      "train_step": 4704,
+      "loss": 0.44863915956769396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44863915956769396
+    },
+    {
+      "train_step": 4705,
+      "loss": 0.6433079237227963,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6433079237227963
+    },
+    {
+      "train_step": 4706,
+      "loss": 0.6388502185673555,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6388502185673555
+    },
+    {
+      "train_step": 4707,
+      "loss": 0.45026151138075643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45026151138075643
+    },
+    {
+      "train_step": 4708,
+      "loss": 0.47450263999041115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47450263999041115
+    },
+    {
+      "train_step": 4709,
+      "loss": 0.6846204617515813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6846204617515813
+    },
+    {
+      "train_step": 4710,
+      "loss": 0.49227858954422504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49227858954422504
+    },
+    {
+      "train_step": 4711,
+      "loss": 0.7082833028233173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7082833028233173
+    },
+    {
+      "train_step": 4712,
+      "loss": 0.5394424574230341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5394424574230341
+    },
+    {
+      "train_step": 4713,
+      "loss": 0.4242988219315783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4242988219315783
+    },
+    {
+      "train_step": 4714,
+      "loss": 0.6577012345053039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6577012345053039
+    },
+    {
+      "train_step": 4715,
+      "loss": 0.5037228062846261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5037228062846261
+    },
+    {
+      "train_step": 4716,
+      "loss": 0.6559329246742406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6559329246742406
+    },
+    {
+      "train_step": 4717,
+      "loss": 0.5131791974038624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5131791974038624
+    },
+    {
+      "train_step": 4718,
+      "loss": 0.6783265733223581,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6783265733223581
+    },
+    {
+      "train_step": 4719,
+      "loss": 0.6129345252207933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6129345252207933
+    },
+    {
+      "train_step": 4720,
+      "loss": 0.750134434043501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.750134434043501
+    },
+    {
+      "train_step": 4721,
+      "loss": 0.45613894621906914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45613894621906914
+    },
+    {
+      "train_step": 4722,
+      "loss": 0.5854098898800492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5854098898800492
+    },
+    {
+      "train_step": 4723,
+      "loss": 0.38117107077389145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38117107077389145
+    },
+    {
+      "train_step": 4724,
+      "loss": 0.6681073770144843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6681073770144843
+    },
+    {
+      "train_step": 4725,
+      "loss": 0.6369907853825904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6369907853825904
+    },
+    {
+      "train_step": 4726,
+      "loss": 0.5921624438068014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5921624438068014
+    },
+    {
+      "train_step": 4727,
+      "loss": 0.7903309826541215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7903309826541215
+    },
+    {
+      "train_step": 4728,
+      "loss": 0.6201976315770166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6201976315770166
+    },
+    {
+      "train_step": 4729,
+      "loss": 0.474248904718805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.474248904718805
+    },
+    {
+      "train_step": 4730,
+      "loss": 0.4901282391688598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4901282391688598
+    },
+    {
+      "train_step": 4731,
+      "loss": 0.5104249129307188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5104249129307188
+    },
+    {
+      "train_step": 4732,
+      "loss": 0.33538466366061526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33538466366061526
+    },
+    {
+      "train_step": 4733,
+      "loss": 0.790926291329886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.790926291329886
+    },
+    {
+      "train_step": 4734,
+      "loss": 0.4648764988259518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4648764988259518
+    },
+    {
+      "train_step": 4735,
+      "loss": 0.5618355582820591,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5618355582820591
+    },
+    {
+      "train_step": 4736,
+      "loss": 0.4875087518454645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4875087518454645
+    },
+    {
+      "train_step": 4737,
+      "loss": 0.39140362518501265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39140362518501265
+    },
+    {
+      "train_step": 4738,
+      "loss": 0.4506249144450723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4506249144450723
+    },
+    {
+      "train_step": 4739,
+      "loss": 0.5867110629072138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5867110629072138
+    },
+    {
+      "train_step": 4740,
+      "loss": 0.8216539479324176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8216539479324176
+    },
+    {
+      "train_step": 4741,
+      "loss": 0.5557082535813667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5557082535813667
+    },
+    {
+      "train_step": 4742,
+      "loss": 0.5131588218727583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5131588218727583
+    },
+    {
+      "train_step": 4743,
+      "loss": 0.6469354663773085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6469354663773085
+    },
+    {
+      "train_step": 4744,
+      "loss": 0.49486722566869434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49486722566869434
+    },
+    {
+      "train_step": 4745,
+      "loss": 0.4871089265328642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4871089265328642
+    },
+    {
+      "train_step": 4746,
+      "loss": 0.5636380781539287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5636380781539287
+    },
+    {
+      "train_step": 4747,
+      "loss": 0.7464838630280463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7464838630280463
+    },
+    {
+      "train_step": 4748,
+      "loss": 0.6923983738859683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6923983738859683
+    },
+    {
+      "train_step": 4749,
+      "loss": 0.6508297340902508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6508297340902508
+    },
+    {
+      "train_step": 4750,
+      "loss": 0.40045218102914815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40045218102914815
+    },
+    {
+      "train_step": 4751,
+      "loss": 0.5848343101319066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5848343101319066
+    },
+    {
+      "train_step": 4752,
+      "loss": 0.5584736200746565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5584736200746565
+    },
+    {
+      "train_step": 4753,
+      "loss": 0.6030882178304668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6030882178304668
+    },
+    {
+      "train_step": 4754,
+      "loss": 0.6085074345488957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6085074345488957
+    },
+    {
+      "train_step": 4755,
+      "loss": 0.5352699310816026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5352699310816026
+    },
+    {
+      "train_step": 4756,
+      "loss": 0.42102736642351857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42102736642351857
+    },
+    {
+      "train_step": 4757,
+      "loss": 0.45253910344619697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45253910344619697
+    },
+    {
+      "train_step": 4758,
+      "loss": 0.43079360032072034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43079360032072034
+    },
+    {
+      "train_step": 4759,
+      "loss": 0.6191728258828714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6191728258828714
+    },
+    {
+      "train_step": 4760,
+      "loss": 0.6213688442829031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6213688442829031
+    },
+    {
+      "train_step": 4761,
+      "loss": 0.36700596237758926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36700596237758926
+    },
+    {
+      "train_step": 4762,
+      "loss": 0.6232052848139282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6232052848139282
+    },
+    {
+      "train_step": 4763,
+      "loss": 0.7660566958348748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7660566958348748
+    },
+    {
+      "train_step": 4764,
+      "loss": 0.8089665666836294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8089665666836294
+    },
+    {
+      "train_step": 4765,
+      "loss": 0.41487801963368703,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41487801963368703
+    },
+    {
+      "train_step": 4766,
+      "loss": 0.6353860397092249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6353860397092249
+    },
+    {
+      "train_step": 4767,
+      "loss": 0.3306631606094529,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3306631606094529
+    },
+    {
+      "train_step": 4768,
+      "loss": 0.5623863984917976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5623863984917976
+    },
+    {
+      "train_step": 4769,
+      "loss": 0.3709823404485886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3709823404485886
+    },
+    {
+      "train_step": 4770,
+      "loss": 0.5619296669518258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5619296669518258
+    },
+    {
+      "train_step": 4771,
+      "loss": 0.5438478198537312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5438478198537312
+    },
+    {
+      "train_step": 4772,
+      "loss": 0.6068493234695124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6068493234695124
+    },
+    {
+      "train_step": 4773,
+      "loss": 0.42905556275404183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42905556275404183
+    },
+    {
+      "train_step": 4774,
+      "loss": 0.4870541685489503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4870541685489503
+    },
+    {
+      "train_step": 4775,
+      "loss": 0.5226764337320737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5226764337320737
+    },
+    {
+      "train_step": 4776,
+      "loss": 0.552096762527784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.552096762527784
+    },
+    {
+      "train_step": 4777,
+      "loss": 0.6839288981616809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6839288981616809
+    },
+    {
+      "train_step": 4778,
+      "loss": 0.6348500501509278,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6348500501509278
+    },
+    {
+      "train_step": 4779,
+      "loss": 0.6614837590308021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6614837590308021
+    },
+    {
+      "train_step": 4780,
+      "loss": 0.48324781024443536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48324781024443536
+    },
+    {
+      "train_step": 4781,
+      "loss": 0.6285091457511598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6285091457511598
+    },
+    {
+      "train_step": 4782,
+      "loss": 0.6708091226926607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6708091226926607
+    },
+    {
+      "train_step": 4783,
+      "loss": 0.530583378229365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.530583378229365
+    },
+    {
+      "train_step": 4784,
+      "loss": 0.6883297378367466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6883297378367466
+    },
+    {
+      "train_step": 4785,
+      "loss": 0.5834896506881476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5834896506881476
+    },
+    {
+      "train_step": 4786,
+      "loss": 0.856451252979582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.856451252979582
+    },
+    {
+      "train_step": 4787,
+      "loss": 0.5853090676818488,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5853090676818488
+    },
+    {
+      "train_step": 4788,
+      "loss": 0.6275380205335792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6275380205335792
+    },
+    {
+      "train_step": 4789,
+      "loss": 0.5775080322456909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5775080322456909
+    },
+    {
+      "train_step": 4790,
+      "loss": 0.5151622665349636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5151622665349636
+    },
+    {
+      "train_step": 4791,
+      "loss": 0.6120767815113177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6120767815113177
+    },
+    {
+      "train_step": 4792,
+      "loss": 0.41824507950795864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41824507950795864
+    },
+    {
+      "train_step": 4793,
+      "loss": 0.6491939411615837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6491939411615837
+    },
+    {
+      "train_step": 4794,
+      "loss": 0.6576437875270144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6576437875270144
+    },
+    {
+      "train_step": 4795,
+      "loss": 0.5036583334805641,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5036583334805641
+    },
+    {
+      "train_step": 4796,
+      "loss": 0.5192489729206455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5192489729206455
+    },
+    {
+      "train_step": 4797,
+      "loss": 0.5736259273983259,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5736259273983259
+    },
+    {
+      "train_step": 4798,
+      "loss": 0.7637404866832789,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7637404866832789
+    },
+    {
+      "train_step": 4799,
+      "loss": 0.6645240150255288,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6645240150255288
+    },
+    {
+      "train_step": 4800,
+      "loss": 0.510846385957795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.510846385957795
+    },
+    {
+      "train_step": 4801,
+      "loss": 0.6070001582488835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6070001582488835
+    },
+    {
+      "train_step": 4802,
+      "loss": 0.7051877073160792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7051877073160792
+    },
+    {
+      "train_step": 4803,
+      "loss": 0.5784581291308669,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5784581291308669
+    },
+    {
+      "train_step": 4804,
+      "loss": 0.5292941326751325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5292941326751325
+    },
+    {
+      "train_step": 4805,
+      "loss": 0.46296283691764306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46296283691764306
+    },
+    {
+      "train_step": 4806,
+      "loss": 0.6705095702500802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6705095702500802
+    },
+    {
+      "train_step": 4807,
+      "loss": 0.5295969738404236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5295969738404236
+    },
+    {
+      "train_step": 4808,
+      "loss": 0.33076293992877454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33076293992877454
+    },
+    {
+      "train_step": 4809,
+      "loss": 0.7946432629201419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7946432629201419
+    },
+    {
+      "train_step": 4810,
+      "loss": 0.6101858465079348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6101858465079348
+    },
+    {
+      "train_step": 4811,
+      "loss": 0.4594126025199609,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4594126025199609
+    },
+    {
+      "train_step": 4812,
+      "loss": 0.6034561685423454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6034561685423454
+    },
+    {
+      "train_step": 4813,
+      "loss": 0.4375630432320198,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4375630432320198
+    },
+    {
+      "train_step": 4814,
+      "loss": 0.6919267342996458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6919267342996458
+    },
+    {
+      "train_step": 4815,
+      "loss": 0.6891593640483248,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6891593640483248
+    },
+    {
+      "train_step": 4816,
+      "loss": 0.5079385412842561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5079385412842561
+    },
+    {
+      "train_step": 4817,
+      "loss": 0.5850016873314441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5850016873314441
+    },
+    {
+      "train_step": 4818,
+      "loss": 0.5693548307719835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5693548307719835
+    },
+    {
+      "train_step": 4819,
+      "loss": 0.7155746133884796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7155746133884796
+    },
+    {
+      "train_step": 4820,
+      "loss": 0.6318623154350126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6318623154350126
+    },
+    {
+      "train_step": 4821,
+      "loss": 0.5661892698578818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5661892698578818
+    },
+    {
+      "train_step": 4822,
+      "loss": 0.6551008292320055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6551008292320055
+    },
+    {
+      "train_step": 4823,
+      "loss": 0.6681568832788829,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6681568832788829
+    },
+    {
+      "train_step": 4824,
+      "loss": 0.40851188369983077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40851188369983077
+    },
+    {
+      "train_step": 4825,
+      "loss": 0.5843228844744605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5843228844744605
+    },
+    {
+      "train_step": 4826,
+      "loss": 0.5010719168991121,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5010719168991121
+    },
+    {
+      "train_step": 4827,
+      "loss": 0.5711886421856287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5711886421856287
+    },
+    {
+      "train_step": 4828,
+      "loss": 0.6401271368619218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6401271368619218
+    },
+    {
+      "train_step": 4829,
+      "loss": 0.43256073470480566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43256073470480566
+    },
+    {
+      "train_step": 4830,
+      "loss": 0.6504512850224966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6504512850224966
+    },
+    {
+      "train_step": 4831,
+      "loss": 0.6557361772861748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6557361772861748
+    },
+    {
+      "train_step": 4832,
+      "loss": 0.5339740804088066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5339740804088066
+    },
+    {
+      "train_step": 4833,
+      "loss": 0.5880120379619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5880120379619
+    },
+    {
+      "train_step": 4834,
+      "loss": 0.5895427677952341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5895427677952341
+    },
+    {
+      "train_step": 4835,
+      "loss": 0.6991060212957432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6991060212957432
+    },
+    {
+      "train_step": 4836,
+      "loss": 0.4599130594089354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4599130594089354
+    },
+    {
+      "train_step": 4837,
+      "loss": 0.6879901858625064,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6879901858625064
+    },
+    {
+      "train_step": 4838,
+      "loss": 0.7766878259170467,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7766878259170467
+    },
+    {
+      "train_step": 4839,
+      "loss": 0.7495709998321745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7495709998321745
+    },
+    {
+      "train_step": 4840,
+      "loss": 0.5996043068403476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5996043068403476
+    },
+    {
+      "train_step": 4841,
+      "loss": 0.7237292292846442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7237292292846442
+    },
+    {
+      "train_step": 4842,
+      "loss": 0.3768733555458033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3768733555458033
+    },
+    {
+      "train_step": 4843,
+      "loss": 0.412040101227004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.412040101227004
+    },
+    {
+      "train_step": 4844,
+      "loss": 0.5279722012710509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5279722012710509
+    },
+    {
+      "train_step": 4845,
+      "loss": 0.5962600713843794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962600713843794
+    },
+    {
+      "train_step": 4846,
+      "loss": 0.5452129169457166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5452129169457166
+    },
+    {
+      "train_step": 4847,
+      "loss": 0.6105928911420744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6105928911420744
+    },
+    {
+      "train_step": 4848,
+      "loss": 0.4683748308590047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4683748308590047
+    },
+    {
+      "train_step": 4849,
+      "loss": 0.6114295200934003,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6114295200934003
+    },
+    {
+      "train_step": 4850,
+      "loss": 0.6144805116616029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6144805116616029
+    },
+    {
+      "train_step": 4851,
+      "loss": 0.6382022507674205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6382022507674205
+    },
+    {
+      "train_step": 4852,
+      "loss": 0.5393772234599057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5393772234599057
+    },
+    {
+      "train_step": 4853,
+      "loss": 0.5613109628678297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5613109628678297
+    },
+    {
+      "train_step": 4854,
+      "loss": 0.5483897553050968,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5483897553050968
+    },
+    {
+      "train_step": 4855,
+      "loss": 0.46698836072603617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46698836072603617
+    },
+    {
+      "train_step": 4856,
+      "loss": 0.5471902321193409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5471902321193409
+    },
+    {
+      "train_step": 4857,
+      "loss": 0.7435087370803475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7435087370803475
+    },
+    {
+      "train_step": 4858,
+      "loss": 0.6109530366194408,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6109530366194408
+    },
+    {
+      "train_step": 4859,
+      "loss": 0.612217011827573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.612217011827573
+    },
+    {
+      "train_step": 4860,
+      "loss": 0.6833771119831706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6833771119831706
+    },
+    {
+      "train_step": 4861,
+      "loss": 0.5235114232594436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5235114232594436
+    },
+    {
+      "train_step": 4862,
+      "loss": 0.6308332587432355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6308332587432355
+    },
+    {
+      "train_step": 4863,
+      "loss": 0.5999802281444001,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5999802281444001
+    },
+    {
+      "train_step": 4864,
+      "loss": 0.5702196345189242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5702196345189242
+    },
+    {
+      "train_step": 4865,
+      "loss": 0.53622933981548,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.53622933981548
+    },
+    {
+      "train_step": 4866,
+      "loss": 0.6387866793017145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6387866793017145
+    },
+    {
+      "train_step": 4867,
+      "loss": 0.37124407483394695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37124407483394695
+    },
+    {
+      "train_step": 4868,
+      "loss": 0.6780935498520976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6780935498520976
+    },
+    {
+      "train_step": 4869,
+      "loss": 0.5964359038820612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5964359038820612
+    },
+    {
+      "train_step": 4870,
+      "loss": 0.5281732227412521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5281732227412521
+    },
+    {
+      "train_step": 4871,
+      "loss": 0.6639368023393906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6639368023393906
+    },
+    {
+      "train_step": 4872,
+      "loss": 0.5903229151996435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5903229151996435
+    },
+    {
+      "train_step": 4873,
+      "loss": 0.7153492176734229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7153492176734229
+    },
+    {
+      "train_step": 4874,
+      "loss": 0.759936350424824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.759936350424824
+    },
+    {
+      "train_step": 4875,
+      "loss": 0.652052551324242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.652052551324242
+    },
+    {
+      "train_step": 4876,
+      "loss": 0.5786642542175608,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5786642542175608
+    },
+    {
+      "train_step": 4877,
+      "loss": 0.47875482560898674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47875482560898674
+    },
+    {
+      "train_step": 4878,
+      "loss": 0.36859841073132715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36859841073132715
+    },
+    {
+      "train_step": 4879,
+      "loss": 0.4319310573066509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4319310573066509
+    },
+    {
+      "train_step": 4880,
+      "loss": 0.581288553993796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.581288553993796
+    },
+    {
+      "train_step": 4881,
+      "loss": 0.673838188588707,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.673838188588707
+    },
+    {
+      "train_step": 4882,
+      "loss": 0.4563335627102099,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4563335627102099
+    },
+    {
+      "train_step": 4883,
+      "loss": 0.4046691935627416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4046691935627416
+    },
+    {
+      "train_step": 4884,
+      "loss": 0.4915622097529353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4915622097529353
+    },
+    {
+      "train_step": 4885,
+      "loss": 0.6442486995326594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6442486995326594
+    },
+    {
+      "train_step": 4886,
+      "loss": 0.5657930212212935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5657930212212935
+    },
+    {
+      "train_step": 4887,
+      "loss": 0.4457443460018723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4457443460018723
+    },
+    {
+      "train_step": 4888,
+      "loss": 0.3825041158405124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3825041158405124
+    },
+    {
+      "train_step": 4889,
+      "loss": 0.5868406299990734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5868406299990734
+    },
+    {
+      "train_step": 4890,
+      "loss": 0.4990050213249098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4990050213249098
+    },
+    {
+      "train_step": 4891,
+      "loss": 0.7535912998489633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7535912998489633
+    },
+    {
+      "train_step": 4892,
+      "loss": 0.4419121067277134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4419121067277134
+    },
+    {
+      "train_step": 4893,
+      "loss": 0.6960120973373342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6960120973373342
+    },
+    {
+      "train_step": 4894,
+      "loss": 0.7617422478050738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7617422478050738
+    },
+    {
+      "train_step": 4895,
+      "loss": 0.670329819026425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.670329819026425
+    },
+    {
+      "train_step": 4896,
+      "loss": 0.6151936709953832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6151936709953832
+    },
+    {
+      "train_step": 4897,
+      "loss": 0.7409825705444355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7409825705444355
+    },
+    {
+      "train_step": 4898,
+      "loss": 0.5373693093846217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5373693093846217
+    },
+    {
+      "train_step": 4899,
+      "loss": 0.4445819315468129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4445819315468129
+    },
+    {
+      "train_step": 4900,
+      "loss": 0.4527292145423559,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4527292145423559
+    },
+    {
+      "train_step": 4901,
+      "loss": 0.5769326094540821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5769326094540821
+    },
+    {
+      "train_step": 4902,
+      "loss": 0.5107018547387053,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5107018547387053
+    },
+    {
+      "train_step": 4903,
+      "loss": 0.6706661841807794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6706661841807794
+    },
+    {
+      "train_step": 4904,
+      "loss": 0.7062603706051311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7062603706051311
+    },
+    {
+      "train_step": 4905,
+      "loss": 0.5624027624886161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5624027624886161
+    },
+    {
+      "train_step": 4906,
+      "loss": 0.5347460299034666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5347460299034666
+    },
+    {
+      "train_step": 4907,
+      "loss": 0.7104115114766812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7104115114766812
+    },
+    {
+      "train_step": 4908,
+      "loss": 0.6420487570749324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6420487570749324
+    },
+    {
+      "train_step": 4909,
+      "loss": 0.640666999006586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.640666999006586
+    },
+    {
+      "train_step": 4910,
+      "loss": 0.7347366786460092,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7347366786460092
+    },
+    {
+      "train_step": 4911,
+      "loss": 0.4086826400718055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4086826400718055
+    },
+    {
+      "train_step": 4912,
+      "loss": 0.7222312654142637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7222312654142637
+    },
+    {
+      "train_step": 4913,
+      "loss": 0.5188674811704302,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5188674811704302
+    },
+    {
+      "train_step": 4914,
+      "loss": 0.6818051562355021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6818051562355021
+    },
+    {
+      "train_step": 4915,
+      "loss": 0.5148296005595057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5148296005595057
+    },
+    {
+      "train_step": 4916,
+      "loss": 0.48403059174969987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48403059174969987
+    },
+    {
+      "train_step": 4917,
+      "loss": 0.557295401925244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.557295401925244
+    },
+    {
+      "train_step": 4918,
+      "loss": 0.8998728810160959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8998728810160959
+    },
+    {
+      "train_step": 4919,
+      "loss": 0.3098261135332018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3098261135332018
+    },
+    {
+      "train_step": 4920,
+      "loss": 0.6226763956093287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6226763956093287
+    },
+    {
+      "train_step": 4921,
+      "loss": 0.6509824817030719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6509824817030719
+    },
+    {
+      "train_step": 4922,
+      "loss": 0.5467538044669193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5467538044669193
+    },
+    {
+      "train_step": 4923,
+      "loss": 0.5049685217054156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5049685217054156
+    },
+    {
+      "train_step": 4924,
+      "loss": 0.5160663613507372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5160663613507372
+    },
+    {
+      "train_step": 4925,
+      "loss": 0.5373446570675539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5373446570675539
+    },
+    {
+      "train_step": 4926,
+      "loss": 0.5279310962210559,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5279310962210559
+    },
+    {
+      "train_step": 4927,
+      "loss": 0.3887974631239092,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3887974631239092
+    },
+    {
+      "train_step": 4928,
+      "loss": 0.5755327881966975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5755327881966975
+    },
+    {
+      "train_step": 4929,
+      "loss": 0.5434923408125775,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5434923408125775
+    },
+    {
+      "train_step": 4930,
+      "loss": 0.6150006638008249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6150006638008249
+    },
+    {
+      "train_step": 4931,
+      "loss": 0.44442962757853327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44442962757853327
+    },
+    {
+      "train_step": 4932,
+      "loss": 0.63230982124938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.63230982124938
+    },
+    {
+      "train_step": 4933,
+      "loss": 0.5548163340308095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5548163340308095
+    },
+    {
+      "train_step": 4934,
+      "loss": 0.5089316812714719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5089316812714719
+    },
+    {
+      "train_step": 4935,
+      "loss": 0.6154489430765572,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6154489430765572
+    },
+    {
+      "train_step": 4936,
+      "loss": 0.5571652735727737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5571652735727737
+    },
+    {
+      "train_step": 4937,
+      "loss": 0.4935088501094545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4935088501094545
+    },
+    {
+      "train_step": 4938,
+      "loss": 0.7939974089493322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7939974089493322
+    },
+    {
+      "train_step": 4939,
+      "loss": 0.6074253787585843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6074253787585843
+    },
+    {
+      "train_step": 4940,
+      "loss": 0.6128733902391531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6128733902391531
+    },
+    {
+      "train_step": 4941,
+      "loss": 0.48109377937536163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48109377937536163
+    },
+    {
+      "train_step": 4942,
+      "loss": 0.5523620259265847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5523620259265847
+    },
+    {
+      "train_step": 4943,
+      "loss": 0.628495767611015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.628495767611015
+    },
+    {
+      "train_step": 4944,
+      "loss": 0.5634218947910874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5634218947910874
+    },
+    {
+      "train_step": 4945,
+      "loss": 0.4386546753780456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4386546753780456
+    },
+    {
+      "train_step": 4946,
+      "loss": 0.7095074109199854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7095074109199854
+    },
+    {
+      "train_step": 4947,
+      "loss": 0.6518954986300947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6518954986300947
+    },
+    {
+      "train_step": 4948,
+      "loss": 0.5854167248192809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5854167248192809
+    },
+    {
+      "train_step": 4949,
+      "loss": 0.6782662763717539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6782662763717539
+    },
+    {
+      "train_step": 4950,
+      "loss": 0.57222635150328,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.57222635150328
+    },
+    {
+      "train_step": 4951,
+      "loss": 0.5272760580852149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5272760580852149
+    },
+    {
+      "train_step": 4952,
+      "loss": 0.6078737114432267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6078737114432267
+    },
+    {
+      "train_step": 4953,
+      "loss": 0.5878049944861041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5878049944861041
+    },
+    {
+      "train_step": 4954,
+      "loss": 0.5297999029199061,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5297999029199061
+    },
+    {
+      "train_step": 4955,
+      "loss": 0.7441679177553078,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7441679177553078
+    },
+    {
+      "train_step": 4956,
+      "loss": 0.49827195642148775,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49827195642148775
+    },
+    {
+      "train_step": 4957,
+      "loss": 0.4293869315809706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4293869315809706
+    },
+    {
+      "train_step": 4958,
+      "loss": 0.5670198748842792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5670198748842792
+    },
+    {
+      "train_step": 4959,
+      "loss": 0.5910876551159828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5910876551159828
+    },
+    {
+      "train_step": 4960,
+      "loss": 0.6029385010739983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6029385010739983
+    },
+    {
+      "train_step": 4961,
+      "loss": 0.5513379029048059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5513379029048059
+    },
+    {
+      "train_step": 4962,
+      "loss": 0.50926897060163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.50926897060163
+    },
+    {
+      "train_step": 4963,
+      "loss": 0.5692330558841083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5692330558841083
+    },
+    {
+      "train_step": 4964,
+      "loss": 0.6550324614243928,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6550324614243928
+    },
+    {
+      "train_step": 4965,
+      "loss": 0.7198177357892442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7198177357892442
+    },
+    {
+      "train_step": 4966,
+      "loss": 0.5109017628251115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5109017628251115
+    },
+    {
+      "train_step": 4967,
+      "loss": 0.692629593214124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.692629593214124
+    },
+    {
+      "train_step": 4968,
+      "loss": 0.576155785681769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576155785681769
+    },
+    {
+      "train_step": 4969,
+      "loss": 0.4241827866629513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4241827866629513
+    },
+    {
+      "train_step": 4970,
+      "loss": 0.4236611091210412,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4236611091210412
+    },
+    {
+      "train_step": 4971,
+      "loss": 0.5801343865190485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5801343865190485
+    },
+    {
+      "train_step": 4972,
+      "loss": 0.5760997428138994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5760997428138994
+    },
+    {
+      "train_step": 4973,
+      "loss": 0.8003032870964321,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8003032870964321
+    },
+    {
+      "train_step": 4974,
+      "loss": 0.45277779812609714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45277779812609714
+    },
+    {
+      "train_step": 4975,
+      "loss": 0.5224827495866805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5224827495866805
+    },
+    {
+      "train_step": 4976,
+      "loss": 0.636592015788353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.636592015788353
+    },
+    {
+      "train_step": 4977,
+      "loss": 0.7691621922227712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7691621922227712
+    },
+    {
+      "train_step": 4978,
+      "loss": 0.7193467899302055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7193467899302055
+    },
+    {
+      "train_step": 4979,
+      "loss": 0.57214628397224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.57214628397224
+    },
+    {
+      "train_step": 4980,
+      "loss": 0.5640196290993809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5640196290993809
+    },
+    {
+      "train_step": 4981,
+      "loss": 0.4838846380534536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4838846380534536
+    },
+    {
+      "train_step": 4982,
+      "loss": 0.6088792869502039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6088792869502039
+    },
+    {
+      "train_step": 4983,
+      "loss": 0.5076064221124854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5076064221124854
+    },
+    {
+      "train_step": 4984,
+      "loss": 0.4002036521499957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4002036521499957
+    },
+    {
+      "train_step": 4985,
+      "loss": 0.609956143151338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.609956143151338
+    },
+    {
+      "train_step": 4986,
+      "loss": 0.7829350016041445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7829350016041445
+    },
+    {
+      "train_step": 4987,
+      "loss": 0.574381786126849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.574381786126849
+    },
+    {
+      "train_step": 4988,
+      "loss": 0.5570212573128572,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5570212573128572
+    },
+    {
+      "train_step": 4989,
+      "loss": 0.46440858156937986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46440858156937986
+    },
+    {
+      "train_step": 4990,
+      "loss": 0.6473218298482626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6473218298482626
+    },
+    {
+      "train_step": 4991,
+      "loss": 0.38028886584428556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38028886584428556
+    },
+    {
+      "train_step": 4992,
+      "loss": 0.518993047153756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.518993047153756
+    },
+    {
+      "train_step": 4993,
+      "loss": 0.6875019573273625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6875019573273625
+    },
+    {
+      "train_step": 4994,
+      "loss": 0.84619777555015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.84619777555015
+    },
+    {
+      "train_step": 4995,
+      "loss": 0.6872452530224916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6872452530224916
+    },
+    {
+      "train_step": 4996,
+      "loss": 0.6837023374259795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6837023374259795
+    },
+    {
+      "train_step": 4997,
+      "loss": 0.5577067087463846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5577067087463846
+    },
+    {
+      "train_step": 4998,
+      "loss": 0.39829809057185295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39829809057185295
+    },
+    {
+      "train_step": 4999,
+      "loss": 0.5869051009795129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5869051009795129
+    },
+    {
+      "train_step": 5000,
+      "loss": 0.4395944609793066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4395944609793066
+    },
+    {
+      "train_step": 5001,
+      "loss": 0.49678533696722477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49678533696722477
+    },
+    {
+      "train_step": 5002,
+      "loss": 0.5892189794859994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5892189794859994
+    },
+    {
+      "train_step": 5003,
+      "loss": 0.7195198390091924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7195198390091924
+    },
+    {
+      "train_step": 5004,
+      "loss": 0.4921779151709752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4921779151709752
+    },
+    {
+      "train_step": 5005,
+      "loss": 0.623467841521732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.623467841521732
+    },
+    {
+      "train_step": 5006,
+      "loss": 0.5975484698616087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5975484698616087
+    },
+    {
+      "train_step": 5007,
+      "loss": 0.6267663643370077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6267663643370077
+    },
+    {
+      "train_step": 5008,
+      "loss": 0.6242338652660349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6242338652660349
+    },
+    {
+      "train_step": 5009,
+      "loss": 0.6380799793088076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6380799793088076
+    },
+    {
+      "train_step": 5010,
+      "loss": 0.5552206761256447,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5552206761256447
+    },
+    {
+      "train_step": 5011,
+      "loss": 0.6826757542680897,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6826757542680897
+    },
+    {
+      "train_step": 5012,
+      "loss": 0.5704571897136975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5704571897136975
+    },
+    {
+      "train_step": 5013,
+      "loss": 0.45908043224854617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45908043224854617
+    },
+    {
+      "train_step": 5014,
+      "loss": 0.46848393987734704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46848393987734704
+    },
+    {
+      "train_step": 5015,
+      "loss": 0.5049559548199013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5049559548199013
+    },
+    {
+      "train_step": 5016,
+      "loss": 0.7707542800932614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7707542800932614
+    },
+    {
+      "train_step": 5017,
+      "loss": 0.7127137915989296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7127137915989296
+    },
+    {
+      "train_step": 5018,
+      "loss": 0.464010516571282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.464010516571282
+    },
+    {
+      "train_step": 5019,
+      "loss": 0.5180024141535051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5180024141535051
+    },
+    {
+      "train_step": 5020,
+      "loss": 0.34147081606261986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34147081606261986
+    },
+    {
+      "train_step": 5021,
+      "loss": 0.5419435527890947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5419435527890947
+    },
+    {
+      "train_step": 5022,
+      "loss": 0.5114231302231187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5114231302231187
+    },
+    {
+      "train_step": 5023,
+      "loss": 0.46316012592209693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46316012592209693
+    },
+    {
+      "train_step": 5024,
+      "loss": 0.52067553476655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.52067553476655
+    },
+    {
+      "train_step": 5025,
+      "loss": 0.5791397554847119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5791397554847119
+    },
+    {
+      "train_step": 5026,
+      "loss": 0.5460441778143814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5460441778143814
+    },
+    {
+      "train_step": 5027,
+      "loss": 0.5813832264179936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5813832264179936
+    },
+    {
+      "train_step": 5028,
+      "loss": 0.7879887295031459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7879887295031459
+    },
+    {
+      "train_step": 5029,
+      "loss": 0.4668743854821058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4668743854821058
+    },
+    {
+      "train_step": 5030,
+      "loss": 0.5768871391756474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5768871391756474
+    },
+    {
+      "train_step": 5031,
+      "loss": 0.5300658224376533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5300658224376533
+    },
+    {
+      "train_step": 5032,
+      "loss": 0.6991485775466628,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6991485775466628
+    },
+    {
+      "train_step": 5033,
+      "loss": 0.6883726601220457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6883726601220457
+    },
+    {
+      "train_step": 5034,
+      "loss": 0.7308320642478239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7308320642478239
+    },
+    {
+      "train_step": 5035,
+      "loss": 0.6524440946937763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6524440946937763
+    },
+    {
+      "train_step": 5036,
+      "loss": 0.6957638570619379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6957638570619379
+    },
+    {
+      "train_step": 5037,
+      "loss": 0.6771833640099333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6771833640099333
+    },
+    {
+      "train_step": 5038,
+      "loss": 0.48815448550521157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48815448550521157
+    },
+    {
+      "train_step": 5039,
+      "loss": 0.3963731362644833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3963731362644833
+    },
+    {
+      "train_step": 5040,
+      "loss": 0.6766995067418133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6766995067418133
+    },
+    {
+      "train_step": 5041,
+      "loss": 0.5216499105088825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5216499105088825
+    },
+    {
+      "train_step": 5042,
+      "loss": 0.376205861148218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.376205861148218
+    },
+    {
+      "train_step": 5043,
+      "loss": 0.3593689427050495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3593689427050495
+    },
+    {
+      "train_step": 5044,
+      "loss": 0.7827561486500757,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7827561486500757
+    },
+    {
+      "train_step": 5045,
+      "loss": 0.5168632132851245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5168632132851245
+    },
+    {
+      "train_step": 5046,
+      "loss": 0.44414998789961746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44414998789961746
+    },
+    {
+      "train_step": 5047,
+      "loss": 0.5828400325101277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5828400325101277
+    },
+    {
+      "train_step": 5048,
+      "loss": 0.5335830465331551,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5335830465331551
+    },
+    {
+      "train_step": 5049,
+      "loss": 0.6083662123593369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6083662123593369
+    },
+    {
+      "train_step": 5050,
+      "loss": 0.5438407695238922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5438407695238922
+    },
+    {
+      "train_step": 5051,
+      "loss": 0.4818247569539655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4818247569539655
+    },
+    {
+      "train_step": 5052,
+      "loss": 0.7632772414543182,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7632772414543182
+    },
+    {
+      "train_step": 5053,
+      "loss": 0.5207246106258312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5207246106258312
+    },
+    {
+      "train_step": 5054,
+      "loss": 0.5871918137666844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5871918137666844
+    },
+    {
+      "train_step": 5055,
+      "loss": 0.6895218037495434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6895218037495434
+    },
+    {
+      "train_step": 5056,
+      "loss": 0.5344477859783323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5344477859783323
+    },
+    {
+      "train_step": 5057,
+      "loss": 0.49513531507663466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49513531507663466
+    },
+    {
+      "train_step": 5058,
+      "loss": 0.5931120456533282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5931120456533282
+    },
+    {
+      "train_step": 5059,
+      "loss": 0.6264402621231407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6264402621231407
+    },
+    {
+      "train_step": 5060,
+      "loss": 0.37219418641555135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37219418641555135
+    },
+    {
+      "train_step": 5061,
+      "loss": 0.6879685813003174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6879685813003174
+    },
+    {
+      "train_step": 5062,
+      "loss": 0.6182010433668959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6182010433668959
+    },
+    {
+      "train_step": 5063,
+      "loss": 0.6819369873404181,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6819369873404181
+    },
+    {
+      "train_step": 5064,
+      "loss": 0.6301517629758344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6301517629758344
+    },
+    {
+      "train_step": 5065,
+      "loss": 0.684429988439876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.684429988439876
+    },
+    {
+      "train_step": 5066,
+      "loss": 0.5944528293284156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5944528293284156
+    },
+    {
+      "train_step": 5067,
+      "loss": 0.7207741335328612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7207741335328612
+    },
+    {
+      "train_step": 5068,
+      "loss": 0.779973275929771,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.779973275929771
+    },
+    {
+      "train_step": 5069,
+      "loss": 0.505622429373624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.505622429373624
+    },
+    {
+      "train_step": 5070,
+      "loss": 0.543011532980448,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.543011532980448
+    },
+    {
+      "train_step": 5071,
+      "loss": 0.4688570063780346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4688570063780346
+    },
+    {
+      "train_step": 5072,
+      "loss": 0.546896276429192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.546896276429192
+    },
+    {
+      "train_step": 5073,
+      "loss": 0.5204901551849604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5204901551849604
+    },
+    {
+      "train_step": 5074,
+      "loss": 0.49275155234990103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49275155234990103
+    },
+    {
+      "train_step": 5075,
+      "loss": 0.6800140170431865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6800140170431865
+    },
+    {
+      "train_step": 5076,
+      "loss": 0.6524859837430124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6524859837430124
+    },
+    {
+      "train_step": 5077,
+      "loss": 0.7569548894007809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7569548894007809
+    },
+    {
+      "train_step": 5078,
+      "loss": 0.48771024676613195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48771024676613195
+    },
+    {
+      "train_step": 5079,
+      "loss": 0.6108806082078257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6108806082078257
+    },
+    {
+      "train_step": 5080,
+      "loss": 0.6259648359254513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6259648359254513
+    },
+    {
+      "train_step": 5081,
+      "loss": 0.5787236388527579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5787236388527579
+    },
+    {
+      "train_step": 5082,
+      "loss": 0.4387283840463694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4387283840463694
+    },
+    {
+      "train_step": 5083,
+      "loss": 0.4910985872362029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4910985872362029
+    },
+    {
+      "train_step": 5084,
+      "loss": 0.45961055193895917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45961055193895917
+    },
+    {
+      "train_step": 5085,
+      "loss": 0.6963405226373282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6963405226373282
+    },
+    {
+      "train_step": 5086,
+      "loss": 0.405463531806418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.405463531806418
+    },
+    {
+      "train_step": 5087,
+      "loss": 0.523187947821723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.523187947821723
+    },
+    {
+      "train_step": 5088,
+      "loss": 0.6572187235001271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6572187235001271
+    },
+    {
+      "train_step": 5089,
+      "loss": 0.6128171530222343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6128171530222343
+    },
+    {
+      "train_step": 5090,
+      "loss": 0.6426690528997406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6426690528997406
+    },
+    {
+      "train_step": 5091,
+      "loss": 0.5698090736061574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5698090736061574
+    },
+    {
+      "train_step": 5092,
+      "loss": 0.5486959047119514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5486959047119514
+    },
+    {
+      "train_step": 5093,
+      "loss": 0.6413817676658199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6413817676658199
+    },
+    {
+      "train_step": 5094,
+      "loss": 0.5860156993367056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5860156993367056
+    },
+    {
+      "train_step": 5095,
+      "loss": 0.6649260519713608,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6649260519713608
+    },
+    {
+      "train_step": 5096,
+      "loss": 0.6050593516375347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6050593516375347
+    },
+    {
+      "train_step": 5097,
+      "loss": 0.6386224234880246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6386224234880246
+    },
+    {
+      "train_step": 5098,
+      "loss": 0.5112053459833018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5112053459833018
+    },
+    {
+      "train_step": 5099,
+      "loss": 0.42419931344419753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42419931344419753
+    },
+    {
+      "train_step": 5100,
+      "loss": 0.5635315685227983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5635315685227983
+    },
+    {
+      "train_step": 5101,
+      "loss": 0.5750498644419382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5750498644419382
+    },
+    {
+      "train_step": 5102,
+      "loss": 0.5284697251403043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5284697251403043
+    },
+    {
+      "train_step": 5103,
+      "loss": 0.6194520529968471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6194520529968471
+    },
+    {
+      "train_step": 5104,
+      "loss": 0.6727686420769224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6727686420769224
+    },
+    {
+      "train_step": 5105,
+      "loss": 0.4956500145575324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4956500145575324
+    },
+    {
+      "train_step": 5106,
+      "loss": 0.6324384365838581,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6324384365838581
+    },
+    {
+      "train_step": 5107,
+      "loss": 0.49390253672468537,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49390253672468537
+    },
+    {
+      "train_step": 5108,
+      "loss": 0.5743577469427615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5743577469427615
+    },
+    {
+      "train_step": 5109,
+      "loss": 0.775576278898893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.775576278898893
+    },
+    {
+      "train_step": 5110,
+      "loss": 0.812358956908871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.812358956908871
+    },
+    {
+      "train_step": 5111,
+      "loss": 0.47097923046110685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47097923046110685
+    },
+    {
+      "train_step": 5112,
+      "loss": 0.5386616743343268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5386616743343268
+    },
+    {
+      "train_step": 5113,
+      "loss": 0.6404442450953151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6404442450953151
+    },
+    {
+      "train_step": 5114,
+      "loss": 0.4234580502861718,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4234580502861718
+    },
+    {
+      "train_step": 5115,
+      "loss": 0.5594622009502909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5594622009502909
+    },
+    {
+      "train_step": 5116,
+      "loss": 0.5962331960431081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962331960431081
+    },
+    {
+      "train_step": 5117,
+      "loss": 0.7392124080792035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7392124080792035
+    },
+    {
+      "train_step": 5118,
+      "loss": 0.5075928287702888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5075928287702888
+    },
+    {
+      "train_step": 5119,
+      "loss": 0.44851889875739825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44851889875739825
+    },
+    {
+      "train_step": 5120,
+      "loss": 0.5346669139041518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5346669139041518
+    },
+    {
+      "train_step": 5121,
+      "loss": 0.5764693290066665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5764693290066665
+    },
+    {
+      "train_step": 5122,
+      "loss": 0.5266179226220146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5266179226220146
+    },
+    {
+      "train_step": 5123,
+      "loss": 0.31408460046620684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31408460046620684
+    },
+    {
+      "train_step": 5124,
+      "loss": 0.6785859570818036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6785859570818036
+    },
+    {
+      "train_step": 5125,
+      "loss": 0.5731498397167453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5731498397167453
+    },
+    {
+      "train_step": 5126,
+      "loss": 0.6566484807877734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6566484807877734
+    },
+    {
+      "train_step": 5127,
+      "loss": 0.5145870817080298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5145870817080298
+    },
+    {
+      "train_step": 5128,
+      "loss": 0.46146982735600545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46146982735600545
+    },
+    {
+      "train_step": 5129,
+      "loss": 0.6340223456377616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6340223456377616
+    },
+    {
+      "train_step": 5130,
+      "loss": 0.33219040050355836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33219040050355836
+    },
+    {
+      "train_step": 5131,
+      "loss": 0.6361241232841186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6361241232841186
+    },
+    {
+      "train_step": 5132,
+      "loss": 0.62599330619974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.62599330619974
+    },
+    {
+      "train_step": 5133,
+      "loss": 0.47762149867526293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47762149867526293
+    },
+    {
+      "train_step": 5134,
+      "loss": 0.6838241911586781,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6838241911586781
+    },
+    {
+      "train_step": 5135,
+      "loss": 0.6621389043785595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6621389043785595
+    },
+    {
+      "train_step": 5136,
+      "loss": 0.7716082922366087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7716082922366087
+    },
+    {
+      "train_step": 5137,
+      "loss": 0.5846507492489664,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5846507492489664
+    },
+    {
+      "train_step": 5138,
+      "loss": 0.5327019128235998,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5327019128235998
+    },
+    {
+      "train_step": 5139,
+      "loss": 0.3694409397404727,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3694409397404727
+    },
+    {
+      "train_step": 5140,
+      "loss": 0.4870368330740797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4870368330740797
+    },
+    {
+      "train_step": 5141,
+      "loss": 0.6321771834279144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6321771834279144
+    },
+    {
+      "train_step": 5142,
+      "loss": 0.6475640159311955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6475640159311955
+    },
+    {
+      "train_step": 5143,
+      "loss": 0.6927551478689006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6927551478689006
+    },
+    {
+      "train_step": 5144,
+      "loss": 0.704020463723975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.704020463723975
+    },
+    {
+      "train_step": 5145,
+      "loss": 0.5844714366384033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5844714366384033
+    },
+    {
+      "train_step": 5146,
+      "loss": 0.4285679795984579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4285679795984579
+    },
+    {
+      "train_step": 5147,
+      "loss": 0.6781115638992604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6781115638992604
+    },
+    {
+      "train_step": 5148,
+      "loss": 0.5955534491860799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5955534491860799
+    },
+    {
+      "train_step": 5149,
+      "loss": 0.5380501360498258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5380501360498258
+    },
+    {
+      "train_step": 5150,
+      "loss": 0.5159904514426484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5159904514426484
+    },
+    {
+      "train_step": 5151,
+      "loss": 0.7208768374676362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7208768374676362
+    },
+    {
+      "train_step": 5152,
+      "loss": 0.579282780000169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.579282780000169
+    },
+    {
+      "train_step": 5153,
+      "loss": 0.6856962375323046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6856962375323046
+    },
+    {
+      "train_step": 5154,
+      "loss": 0.5013683174625193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5013683174625193
+    },
+    {
+      "train_step": 5155,
+      "loss": 0.6804666760517166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6804666760517166
+    },
+    {
+      "train_step": 5156,
+      "loss": 0.40876717749851377,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40876717749851377
+    },
+    {
+      "train_step": 5157,
+      "loss": 0.47116438611451344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47116438611451344
+    },
+    {
+      "train_step": 5158,
+      "loss": 0.6706424321805874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6706424321805874
+    },
+    {
+      "train_step": 5159,
+      "loss": 0.4963936236624695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4963936236624695
+    },
+    {
+      "train_step": 5160,
+      "loss": 0.6306356532941589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6306356532941589
+    },
+    {
+      "train_step": 5161,
+      "loss": 0.6558097492632073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6558097492632073
+    },
+    {
+      "train_step": 5162,
+      "loss": 0.45863942605505087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45863942605505087
+    },
+    {
+      "train_step": 5163,
+      "loss": 0.8169317395871294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8169317395871294
+    },
+    {
+      "train_step": 5164,
+      "loss": 0.4645649006789666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4645649006789666
+    },
+    {
+      "train_step": 5165,
+      "loss": 0.4779141515498284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4779141515498284
+    },
+    {
+      "train_step": 5166,
+      "loss": 0.6218980714856018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6218980714856018
+    },
+    {
+      "train_step": 5167,
+      "loss": 0.5419003925027065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5419003925027065
+    },
+    {
+      "train_step": 5168,
+      "loss": 0.6399779525262534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6399779525262534
+    },
+    {
+      "train_step": 5169,
+      "loss": 0.46381140753121664,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46381140753121664
+    },
+    {
+      "train_step": 5170,
+      "loss": 0.5127883857787165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5127883857787165
+    },
+    {
+      "train_step": 5171,
+      "loss": 0.6287197845596352,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6287197845596352
+    },
+    {
+      "train_step": 5172,
+      "loss": 0.5156540498395256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5156540498395256
+    },
+    {
+      "train_step": 5173,
+      "loss": 0.5556213735664158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5556213735664158
+    },
+    {
+      "train_step": 5174,
+      "loss": 0.5409799503266521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5409799503266521
+    },
+    {
+      "train_step": 5175,
+      "loss": 0.5945377736034799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5945377736034799
+    },
+    {
+      "train_step": 5176,
+      "loss": 0.6971493015743597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6971493015743597
+    },
+    {
+      "train_step": 5177,
+      "loss": 0.6060666995230466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6060666995230466
+    },
+    {
+      "train_step": 5178,
+      "loss": 0.5393279885094204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5393279885094204
+    },
+    {
+      "train_step": 5179,
+      "loss": 0.47695433462427644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47695433462427644
+    },
+    {
+      "train_step": 5180,
+      "loss": 0.6803886032047328,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6803886032047328
+    },
+    {
+      "train_step": 5181,
+      "loss": 0.719638238178892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.719638238178892
+    },
+    {
+      "train_step": 5182,
+      "loss": 0.743989665078808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.743989665078808
+    },
+    {
+      "train_step": 5183,
+      "loss": 0.5548294617928018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5548294617928018
+    },
+    {
+      "train_step": 5184,
+      "loss": 0.46566179069443087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46566179069443087
+    },
+    {
+      "train_step": 5185,
+      "loss": 0.3641631008421016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3641631008421016
+    },
+    {
+      "train_step": 5186,
+      "loss": 0.7693433146863518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7693433146863518
+    },
+    {
+      "train_step": 5187,
+      "loss": 0.5823171922896953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5823171922896953
+    },
+    {
+      "train_step": 5188,
+      "loss": 0.44453576366309233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44453576366309233
+    },
+    {
+      "train_step": 5189,
+      "loss": 0.6253439881808038,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6253439881808038
+    },
+    {
+      "train_step": 5190,
+      "loss": 0.4894751260962535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4894751260962535
+    },
+    {
+      "train_step": 5191,
+      "loss": 0.5965623614071421,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5965623614071421
+    },
+    {
+      "train_step": 5192,
+      "loss": 0.5915686411735397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5915686411735397
+    },
+    {
+      "train_step": 5193,
+      "loss": 0.5413266802635541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5413266802635541
+    },
+    {
+      "train_step": 5194,
+      "loss": 0.6444296461435074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6444296461435074
+    },
+    {
+      "train_step": 5195,
+      "loss": 0.5445780838564689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5445780838564689
+    },
+    {
+      "train_step": 5196,
+      "loss": 0.7022002343839715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7022002343839715
+    },
+    {
+      "train_step": 5197,
+      "loss": 0.4810226238788926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4810226238788926
+    },
+    {
+      "train_step": 5198,
+      "loss": 0.49546821828157683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49546821828157683
+    },
+    {
+      "train_step": 5199,
+      "loss": 0.5834004594225689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5834004594225689
+    },
+    {
+      "train_step": 5200,
+      "loss": 0.7317619307633825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7317619307633825
+    },
+    {
+      "train_step": 5201,
+      "loss": 0.7617224182830508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7617224182830508
+    },
+    {
+      "train_step": 5202,
+      "loss": 0.5658571954359126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5658571954359126
+    },
+    {
+      "train_step": 5203,
+      "loss": 0.48423425132932923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48423425132932923
+    },
+    {
+      "train_step": 5204,
+      "loss": 0.5852873081692708,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5852873081692708
+    },
+    {
+      "train_step": 5205,
+      "loss": 0.45617592288214376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45617592288214376
+    },
+    {
+      "train_step": 5206,
+      "loss": 0.49909019436089697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49909019436089697
+    },
+    {
+      "train_step": 5207,
+      "loss": 0.49308611007776265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49308611007776265
+    },
+    {
+      "train_step": 5208,
+      "loss": 0.7043406919776293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7043406919776293
+    },
+    {
+      "train_step": 5209,
+      "loss": 0.4875539492329111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4875539492329111
+    },
+    {
+      "train_step": 5210,
+      "loss": 0.4345226090690241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4345226090690241
+    },
+    {
+      "train_step": 5211,
+      "loss": 0.4072826064070675,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4072826064070675
+    },
+    {
+      "train_step": 5212,
+      "loss": 0.4468550002257882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4468550002257882
+    },
+    {
+      "train_step": 5213,
+      "loss": 0.44295467061270793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44295467061270793
+    },
+    {
+      "train_step": 5214,
+      "loss": 0.5545368995100878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5545368995100878
+    },
+    {
+      "train_step": 5215,
+      "loss": 0.5125779608554073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5125779608554073
+    },
+    {
+      "train_step": 5216,
+      "loss": 0.555977564931383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.555977564931383
+    },
+    {
+      "train_step": 5217,
+      "loss": 0.6994922934089833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6994922934089833
+    },
+    {
+      "train_step": 5218,
+      "loss": 0.5867817257400956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5867817257400956
+    },
+    {
+      "train_step": 5219,
+      "loss": 0.5472672889110308,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5472672889110308
+    },
+    {
+      "train_step": 5220,
+      "loss": 0.542058441697149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.542058441697149
+    },
+    {
+      "train_step": 5221,
+      "loss": 0.5299461011403953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5299461011403953
+    },
+    {
+      "train_step": 5222,
+      "loss": 0.5226732033622554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5226732033622554
+    },
+    {
+      "train_step": 5223,
+      "loss": 0.6258802905172915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6258802905172915
+    },
+    {
+      "train_step": 5224,
+      "loss": 0.6905944294477641,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6905944294477641
+    },
+    {
+      "train_step": 5225,
+      "loss": 0.3675035434202874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3675035434202874
+    },
+    {
+      "train_step": 5226,
+      "loss": 0.5474305311366778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5474305311366778
+    },
+    {
+      "train_step": 5227,
+      "loss": 0.6809088427285328,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6809088427285328
+    },
+    {
+      "train_step": 5228,
+      "loss": 0.4930082352655042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4930082352655042
+    },
+    {
+      "train_step": 5229,
+      "loss": 0.7541041163264122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7541041163264122
+    },
+    {
+      "train_step": 5230,
+      "loss": 0.5133110443497304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5133110443497304
+    },
+    {
+      "train_step": 5231,
+      "loss": 0.6164851374822411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6164851374822411
+    },
+    {
+      "train_step": 5232,
+      "loss": 0.6252001226436343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6252001226436343
+    },
+    {
+      "train_step": 5233,
+      "loss": 0.598355121801603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.598355121801603
+    },
+    {
+      "train_step": 5234,
+      "loss": 0.6996899448971384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6996899448971384
+    },
+    {
+      "train_step": 5235,
+      "loss": 0.4686175816245991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4686175816245991
+    },
+    {
+      "train_step": 5236,
+      "loss": 0.5582745358384713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5582745358384713
+    },
+    {
+      "train_step": 5237,
+      "loss": 0.5465027884404796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5465027884404796
+    },
+    {
+      "train_step": 5238,
+      "loss": 0.47707477604604187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47707477604604187
+    },
+    {
+      "train_step": 5239,
+      "loss": 0.5365657582636081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5365657582636081
+    },
+    {
+      "train_step": 5240,
+      "loss": 0.8258206556407882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8258206556407882
+    },
+    {
+      "train_step": 5241,
+      "loss": 0.5896740693814085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5896740693814085
+    },
+    {
+      "train_step": 5242,
+      "loss": 0.5735430316267323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5735430316267323
+    },
+    {
+      "train_step": 5243,
+      "loss": 0.7639548372234475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7639548372234475
+    },
+    {
+      "train_step": 5244,
+      "loss": 0.5869209110707065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5869209110707065
+    },
+    {
+      "train_step": 5245,
+      "loss": 0.5913466100163894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5913466100163894
+    },
+    {
+      "train_step": 5246,
+      "loss": 0.38401242529018503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38401242529018503
+    },
+    {
+      "train_step": 5247,
+      "loss": 0.5572985180242765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5572985180242765
+    },
+    {
+      "train_step": 5248,
+      "loss": 0.4772591898326216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4772591898326216
+    },
+    {
+      "train_step": 5249,
+      "loss": 0.5426345999157531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5426345999157531
+    },
+    {
+      "train_step": 5250,
+      "loss": 0.6117564190113254,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6117564190113254
+    },
+    {
+      "train_step": 5251,
+      "loss": 0.5432348804987169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5432348804987169
+    },
+    {
+      "train_step": 5252,
+      "loss": 0.5783340146541774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5783340146541774
+    },
+    {
+      "train_step": 5253,
+      "loss": 0.5681428678550896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5681428678550896
+    },
+    {
+      "train_step": 5254,
+      "loss": 0.5716987923213631,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5716987923213631
+    },
+    {
+      "train_step": 5255,
+      "loss": 0.4878020202831909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4878020202831909
+    },
+    {
+      "train_step": 5256,
+      "loss": 0.5677744908815694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5677744908815694
+    },
+    {
+      "train_step": 5257,
+      "loss": 0.43123278588827124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43123278588827124
+    },
+    {
+      "train_step": 5258,
+      "loss": 0.6864793857707217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6864793857707217
+    },
+    {
+      "train_step": 5259,
+      "loss": 0.38918460182530534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38918460182530534
+    },
+    {
+      "train_step": 5260,
+      "loss": 0.5048696661722524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5048696661722524
+    },
+    {
+      "train_step": 5261,
+      "loss": 0.48695645183331604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48695645183331604
+    },
+    {
+      "train_step": 5262,
+      "loss": 0.3987281529185119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3987281529185119
+    },
+    {
+      "train_step": 5263,
+      "loss": 0.47757627913878165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47757627913878165
+    },
+    {
+      "train_step": 5264,
+      "loss": 0.5009004693718343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5009004693718343
+    },
+    {
+      "train_step": 5265,
+      "loss": 0.6527560200656058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6527560200656058
+    },
+    {
+      "train_step": 5266,
+      "loss": 0.5654702505647372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5654702505647372
+    },
+    {
+      "train_step": 5267,
+      "loss": 0.5772918440810139,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5772918440810139
+    },
+    {
+      "train_step": 5268,
+      "loss": 0.5837412708176873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5837412708176873
+    },
+    {
+      "train_step": 5269,
+      "loss": 0.455687258866697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.455687258866697
+    },
+    {
+      "train_step": 5270,
+      "loss": 0.6079464939684169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6079464939684169
+    },
+    {
+      "train_step": 5271,
+      "loss": 0.7173938269895086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7173938269895086
+    },
+    {
+      "train_step": 5272,
+      "loss": 0.4665894552225049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4665894552225049
+    },
+    {
+      "train_step": 5273,
+      "loss": 0.5851906207449954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5851906207449954
+    },
+    {
+      "train_step": 5274,
+      "loss": 0.35555361512847794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35555361512847794
+    },
+    {
+      "train_step": 5275,
+      "loss": 0.5789350775393713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5789350775393713
+    },
+    {
+      "train_step": 5276,
+      "loss": 0.36730815862953153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36730815862953153
+    },
+    {
+      "train_step": 5277,
+      "loss": 0.4987789623186372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4987789623186372
+    },
+    {
+      "train_step": 5278,
+      "loss": 0.5729982672127925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5729982672127925
+    },
+    {
+      "train_step": 5279,
+      "loss": 0.5669666963463365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5669666963463365
+    },
+    {
+      "train_step": 5280,
+      "loss": 0.44242802344359844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44242802344359844
+    },
+    {
+      "train_step": 5281,
+      "loss": 0.5738190793233319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5738190793233319
+    },
+    {
+      "train_step": 5282,
+      "loss": 0.5495889076850655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5495889076850655
+    },
+    {
+      "train_step": 5283,
+      "loss": 0.5112881190784724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5112881190784724
+    },
+    {
+      "train_step": 5284,
+      "loss": 0.5350564935040651,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5350564935040651
+    },
+    {
+      "train_step": 5285,
+      "loss": 0.8781027487533536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8781027487533536
+    },
+    {
+      "train_step": 5286,
+      "loss": 0.5919634320370202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5919634320370202
+    },
+    {
+      "train_step": 5287,
+      "loss": 0.4966458513490064,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4966458513490064
+    },
+    {
+      "train_step": 5288,
+      "loss": 0.4477513180507375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4477513180507375
+    },
+    {
+      "train_step": 5289,
+      "loss": 0.5288461331447973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5288461331447973
+    },
+    {
+      "train_step": 5290,
+      "loss": 0.5038152815784923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5038152815784923
+    },
+    {
+      "train_step": 5291,
+      "loss": 0.5178074579614237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5178074579614237
+    },
+    {
+      "train_step": 5292,
+      "loss": 0.6964691733865517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6964691733865517
+    },
+    {
+      "train_step": 5293,
+      "loss": 0.6856775641411174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6856775641411174
+    },
+    {
+      "train_step": 5294,
+      "loss": 0.558564832524754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.558564832524754
+    },
+    {
+      "train_step": 5295,
+      "loss": 0.5162648693378294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5162648693378294
+    },
+    {
+      "train_step": 5296,
+      "loss": 0.6142535642663178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6142535642663178
+    },
+    {
+      "train_step": 5297,
+      "loss": 0.4707944713433995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4707944713433995
+    },
+    {
+      "train_step": 5298,
+      "loss": 0.4872382592464741,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4872382592464741
+    },
+    {
+      "train_step": 5299,
+      "loss": 0.5469695401619342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5469695401619342
+    },
+    {
+      "train_step": 5300,
+      "loss": 0.7157034126281235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7157034126281235
+    },
+    {
+      "train_step": 5301,
+      "loss": 0.5266367893989298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5266367893989298
+    },
+    {
+      "train_step": 5302,
+      "loss": 0.47640157095931224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47640157095931224
+    },
+    {
+      "train_step": 5303,
+      "loss": 0.435375686324959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.435375686324959
+    },
+    {
+      "train_step": 5304,
+      "loss": 0.5867455167815916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5867455167815916
+    },
+    {
+      "train_step": 5305,
+      "loss": 0.44718426610147055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44718426610147055
+    },
+    {
+      "train_step": 5306,
+      "loss": 0.5437065785764399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5437065785764399
+    },
+    {
+      "train_step": 5307,
+      "loss": 0.7783888617788428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7783888617788428
+    },
+    {
+      "train_step": 5308,
+      "loss": 0.5480048747881063,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5480048747881063
+    },
+    {
+      "train_step": 5309,
+      "loss": 0.4719156257356866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4719156257356866
+    },
+    {
+      "train_step": 5310,
+      "loss": 0.5459129206518303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5459129206518303
+    },
+    {
+      "train_step": 5311,
+      "loss": 0.6729600691659228,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6729600691659228
+    },
+    {
+      "train_step": 5312,
+      "loss": 0.5190847303233027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5190847303233027
+    },
+    {
+      "train_step": 5313,
+      "loss": 0.46379758877126576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46379758877126576
+    },
+    {
+      "train_step": 5314,
+      "loss": 0.6747652460155794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6747652460155794
+    },
+    {
+      "train_step": 5315,
+      "loss": 0.7502315954603513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7502315954603513
+    },
+    {
+      "train_step": 5316,
+      "loss": 0.5665006140646851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5665006140646851
+    },
+    {
+      "train_step": 5317,
+      "loss": 0.5884539136473022,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5884539136473022
+    },
+    {
+      "train_step": 5318,
+      "loss": 0.6710619577987982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6710619577987982
+    },
+    {
+      "train_step": 5319,
+      "loss": 0.6089008454823114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6089008454823114
+    },
+    {
+      "train_step": 5320,
+      "loss": 0.4981918167164355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4981918167164355
+    },
+    {
+      "train_step": 5321,
+      "loss": 0.6448618133601821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6448618133601821
+    },
+    {
+      "train_step": 5322,
+      "loss": 0.7132242968637599,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7132242968637599
+    },
+    {
+      "train_step": 5323,
+      "loss": 0.6260195292756539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6260195292756539
+    },
+    {
+      "train_step": 5324,
+      "loss": 0.6180004394909769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6180004394909769
+    },
+    {
+      "train_step": 5325,
+      "loss": 0.4896489041410465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4896489041410465
+    },
+    {
+      "train_step": 5326,
+      "loss": 0.470852869105086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.470852869105086
+    },
+    {
+      "train_step": 5327,
+      "loss": 0.658749580585147,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.658749580585147
+    },
+    {
+      "train_step": 5328,
+      "loss": 0.6543754308214101,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6543754308214101
+    },
+    {
+      "train_step": 5329,
+      "loss": 0.560123243637374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.560123243637374
+    },
+    {
+      "train_step": 5330,
+      "loss": 0.5051487340454545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5051487340454545
+    },
+    {
+      "train_step": 5331,
+      "loss": 0.367934662722497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.367934662722497
+    },
+    {
+      "train_step": 5332,
+      "loss": 0.5585739157412255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5585739157412255
+    },
+    {
+      "train_step": 5333,
+      "loss": 0.6923142843401114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6923142843401114
+    },
+    {
+      "train_step": 5334,
+      "loss": 0.527728410109579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.527728410109579
+    },
+    {
+      "train_step": 5335,
+      "loss": 0.6867552913023027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6867552913023027
+    },
+    {
+      "train_step": 5336,
+      "loss": 0.288110545683233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.288110545683233
+    },
+    {
+      "train_step": 5337,
+      "loss": 0.7629469411434143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7629469411434143
+    },
+    {
+      "train_step": 5338,
+      "loss": 0.6263440766196696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6263440766196696
+    },
+    {
+      "train_step": 5339,
+      "loss": 0.6522746251414115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6522746251414115
+    },
+    {
+      "train_step": 5340,
+      "loss": 0.6275241833736648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6275241833736648
+    },
+    {
+      "train_step": 5341,
+      "loss": 0.4763924766111265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4763924766111265
+    },
+    {
+      "train_step": 5342,
+      "loss": 0.7023655413664627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7023655413664627
+    },
+    {
+      "train_step": 5343,
+      "loss": 0.4913505456913833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4913505456913833
+    },
+    {
+      "train_step": 5344,
+      "loss": 0.6335330697338236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6335330697338236
+    },
+    {
+      "train_step": 5345,
+      "loss": 0.36518153369830797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36518153369830797
+    },
+    {
+      "train_step": 5346,
+      "loss": 0.7444348507783849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7444348507783849
+    },
+    {
+      "train_step": 5347,
+      "loss": 0.5864648774445509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5864648774445509
+    },
+    {
+      "train_step": 5348,
+      "loss": 0.5193502242587077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5193502242587077
+    },
+    {
+      "train_step": 5349,
+      "loss": 0.6336125750479624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6336125750479624
+    },
+    {
+      "train_step": 5350,
+      "loss": 0.7130704332641378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7130704332641378
+    },
+    {
+      "train_step": 5351,
+      "loss": 0.7724300608943991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7724300608943991
+    },
+    {
+      "train_step": 5352,
+      "loss": 0.45204795071381804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45204795071381804
+    },
+    {
+      "train_step": 5353,
+      "loss": 0.7255064212214304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7255064212214304
+    },
+    {
+      "train_step": 5354,
+      "loss": 0.7757154518510164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7757154518510164
+    },
+    {
+      "train_step": 5355,
+      "loss": 0.452385264365129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.452385264365129
+    },
+    {
+      "train_step": 5356,
+      "loss": 0.6869796703989641,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6869796703989641
+    },
+    {
+      "train_step": 5357,
+      "loss": 0.5559667872494668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5559667872494668
+    },
+    {
+      "train_step": 5358,
+      "loss": 0.4608546629319669,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4608546629319669
+    },
+    {
+      "train_step": 5359,
+      "loss": 0.6548065978024615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6548065978024615
+    },
+    {
+      "train_step": 5360,
+      "loss": 0.6046946498283435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6046946498283435
+    },
+    {
+      "train_step": 5361,
+      "loss": 0.5622975296670383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5622975296670383
+    },
+    {
+      "train_step": 5362,
+      "loss": 0.6389868706774862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6389868706774862
+    },
+    {
+      "train_step": 5363,
+      "loss": 0.45242372506962414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45242372506962414
+    },
+    {
+      "train_step": 5364,
+      "loss": 0.48878861303677545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48878861303677545
+    },
+    {
+      "train_step": 5365,
+      "loss": 0.29568682690620435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.29568682690620435
+    },
+    {
+      "train_step": 5366,
+      "loss": 0.5546509966964902,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5546509966964902
+    },
+    {
+      "train_step": 5367,
+      "loss": 0.6439658895531511,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6439658895531511
+    },
+    {
+      "train_step": 5368,
+      "loss": 0.5287560403046141,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5287560403046141
+    },
+    {
+      "train_step": 5369,
+      "loss": 0.49757559803066825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49757559803066825
+    },
+    {
+      "train_step": 5370,
+      "loss": 0.5573748663080206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5573748663080206
+    },
+    {
+      "train_step": 5371,
+      "loss": 0.603830119274421,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.603830119274421
+    },
+    {
+      "train_step": 5372,
+      "loss": 0.43140978110572953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43140978110572953
+    },
+    {
+      "train_step": 5373,
+      "loss": 0.5130015061783217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5130015061783217
+    },
+    {
+      "train_step": 5374,
+      "loss": 0.6929339061483586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6929339061483586
+    },
+    {
+      "train_step": 5375,
+      "loss": 0.7630255658216326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7630255658216326
+    },
+    {
+      "train_step": 5376,
+      "loss": 0.4637254932610912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4637254932610912
+    },
+    {
+      "train_step": 5377,
+      "loss": 0.6700725206277739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6700725206277739
+    },
+    {
+      "train_step": 5378,
+      "loss": 0.7695790827627131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7695790827627131
+    },
+    {
+      "train_step": 5379,
+      "loss": 0.6959818831548953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6959818831548953
+    },
+    {
+      "train_step": 5380,
+      "loss": 0.5409272763382422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5409272763382422
+    },
+    {
+      "train_step": 5381,
+      "loss": 0.5752875934523929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5752875934523929
+    },
+    {
+      "train_step": 5382,
+      "loss": 0.46873993300762523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46873993300762523
+    },
+    {
+      "train_step": 5383,
+      "loss": 0.5704348406613327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5704348406613327
+    },
+    {
+      "train_step": 5384,
+      "loss": 0.7989070628406588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7989070628406588
+    },
+    {
+      "train_step": 5385,
+      "loss": 0.582465529100225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.582465529100225
+    },
+    {
+      "train_step": 5386,
+      "loss": 0.43029120820573763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43029120820573763
+    },
+    {
+      "train_step": 5387,
+      "loss": 0.5955960338911478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5955960338911478
+    },
+    {
+      "train_step": 5388,
+      "loss": 0.4409445332590409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4409445332590409
+    },
+    {
+      "train_step": 5389,
+      "loss": 0.5693703957781819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5693703957781819
+    },
+    {
+      "train_step": 5390,
+      "loss": 0.6438189265793908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6438189265793908
+    },
+    {
+      "train_step": 5391,
+      "loss": 0.6844321991671442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6844321991671442
+    },
+    {
+      "train_step": 5392,
+      "loss": 0.43349434709332646,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43349434709332646
+    },
+    {
+      "train_step": 5393,
+      "loss": 0.42675317629074905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42675317629074905
+    },
+    {
+      "train_step": 5394,
+      "loss": 0.47407166304232273,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47407166304232273
+    },
+    {
+      "train_step": 5395,
+      "loss": 0.7894774508354896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7894774508354896
+    },
+    {
+      "train_step": 5396,
+      "loss": 0.5121296098798164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5121296098798164
+    },
+    {
+      "train_step": 5397,
+      "loss": 0.45006262460979185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45006262460979185
+    },
+    {
+      "train_step": 5398,
+      "loss": 0.351578130118411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.351578130118411
+    },
+    {
+      "train_step": 5399,
+      "loss": 0.589275630686747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.589275630686747
+    },
+    {
+      "train_step": 5400,
+      "loss": 0.6318491180066351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6318491180066351
+    },
+    {
+      "train_step": 5401,
+      "loss": 0.4419181764678938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4419181764678938
+    },
+    {
+      "train_step": 5402,
+      "loss": 0.2677039023941057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2677039023941057
+    },
+    {
+      "train_step": 5403,
+      "loss": 0.47507159712864533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47507159712864533
+    },
+    {
+      "train_step": 5404,
+      "loss": 0.7659514369260386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7659514369260386
+    },
+    {
+      "train_step": 5405,
+      "loss": 0.4620247804647361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4620247804647361
+    },
+    {
+      "train_step": 5406,
+      "loss": 0.7974103882600193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7974103882600193
+    },
+    {
+      "train_step": 5407,
+      "loss": 0.47532953164902153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47532953164902153
+    },
+    {
+      "train_step": 5408,
+      "loss": 0.47703373633413715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47703373633413715
+    },
+    {
+      "train_step": 5409,
+      "loss": 0.4777815884309103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4777815884309103
+    },
+    {
+      "train_step": 5410,
+      "loss": 0.5558120500784562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5558120500784562
+    },
+    {
+      "train_step": 5411,
+      "loss": 0.5779341329246016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5779341329246016
+    },
+    {
+      "train_step": 5412,
+      "loss": 0.3227187030761965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3227187030761965
+    },
+    {
+      "train_step": 5413,
+      "loss": 0.4167266687348493,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4167266687348493
+    },
+    {
+      "train_step": 5414,
+      "loss": 0.4307405397042603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4307405397042603
+    },
+    {
+      "train_step": 5415,
+      "loss": 0.4241844302150666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4241844302150666
+    },
+    {
+      "train_step": 5416,
+      "loss": 0.49272864328607724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49272864328607724
+    },
+    {
+      "train_step": 5417,
+      "loss": 0.5594561259827229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5594561259827229
+    },
+    {
+      "train_step": 5418,
+      "loss": 0.5519934442902686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5519934442902686
+    },
+    {
+      "train_step": 5419,
+      "loss": 0.42046561098804186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42046561098804186
+    },
+    {
+      "train_step": 5420,
+      "loss": 0.4380114068259723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4380114068259723
+    },
+    {
+      "train_step": 5421,
+      "loss": 0.5224209474649089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5224209474649089
+    },
+    {
+      "train_step": 5422,
+      "loss": 0.5921968759723799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5921968759723799
+    },
+    {
+      "train_step": 5423,
+      "loss": 0.5454734674874597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5454734674874597
+    },
+    {
+      "train_step": 5424,
+      "loss": 0.40245573880721763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40245573880721763
+    },
+    {
+      "train_step": 5425,
+      "loss": 0.6414862628303841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6414862628303841
+    },
+    {
+      "train_step": 5426,
+      "loss": 0.56472240400867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.56472240400867
+    },
+    {
+      "train_step": 5427,
+      "loss": 0.5960142810782607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5960142810782607
+    },
+    {
+      "train_step": 5428,
+      "loss": 0.40106013541953345,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40106013541953345
+    },
+    {
+      "train_step": 5429,
+      "loss": 0.493546051961351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.493546051961351
+    },
+    {
+      "train_step": 5430,
+      "loss": 0.29966963393838186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.29966963393838186
+    },
+    {
+      "train_step": 5431,
+      "loss": 0.7575043045975975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7575043045975975
+    },
+    {
+      "train_step": 5432,
+      "loss": 0.5073438741970729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5073438741970729
+    },
+    {
+      "train_step": 5433,
+      "loss": 0.437390949501387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.437390949501387
+    },
+    {
+      "train_step": 5434,
+      "loss": 0.5290472386060054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5290472386060054
+    },
+    {
+      "train_step": 5435,
+      "loss": 0.657125037002092,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.657125037002092
+    },
+    {
+      "train_step": 5436,
+      "loss": 0.6055139953866485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6055139953866485
+    },
+    {
+      "train_step": 5437,
+      "loss": 0.46786555504419625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46786555504419625
+    },
+    {
+      "train_step": 5438,
+      "loss": 0.4635239586110559,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4635239586110559
+    },
+    {
+      "train_step": 5439,
+      "loss": 0.7155146005530413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7155146005530413
+    },
+    {
+      "train_step": 5440,
+      "loss": 0.5655014875378801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5655014875378801
+    },
+    {
+      "train_step": 5441,
+      "loss": 0.605342174840425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.605342174840425
+    },
+    {
+      "train_step": 5442,
+      "loss": 0.4109906532267527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4109906532267527
+    },
+    {
+      "train_step": 5443,
+      "loss": 0.3291496847699033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3291496847699033
+    },
+    {
+      "train_step": 5444,
+      "loss": 0.4074768136290779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4074768136290779
+    },
+    {
+      "train_step": 5445,
+      "loss": 0.8637543331776617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8637543331776617
+    },
+    {
+      "train_step": 5446,
+      "loss": 0.5745933578437926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5745933578437926
+    },
+    {
+      "train_step": 5447,
+      "loss": 0.42532940080351667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42532940080351667
+    },
+    {
+      "train_step": 5448,
+      "loss": 0.5224726776064144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5224726776064144
+    },
+    {
+      "train_step": 5449,
+      "loss": 0.6868449828582658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6868449828582658
+    },
+    {
+      "train_step": 5450,
+      "loss": 0.4669332467296814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4669332467296814
+    },
+    {
+      "train_step": 5451,
+      "loss": 0.4283683250225617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4283683250225617
+    },
+    {
+      "train_step": 5452,
+      "loss": 0.5494358359834496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5494358359834496
+    },
+    {
+      "train_step": 5453,
+      "loss": 0.7537586036523068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7537586036523068
+    },
+    {
+      "train_step": 5454,
+      "loss": 0.46894721186796356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46894721186796356
+    },
+    {
+      "train_step": 5455,
+      "loss": 0.6154547359146353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6154547359146353
+    },
+    {
+      "train_step": 5456,
+      "loss": 0.5336455492134043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5336455492134043
+    },
+    {
+      "train_step": 5457,
+      "loss": 0.6146762934994805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6146762934994805
+    },
+    {
+      "train_step": 5458,
+      "loss": 0.5087826085579504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5087826085579504
+    },
+    {
+      "train_step": 5459,
+      "loss": 0.6335254747536018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6335254747536018
+    },
+    {
+      "train_step": 5460,
+      "loss": 0.4902302852953227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4902302852953227
+    },
+    {
+      "train_step": 5461,
+      "loss": 0.49186251595037184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49186251595037184
+    },
+    {
+      "train_step": 5462,
+      "loss": 0.5533326097861672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5533326097861672
+    },
+    {
+      "train_step": 5463,
+      "loss": 0.5965544614370819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5965544614370819
+    },
+    {
+      "train_step": 5464,
+      "loss": 0.6060743172070495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6060743172070495
+    },
+    {
+      "train_step": 5465,
+      "loss": 0.50100680824884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.50100680824884
+    },
+    {
+      "train_step": 5466,
+      "loss": 0.5174619600334418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5174619600334418
+    },
+    {
+      "train_step": 5467,
+      "loss": 0.4936153141595463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4936153141595463
+    },
+    {
+      "train_step": 5468,
+      "loss": 0.5023880386163433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5023880386163433
+    },
+    {
+      "train_step": 5469,
+      "loss": 0.6960200936847165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6960200936847165
+    },
+    {
+      "train_step": 5470,
+      "loss": 0.6358551229308157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6358551229308157
+    },
+    {
+      "train_step": 5471,
+      "loss": 0.4254724776093944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4254724776093944
+    },
+    {
+      "train_step": 5472,
+      "loss": 0.6002403702006397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6002403702006397
+    },
+    {
+      "train_step": 5473,
+      "loss": 0.6296731125218717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6296731125218717
+    },
+    {
+      "train_step": 5474,
+      "loss": 0.5334780952053146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5334780952053146
+    },
+    {
+      "train_step": 5475,
+      "loss": 0.5025504583270803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5025504583270803
+    },
+    {
+      "train_step": 5476,
+      "loss": 0.6427489442560873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6427489442560873
+    },
+    {
+      "train_step": 5477,
+      "loss": 0.5747183465378561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5747183465378561
+    },
+    {
+      "train_step": 5478,
+      "loss": 0.7607709385423189,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7607709385423189
+    },
+    {
+      "train_step": 5479,
+      "loss": 0.5229567119118167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5229567119118167
+    },
+    {
+      "train_step": 5480,
+      "loss": 0.5709218000862099,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5709218000862099
+    },
+    {
+      "train_step": 5481,
+      "loss": 0.5498636895965378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5498636895965378
+    },
+    {
+      "train_step": 5482,
+      "loss": 0.4349807620442156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4349807620442156
+    },
+    {
+      "train_step": 5483,
+      "loss": 0.6386702291988423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6386702291988423
+    },
+    {
+      "train_step": 5484,
+      "loss": 0.6676530237911675,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6676530237911675
+    },
+    {
+      "train_step": 5485,
+      "loss": 0.4658753728845741,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4658753728845741
+    },
+    {
+      "train_step": 5486,
+      "loss": 0.728766701610841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.728766701610841
+    },
+    {
+      "train_step": 5487,
+      "loss": 0.6868915953128962,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6868915953128962
+    },
+    {
+      "train_step": 5488,
+      "loss": 0.7285637179472428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7285637179472428
+    },
+    {
+      "train_step": 5489,
+      "loss": 0.5687898090800714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5687898090800714
+    },
+    {
+      "train_step": 5490,
+      "loss": 0.4121926579776018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4121926579776018
+    },
+    {
+      "train_step": 5491,
+      "loss": 0.5369731434650417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5369731434650417
+    },
+    {
+      "train_step": 5492,
+      "loss": 0.6871808625069873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6871808625069873
+    },
+    {
+      "train_step": 5493,
+      "loss": 0.5066990657459994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5066990657459994
+    },
+    {
+      "train_step": 5494,
+      "loss": 0.7494670920068728,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7494670920068728
+    },
+    {
+      "train_step": 5495,
+      "loss": 0.7222093218584833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7222093218584833
+    },
+    {
+      "train_step": 5496,
+      "loss": 0.8084914565954593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8084914565954593
+    },
+    {
+      "train_step": 5497,
+      "loss": 0.654275574560277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.654275574560277
+    },
+    {
+      "train_step": 5498,
+      "loss": 0.6047913287781628,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6047913287781628
+    },
+    {
+      "train_step": 5499,
+      "loss": 0.6296343407524202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6296343407524202
+    },
+    {
+      "train_step": 5500,
+      "loss": 0.44391966607506256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44391966607506256
+    },
+    {
+      "train_step": 5501,
+      "loss": 0.5349446543945684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5349446543945684
+    },
+    {
+      "train_step": 5502,
+      "loss": 0.4509118120221773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4509118120221773
+    },
+    {
+      "train_step": 5503,
+      "loss": 0.7115684437814799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7115684437814799
+    },
+    {
+      "train_step": 5504,
+      "loss": 0.6842594920456808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6842594920456808
+    },
+    {
+      "train_step": 5505,
+      "loss": 0.790333399269499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.790333399269499
+    },
+    {
+      "train_step": 5506,
+      "loss": 0.49294740813487115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49294740813487115
+    },
+    {
+      "train_step": 5507,
+      "loss": 0.46127406135933263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46127406135933263
+    },
+    {
+      "train_step": 5508,
+      "loss": 0.6649663042662476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6649663042662476
+    },
+    {
+      "train_step": 5509,
+      "loss": 0.6019149092550841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6019149092550841
+    },
+    {
+      "train_step": 5510,
+      "loss": 0.4111135437830987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4111135437830987
+    },
+    {
+      "train_step": 5511,
+      "loss": 0.6559226870399587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6559226870399587
+    },
+    {
+      "train_step": 5512,
+      "loss": 0.3474095638776605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3474095638776605
+    },
+    {
+      "train_step": 5513,
+      "loss": 0.5159162874283361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5159162874283361
+    },
+    {
+      "train_step": 5514,
+      "loss": 0.569636078331054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.569636078331054
+    },
+    {
+      "train_step": 5515,
+      "loss": 0.4695423176065475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4695423176065475
+    },
+    {
+      "train_step": 5516,
+      "loss": 0.7150008667277574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7150008667277574
+    },
+    {
+      "train_step": 5517,
+      "loss": 0.557589245465244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.557589245465244
+    },
+    {
+      "train_step": 5518,
+      "loss": 0.5675921713312237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5675921713312237
+    },
+    {
+      "train_step": 5519,
+      "loss": 0.47771295269077757,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47771295269077757
+    },
+    {
+      "train_step": 5520,
+      "loss": 0.5940297288904753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5940297288904753
+    },
+    {
+      "train_step": 5521,
+      "loss": 0.5329464957731147,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5329464957731147
+    },
+    {
+      "train_step": 5522,
+      "loss": 0.5725491333459396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5725491333459396
+    },
+    {
+      "train_step": 5523,
+      "loss": 0.7161780953979071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7161780953979071
+    },
+    {
+      "train_step": 5524,
+      "loss": 0.5919946194429629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5919946194429629
+    },
+    {
+      "train_step": 5525,
+      "loss": 0.572931866597417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.572931866597417
+    },
+    {
+      "train_step": 5526,
+      "loss": 0.3834488858909026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3834488858909026
+    },
+    {
+      "train_step": 5527,
+      "loss": 0.5724434624968618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5724434624968618
+    },
+    {
+      "train_step": 5528,
+      "loss": 0.5386932288322605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5386932288322605
+    },
+    {
+      "train_step": 5529,
+      "loss": 0.5816251163041084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5816251163041084
+    },
+    {
+      "train_step": 5530,
+      "loss": 0.6007212107893224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6007212107893224
+    },
+    {
+      "train_step": 5531,
+      "loss": 0.36551437031273415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36551437031273415
+    },
+    {
+      "train_step": 5532,
+      "loss": 0.5803560911079704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5803560911079704
+    },
+    {
+      "train_step": 5533,
+      "loss": 0.5277195310556788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5277195310556788
+    },
+    {
+      "train_step": 5534,
+      "loss": 0.7002381191075187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7002381191075187
+    },
+    {
+      "train_step": 5535,
+      "loss": 0.5422440585585588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5422440585585588
+    },
+    {
+      "train_step": 5536,
+      "loss": 0.5387540768583721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5387540768583721
+    },
+    {
+      "train_step": 5537,
+      "loss": 0.364368577013671,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.364368577013671
+    },
+    {
+      "train_step": 5538,
+      "loss": 0.4696703858384406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4696703858384406
+    },
+    {
+      "train_step": 5539,
+      "loss": 0.7044650923375528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7044650923375528
+    },
+    {
+      "train_step": 5540,
+      "loss": 0.4525352200316185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4525352200316185
+    },
+    {
+      "train_step": 5541,
+      "loss": 0.39266749327971406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39266749327971406
+    },
+    {
+      "train_step": 5542,
+      "loss": 0.6050983973946021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6050983973946021
+    },
+    {
+      "train_step": 5543,
+      "loss": 0.5934518574959089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5934518574959089
+    },
+    {
+      "train_step": 5544,
+      "loss": 0.31581435890213705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31581435890213705
+    },
+    {
+      "train_step": 5545,
+      "loss": 0.5139086408959194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5139086408959194
+    },
+    {
+      "train_step": 5546,
+      "loss": 0.37309883637677893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37309883637677893
+    },
+    {
+      "train_step": 5547,
+      "loss": 0.6199803644831515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6199803644831515
+    },
+    {
+      "train_step": 5548,
+      "loss": 0.6617004100489509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6617004100489509
+    },
+    {
+      "train_step": 5549,
+      "loss": 0.4871648044345487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4871648044345487
+    },
+    {
+      "train_step": 5550,
+      "loss": 0.6486888890546856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6486888890546856
+    },
+    {
+      "train_step": 5551,
+      "loss": 0.5726325989584806,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5726325989584806
+    },
+    {
+      "train_step": 5552,
+      "loss": 0.6109746762039369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6109746762039369
+    },
+    {
+      "train_step": 5553,
+      "loss": 0.5357076546735644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5357076546735644
+    },
+    {
+      "train_step": 5554,
+      "loss": 0.47897646083948575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47897646083948575
+    },
+    {
+      "train_step": 5555,
+      "loss": 0.4584544014637548,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4584544014637548
+    },
+    {
+      "train_step": 5556,
+      "loss": 0.48772143734670015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48772143734670015
+    },
+    {
+      "train_step": 5557,
+      "loss": 0.6025175383017322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6025175383017322
+    },
+    {
+      "train_step": 5558,
+      "loss": 0.4904117896339677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4904117896339677
+    },
+    {
+      "train_step": 5559,
+      "loss": 0.6037806255862411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6037806255862411
+    },
+    {
+      "train_step": 5560,
+      "loss": 0.34948141248800046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34948141248800046
+    },
+    {
+      "train_step": 5561,
+      "loss": 0.48967322948746866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48967322948746866
+    },
+    {
+      "train_step": 5562,
+      "loss": 0.6086384123082095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6086384123082095
+    },
+    {
+      "train_step": 5563,
+      "loss": 0.5098446318979757,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5098446318979757
+    },
+    {
+      "train_step": 5564,
+      "loss": 0.7085089567356112,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7085089567356112
+    },
+    {
+      "train_step": 5565,
+      "loss": 0.6811803254616163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6811803254616163
+    },
+    {
+      "train_step": 5566,
+      "loss": 0.7999332770601929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7999332770601929
+    },
+    {
+      "train_step": 5567,
+      "loss": 0.6909967463340612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6909967463340612
+    },
+    {
+      "train_step": 5568,
+      "loss": 0.4976647428706458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4976647428706458
+    },
+    {
+      "train_step": 5569,
+      "loss": 0.6981983076809933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6981983076809933
+    },
+    {
+      "train_step": 5570,
+      "loss": 0.6896953426431461,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6896953426431461
+    },
+    {
+      "train_step": 5571,
+      "loss": 0.4753671388675059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4753671388675059
+    },
+    {
+      "train_step": 5572,
+      "loss": 0.6113912272104853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6113912272104853
+    },
+    {
+      "train_step": 5573,
+      "loss": 0.7270150343472583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7270150343472583
+    },
+    {
+      "train_step": 5574,
+      "loss": 0.6997442457001084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6997442457001084
+    },
+    {
+      "train_step": 5575,
+      "loss": 0.6832049412223451,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6832049412223451
+    },
+    {
+      "train_step": 5576,
+      "loss": 0.5185180402820478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5185180402820478
+    },
+    {
+      "train_step": 5577,
+      "loss": 0.4081818857409452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4081818857409452
+    },
+    {
+      "train_step": 5578,
+      "loss": 0.5656448015218907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5656448015218907
+    },
+    {
+      "train_step": 5579,
+      "loss": 0.5518420299435967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5518420299435967
+    },
+    {
+      "train_step": 5580,
+      "loss": 0.4976892027548853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4976892027548853
+    },
+    {
+      "train_step": 5581,
+      "loss": 0.6994329451330745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6994329451330745
+    },
+    {
+      "train_step": 5582,
+      "loss": 0.6312859104698912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6312859104698912
+    },
+    {
+      "train_step": 5583,
+      "loss": 0.5583879702758855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5583879702758855
+    },
+    {
+      "train_step": 5584,
+      "loss": 0.6214213265045705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6214213265045705
+    },
+    {
+      "train_step": 5585,
+      "loss": 0.571377377303522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.571377377303522
+    },
+    {
+      "train_step": 5586,
+      "loss": 0.3536686950613737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3536686950613737
+    },
+    {
+      "train_step": 5587,
+      "loss": 0.3422283245149494,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3422283245149494
+    },
+    {
+      "train_step": 5588,
+      "loss": 0.6900609407702406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6900609407702406
+    },
+    {
+      "train_step": 5589,
+      "loss": 0.3793927597721521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3793927597721521
+    },
+    {
+      "train_step": 5590,
+      "loss": 0.5786645112112688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5786645112112688
+    },
+    {
+      "train_step": 5591,
+      "loss": 0.6577418184549164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6577418184549164
+    },
+    {
+      "train_step": 5592,
+      "loss": 0.49331141626022434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49331141626022434
+    },
+    {
+      "train_step": 5593,
+      "loss": 0.6314599852462942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6314599852462942
+    },
+    {
+      "train_step": 5594,
+      "loss": 0.3305546330784408,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3305546330784408
+    },
+    {
+      "train_step": 5595,
+      "loss": 0.44188418065195206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44188418065195206
+    },
+    {
+      "train_step": 5596,
+      "loss": 0.6757147803023326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6757147803023326
+    },
+    {
+      "train_step": 5597,
+      "loss": 0.37353405066090556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37353405066090556
+    },
+    {
+      "train_step": 5598,
+      "loss": 0.7499742047115369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7499742047115369
+    },
+    {
+      "train_step": 5599,
+      "loss": 0.5757635762229992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5757635762229992
+    },
+    {
+      "train_step": 5600,
+      "loss": 0.7901542759901854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7901542759901854
+    },
+    {
+      "train_step": 5601,
+      "loss": 0.5311039345360868,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5311039345360868
+    },
+    {
+      "train_step": 5602,
+      "loss": 0.5120346502113113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5120346502113113
+    },
+    {
+      "train_step": 5603,
+      "loss": 0.5923852639823923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5923852639823923
+    },
+    {
+      "train_step": 5604,
+      "loss": 0.48131691303042473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48131691303042473
+    },
+    {
+      "train_step": 5605,
+      "loss": 0.4839509071033908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4839509071033908
+    },
+    {
+      "train_step": 5606,
+      "loss": 0.4427503517006143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4427503517006143
+    },
+    {
+      "train_step": 5607,
+      "loss": 0.6186034443477593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6186034443477593
+    },
+    {
+      "train_step": 5608,
+      "loss": 0.46886130962961414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46886130962961414
+    },
+    {
+      "train_step": 5609,
+      "loss": 0.6001858574612771,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6001858574612771
+    },
+    {
+      "train_step": 5610,
+      "loss": 0.579182214935165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.579182214935165
+    },
+    {
+      "train_step": 5611,
+      "loss": 0.5505738212568422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5505738212568422
+    },
+    {
+      "train_step": 5612,
+      "loss": 0.4335102924652563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4335102924652563
+    },
+    {
+      "train_step": 5613,
+      "loss": 0.4784079890730053,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4784079890730053
+    },
+    {
+      "train_step": 5614,
+      "loss": 0.4396856261685359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4396856261685359
+    },
+    {
+      "train_step": 5615,
+      "loss": 0.4765577470093241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4765577470093241
+    },
+    {
+      "train_step": 5616,
+      "loss": 0.5468169848084072,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5468169848084072
+    },
+    {
+      "train_step": 5617,
+      "loss": 0.6325778218034208,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6325778218034208
+    },
+    {
+      "train_step": 5618,
+      "loss": 0.7690841357404266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7690841357404266
+    },
+    {
+      "train_step": 5619,
+      "loss": 0.7154815625426912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7154815625426912
+    },
+    {
+      "train_step": 5620,
+      "loss": 0.661270471109776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.661270471109776
+    },
+    {
+      "train_step": 5621,
+      "loss": 0.5657107880694899,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5657107880694899
+    },
+    {
+      "train_step": 5622,
+      "loss": 0.5973796294385653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5973796294385653
+    },
+    {
+      "train_step": 5623,
+      "loss": 0.5863855564658903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5863855564658903
+    },
+    {
+      "train_step": 5624,
+      "loss": 0.5961008382156936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5961008382156936
+    },
+    {
+      "train_step": 5625,
+      "loss": 0.4840238936713843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4840238936713843
+    },
+    {
+      "train_step": 5626,
+      "loss": 0.6174188319560093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6174188319560093
+    },
+    {
+      "train_step": 5627,
+      "loss": 0.5437811205725361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5437811205725361
+    },
+    {
+      "train_step": 5628,
+      "loss": 0.5780408387985077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5780408387985077
+    },
+    {
+      "train_step": 5629,
+      "loss": 0.5259224750315827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5259224750315827
+    },
+    {
+      "train_step": 5630,
+      "loss": 0.5155755534755858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5155755534755858
+    },
+    {
+      "train_step": 5631,
+      "loss": 0.2680046763956543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2680046763956543
+    },
+    {
+      "train_step": 5632,
+      "loss": 0.7282640651213922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7282640651213922
+    },
+    {
+      "train_step": 5633,
+      "loss": 0.34467127509768736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34467127509768736
+    },
+    {
+      "train_step": 5634,
+      "loss": 0.7120327438186163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7120327438186163
+    },
+    {
+      "train_step": 5635,
+      "loss": 0.6387712293219422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6387712293219422
+    },
+    {
+      "train_step": 5636,
+      "loss": 0.44092299551480724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44092299551480724
+    },
+    {
+      "train_step": 5637,
+      "loss": 0.5812256155440186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5812256155440186
+    },
+    {
+      "train_step": 5638,
+      "loss": 0.49696043812794244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49696043812794244
+    },
+    {
+      "train_step": 5639,
+      "loss": 0.6176976099959032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6176976099959032
+    },
+    {
+      "train_step": 5640,
+      "loss": 0.5460641937482815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5460641937482815
+    },
+    {
+      "train_step": 5641,
+      "loss": 0.6263132706572825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6263132706572825
+    },
+    {
+      "train_step": 5642,
+      "loss": 0.3591193366505479,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3591193366505479
+    },
+    {
+      "train_step": 5643,
+      "loss": 0.6338900411914415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6338900411914415
+    },
+    {
+      "train_step": 5644,
+      "loss": 0.7678590757938074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7678590757938074
+    },
+    {
+      "train_step": 5645,
+      "loss": 0.4933657099441167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4933657099441167
+    },
+    {
+      "train_step": 5646,
+      "loss": 0.4721487051696976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4721487051696976
+    },
+    {
+      "train_step": 5647,
+      "loss": 0.47550568535136856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47550568535136856
+    },
+    {
+      "train_step": 5648,
+      "loss": 0.5943094504494161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5943094504494161
+    },
+    {
+      "train_step": 5649,
+      "loss": 0.4807440227650459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4807440227650459
+    },
+    {
+      "train_step": 5650,
+      "loss": 0.6544494889466462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6544494889466462
+    },
+    {
+      "train_step": 5651,
+      "loss": 0.2961391372042367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2961391372042367
+    },
+    {
+      "train_step": 5652,
+      "loss": 0.7554747816328125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7554747816328125
+    },
+    {
+      "train_step": 5653,
+      "loss": 0.48477536958524536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48477536958524536
+    },
+    {
+      "train_step": 5654,
+      "loss": 0.7803395652406432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7803395652406432
+    },
+    {
+      "train_step": 5655,
+      "loss": 0.7491188695204133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7491188695204133
+    },
+    {
+      "train_step": 5656,
+      "loss": 0.40483449820256234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40483449820256234
+    },
+    {
+      "train_step": 5657,
+      "loss": 0.5927109156952179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5927109156952179
+    },
+    {
+      "train_step": 5658,
+      "loss": 0.4805529764491183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4805529764491183
+    },
+    {
+      "train_step": 5659,
+      "loss": 0.584857450705443,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.584857450705443
+    },
+    {
+      "train_step": 5660,
+      "loss": 0.4636245330791013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4636245330791013
+    },
+    {
+      "train_step": 5661,
+      "loss": 0.49786340055638384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49786340055638384
+    },
+    {
+      "train_step": 5662,
+      "loss": 0.552966386824624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.552966386824624
+    },
+    {
+      "train_step": 5663,
+      "loss": 0.5757159900101981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5757159900101981
+    },
+    {
+      "train_step": 5664,
+      "loss": 0.5387715183898252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5387715183898252
+    },
+    {
+      "train_step": 5665,
+      "loss": 0.6959847102785093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6959847102785093
+    },
+    {
+      "train_step": 5666,
+      "loss": 0.500590408300312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.500590408300312
+    },
+    {
+      "train_step": 5667,
+      "loss": 0.5554439676418975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5554439676418975
+    },
+    {
+      "train_step": 5668,
+      "loss": 0.5558023354781199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5558023354781199
+    },
+    {
+      "train_step": 5669,
+      "loss": 0.5029964967078863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5029964967078863
+    },
+    {
+      "train_step": 5670,
+      "loss": 0.5299647431244384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5299647431244384
+    },
+    {
+      "train_step": 5671,
+      "loss": 0.6459164645560851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6459164645560851
+    },
+    {
+      "train_step": 5672,
+      "loss": 0.4003566314408361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4003566314408361
+    },
+    {
+      "train_step": 5673,
+      "loss": 0.5252494171876446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5252494171876446
+    },
+    {
+      "train_step": 5674,
+      "loss": 0.7935181332691009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7935181332691009
+    },
+    {
+      "train_step": 5675,
+      "loss": 0.4885373666410123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4885373666410123
+    },
+    {
+      "train_step": 5676,
+      "loss": 0.7856178621632554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7856178621632554
+    },
+    {
+      "train_step": 5677,
+      "loss": 0.5350418949170859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5350418949170859
+    },
+    {
+      "train_step": 5678,
+      "loss": 0.6028385436303347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6028385436303347
+    },
+    {
+      "train_step": 5679,
+      "loss": 0.5061636199554311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5061636199554311
+    },
+    {
+      "train_step": 5680,
+      "loss": 0.5680238902875591,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5680238902875591
+    },
+    {
+      "train_step": 5681,
+      "loss": 0.5942770951316012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5942770951316012
+    },
+    {
+      "train_step": 5682,
+      "loss": 0.521483016288844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.521483016288844
+    },
+    {
+      "train_step": 5683,
+      "loss": 0.6633911517371119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6633911517371119
+    },
+    {
+      "train_step": 5684,
+      "loss": 0.5356160255394613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5356160255394613
+    },
+    {
+      "train_step": 5685,
+      "loss": 0.7335702305757513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7335702305757513
+    },
+    {
+      "train_step": 5686,
+      "loss": 0.5862039879894209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5862039879894209
+    },
+    {
+      "train_step": 5687,
+      "loss": 0.4242389823469002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4242389823469002
+    },
+    {
+      "train_step": 5688,
+      "loss": 0.4882433795916252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4882433795916252
+    },
+    {
+      "train_step": 5689,
+      "loss": 0.6668179676931797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6668179676931797
+    },
+    {
+      "train_step": 5690,
+      "loss": 0.4268256489358557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4268256489358557
+    },
+    {
+      "train_step": 5691,
+      "loss": 0.41273678075905684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41273678075905684
+    },
+    {
+      "train_step": 5692,
+      "loss": 0.7969184699574767,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7969184699574767
+    },
+    {
+      "train_step": 5693,
+      "loss": 0.5352676171439485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5352676171439485
+    },
+    {
+      "train_step": 5694,
+      "loss": 0.6479737437239589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6479737437239589
+    },
+    {
+      "train_step": 5695,
+      "loss": 0.5412606153371536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5412606153371536
+    },
+    {
+      "train_step": 5696,
+      "loss": 0.6468129685217003,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6468129685217003
+    },
+    {
+      "train_step": 5697,
+      "loss": 0.588842284221728,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.588842284221728
+    },
+    {
+      "train_step": 5698,
+      "loss": 0.2983960580235058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2983960580235058
+    },
+    {
+      "train_step": 5699,
+      "loss": 0.540779267023764,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.540779267023764
+    },
+    {
+      "train_step": 5700,
+      "loss": 0.5707344073419838,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5707344073419838
+    },
+    {
+      "train_step": 5701,
+      "loss": 0.5402105379207002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5402105379207002
+    },
+    {
+      "train_step": 5702,
+      "loss": 0.7170249402367432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7170249402367432
+    },
+    {
+      "train_step": 5703,
+      "loss": 0.5643801117347405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5643801117347405
+    },
+    {
+      "train_step": 5704,
+      "loss": 0.5693774386425843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5693774386425843
+    },
+    {
+      "train_step": 5705,
+      "loss": 0.6957384704175604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6957384704175604
+    },
+    {
+      "train_step": 5706,
+      "loss": 0.5015140892555898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5015140892555898
+    },
+    {
+      "train_step": 5707,
+      "loss": 0.5406089005521095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5406089005521095
+    },
+    {
+      "train_step": 5708,
+      "loss": 0.3916612670300589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3916612670300589
+    },
+    {
+      "train_step": 5709,
+      "loss": 0.6554093256607866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6554093256607866
+    },
+    {
+      "train_step": 5710,
+      "loss": 0.4581566285040571,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4581566285040571
+    },
+    {
+      "train_step": 5711,
+      "loss": 0.47590889056629215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47590889056629215
+    },
+    {
+      "train_step": 5712,
+      "loss": 0.4329279493408988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4329279493408988
+    },
+    {
+      "train_step": 5713,
+      "loss": 0.5887528741251542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5887528741251542
+    },
+    {
+      "train_step": 5714,
+      "loss": 0.5114949114915224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5114949114915224
+    },
+    {
+      "train_step": 5715,
+      "loss": 0.5578442356736917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5578442356736917
+    },
+    {
+      "train_step": 5716,
+      "loss": 0.5377153981094138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5377153981094138
+    },
+    {
+      "train_step": 5717,
+      "loss": 0.41967463402748534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41967463402748534
+    },
+    {
+      "train_step": 5718,
+      "loss": 0.6454576927897211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6454576927897211
+    },
+    {
+      "train_step": 5719,
+      "loss": 0.4947844703445855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4947844703445855
+    },
+    {
+      "train_step": 5720,
+      "loss": 0.6480601394498436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6480601394498436
+    },
+    {
+      "train_step": 5721,
+      "loss": 0.4200508464244444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4200508464244444
+    },
+    {
+      "train_step": 5722,
+      "loss": 0.45394879337510885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45394879337510885
+    },
+    {
+      "train_step": 5723,
+      "loss": 0.5902751460766362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5902751460766362
+    },
+    {
+      "train_step": 5724,
+      "loss": 0.46057310354900627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46057310354900627
+    },
+    {
+      "train_step": 5725,
+      "loss": 0.499362461885274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.499362461885274
+    },
+    {
+      "train_step": 5726,
+      "loss": 0.7481167213465253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7481167213465253
+    },
+    {
+      "train_step": 5727,
+      "loss": 0.5958433223053745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5958433223053745
+    },
+    {
+      "train_step": 5728,
+      "loss": 0.7567393196082901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7567393196082901
+    },
+    {
+      "train_step": 5729,
+      "loss": 0.6699423227179934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6699423227179934
+    },
+    {
+      "train_step": 5730,
+      "loss": 0.468975999449661,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.468975999449661
+    },
+    {
+      "train_step": 5731,
+      "loss": 0.4817494926176307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4817494926176307
+    },
+    {
+      "train_step": 5732,
+      "loss": 0.4789017961978869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4789017961978869
+    },
+    {
+      "train_step": 5733,
+      "loss": 0.66039962597878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66039962597878
+    },
+    {
+      "train_step": 5734,
+      "loss": 0.6794956497618314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6794956497618314
+    },
+    {
+      "train_step": 5735,
+      "loss": 0.6357730443764256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6357730443764256
+    },
+    {
+      "train_step": 5736,
+      "loss": 0.6261460896414062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6261460896414062
+    },
+    {
+      "train_step": 5737,
+      "loss": 0.5751028340421844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5751028340421844
+    },
+    {
+      "train_step": 5738,
+      "loss": 0.5151341967700453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5151341967700453
+    },
+    {
+      "train_step": 5739,
+      "loss": 0.49849334324700567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49849334324700567
+    },
+    {
+      "train_step": 5740,
+      "loss": 0.479859664314855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.479859664314855
+    },
+    {
+      "train_step": 5741,
+      "loss": 0.6427665969554464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6427665969554464
+    },
+    {
+      "train_step": 5742,
+      "loss": 0.5940056755618282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5940056755618282
+    },
+    {
+      "train_step": 5743,
+      "loss": 0.5832070669652105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5832070669652105
+    },
+    {
+      "train_step": 5744,
+      "loss": 0.694328010502108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.694328010502108
+    },
+    {
+      "train_step": 5745,
+      "loss": 0.6404817122278341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6404817122278341
+    },
+    {
+      "train_step": 5746,
+      "loss": 0.7213571655502223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7213571655502223
+    },
+    {
+      "train_step": 5747,
+      "loss": 0.6090939481810378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6090939481810378
+    },
+    {
+      "train_step": 5748,
+      "loss": 0.6248111116602622,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6248111116602622
+    },
+    {
+      "train_step": 5749,
+      "loss": 0.6703285839804461,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6703285839804461
+    },
+    {
+      "train_step": 5750,
+      "loss": 0.6306101220454102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6306101220454102
+    },
+    {
+      "train_step": 5751,
+      "loss": 0.6627094661277214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6627094661277214
+    },
+    {
+      "train_step": 5752,
+      "loss": 0.501367839179562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.501367839179562
+    },
+    {
+      "train_step": 5753,
+      "loss": 0.46109597908604805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46109597908604805
+    },
+    {
+      "train_step": 5754,
+      "loss": 0.4289314206111797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4289314206111797
+    },
+    {
+      "train_step": 5755,
+      "loss": 0.43324496392189965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43324496392189965
+    },
+    {
+      "train_step": 5756,
+      "loss": 0.42192680804496974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42192680804496974
+    },
+    {
+      "train_step": 5757,
+      "loss": 0.46775931490800016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46775931490800016
+    },
+    {
+      "train_step": 5758,
+      "loss": 0.6825406208543825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6825406208543825
+    },
+    {
+      "train_step": 5759,
+      "loss": 0.4350845855122929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4350845855122929
+    },
+    {
+      "train_step": 5760,
+      "loss": 0.7076533610062371,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7076533610062371
+    },
+    {
+      "train_step": 5761,
+      "loss": 0.3708455791003363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3708455791003363
+    },
+    {
+      "train_step": 5762,
+      "loss": 0.6770544033548369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6770544033548369
+    },
+    {
+      "train_step": 5763,
+      "loss": 0.49915419053955784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49915419053955784
+    },
+    {
+      "train_step": 5764,
+      "loss": 0.6842885879306929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6842885879306929
+    },
+    {
+      "train_step": 5765,
+      "loss": 0.5693771886434081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5693771886434081
+    },
+    {
+      "train_step": 5766,
+      "loss": 0.4947481247065517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4947481247065517
+    },
+    {
+      "train_step": 5767,
+      "loss": 0.35603920215720647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35603920215720647
+    },
+    {
+      "train_step": 5768,
+      "loss": 0.544603550247698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.544603550247698
+    },
+    {
+      "train_step": 5769,
+      "loss": 0.49908231712814843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49908231712814843
+    },
+    {
+      "train_step": 5770,
+      "loss": 0.5874256543979213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5874256543979213
+    },
+    {
+      "train_step": 5771,
+      "loss": 0.5699866209610776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5699866209610776
+    },
+    {
+      "train_step": 5772,
+      "loss": 0.49723057425075284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49723057425075284
+    },
+    {
+      "train_step": 5773,
+      "loss": 0.568474334779237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.568474334779237
+    },
+    {
+      "train_step": 5774,
+      "loss": 0.7759768482425047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7759768482425047
+    },
+    {
+      "train_step": 5775,
+      "loss": 0.5332237750123892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5332237750123892
+    },
+    {
+      "train_step": 5776,
+      "loss": 0.6033066225011262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6033066225011262
+    },
+    {
+      "train_step": 5777,
+      "loss": 0.706915292483231,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.706915292483231
+    },
+    {
+      "train_step": 5778,
+      "loss": 0.6286021147158266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6286021147158266
+    },
+    {
+      "train_step": 5779,
+      "loss": 0.8741350990620207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8741350990620207
+    },
+    {
+      "train_step": 5780,
+      "loss": 0.42549635708582795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42549635708582795
+    },
+    {
+      "train_step": 5781,
+      "loss": 0.6918404714330053,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6918404714330053
+    },
+    {
+      "train_step": 5782,
+      "loss": 0.6540119117827775,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6540119117827775
+    },
+    {
+      "train_step": 5783,
+      "loss": 0.5664472952472104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5664472952472104
+    },
+    {
+      "train_step": 5784,
+      "loss": 0.48651060621145226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48651060621145226
+    },
+    {
+      "train_step": 5785,
+      "loss": 0.665211081120691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.665211081120691
+    },
+    {
+      "train_step": 5786,
+      "loss": 0.6102101782604947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6102101782604947
+    },
+    {
+      "train_step": 5787,
+      "loss": 0.4855511104440615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4855511104440615
+    },
+    {
+      "train_step": 5788,
+      "loss": 0.6338121043245797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6338121043245797
+    },
+    {
+      "train_step": 5789,
+      "loss": 0.4888916202731932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4888916202731932
+    },
+    {
+      "train_step": 5790,
+      "loss": 0.682716697173904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.682716697173904
+    },
+    {
+      "train_step": 5791,
+      "loss": 0.5328719928727689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5328719928727689
+    },
+    {
+      "train_step": 5792,
+      "loss": 0.5492773033201865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5492773033201865
+    },
+    {
+      "train_step": 5793,
+      "loss": 0.4805434671001355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4805434671001355
+    },
+    {
+      "train_step": 5794,
+      "loss": 0.6141312955515779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6141312955515779
+    },
+    {
+      "train_step": 5795,
+      "loss": 0.49315980634825984,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49315980634825984
+    },
+    {
+      "train_step": 5796,
+      "loss": 0.634599620492337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.634599620492337
+    },
+    {
+      "train_step": 5797,
+      "loss": 0.4729565538744347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4729565538744347
+    },
+    {
+      "train_step": 5798,
+      "loss": 0.5619966978568037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5619966978568037
+    },
+    {
+      "train_step": 5799,
+      "loss": 0.4181336689644669,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4181336689644669
+    },
+    {
+      "train_step": 5800,
+      "loss": 0.40049172738890293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40049172738890293
+    },
+    {
+      "train_step": 5801,
+      "loss": 0.5807253859200573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5807253859200573
+    },
+    {
+      "train_step": 5802,
+      "loss": 0.6405490005255505,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6405490005255505
+    },
+    {
+      "train_step": 5803,
+      "loss": 0.5111327191815483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5111327191815483
+    },
+    {
+      "train_step": 5804,
+      "loss": 0.6732573976769322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6732573976769322
+    },
+    {
+      "train_step": 5805,
+      "loss": 0.7329792880372615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7329792880372615
+    },
+    {
+      "train_step": 5806,
+      "loss": 0.49762804481763456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49762804481763456
+    },
+    {
+      "train_step": 5807,
+      "loss": 0.48856192060482184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48856192060482184
+    },
+    {
+      "train_step": 5808,
+      "loss": 0.44045184769431783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44045184769431783
+    },
+    {
+      "train_step": 5809,
+      "loss": 0.45806096253619727,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45806096253619727
+    },
+    {
+      "train_step": 5810,
+      "loss": 0.4324452935740864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4324452935740864
+    },
+    {
+      "train_step": 5811,
+      "loss": 0.5519795991467171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5519795991467171
+    },
+    {
+      "train_step": 5812,
+      "loss": 0.42387668870239537,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42387668870239537
+    },
+    {
+      "train_step": 5813,
+      "loss": 0.5067077193248527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5067077193248527
+    },
+    {
+      "train_step": 5814,
+      "loss": 0.5329288901957879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5329288901957879
+    },
+    {
+      "train_step": 5815,
+      "loss": 0.5261744531026981,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5261744531026981
+    },
+    {
+      "train_step": 5816,
+      "loss": 0.46689242247317186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46689242247317186
+    },
+    {
+      "train_step": 5817,
+      "loss": 0.619467347256375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.619467347256375
+    },
+    {
+      "train_step": 5818,
+      "loss": 0.45046363506005427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45046363506005427
+    },
+    {
+      "train_step": 5819,
+      "loss": 0.4970569459563667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4970569459563667
+    },
+    {
+      "train_step": 5820,
+      "loss": 0.6732473252865286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6732473252865286
+    },
+    {
+      "train_step": 5821,
+      "loss": 0.6530285244946179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6530285244946179
+    },
+    {
+      "train_step": 5822,
+      "loss": 0.4137381525595304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4137381525595304
+    },
+    {
+      "train_step": 5823,
+      "loss": 0.4324840401080042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4324840401080042
+    },
+    {
+      "train_step": 5824,
+      "loss": 0.5219007227388582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5219007227388582
+    },
+    {
+      "train_step": 5825,
+      "loss": 0.3865970861274348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3865970861274348
+    },
+    {
+      "train_step": 5826,
+      "loss": 0.42484551601946685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42484551601946685
+    },
+    {
+      "train_step": 5827,
+      "loss": 0.43510817516477174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43510817516477174
+    },
+    {
+      "train_step": 5828,
+      "loss": 0.44021018776390214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44021018776390214
+    },
+    {
+      "train_step": 5829,
+      "loss": 0.7734901753749567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7734901753749567
+    },
+    {
+      "train_step": 5830,
+      "loss": 0.6370176607027597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6370176607027597
+    },
+    {
+      "train_step": 5831,
+      "loss": 0.5621220083185815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5621220083185815
+    },
+    {
+      "train_step": 5832,
+      "loss": 0.6209988379120109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6209988379120109
+    },
+    {
+      "train_step": 5833,
+      "loss": 0.5123525017438275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5123525017438275
+    },
+    {
+      "train_step": 5834,
+      "loss": 0.5093573942157361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5093573942157361
+    },
+    {
+      "train_step": 5835,
+      "loss": 0.5553593359057177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5553593359057177
+    },
+    {
+      "train_step": 5836,
+      "loss": 0.4136631980611801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4136631980611801
+    },
+    {
+      "train_step": 5837,
+      "loss": 0.28527955663135607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.28527955663135607
+    },
+    {
+      "train_step": 5838,
+      "loss": 0.48741115889745174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48741115889745174
+    },
+    {
+      "train_step": 5839,
+      "loss": 0.5153821090225023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5153821090225023
+    },
+    {
+      "train_step": 5840,
+      "loss": 0.1736989760771528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.1736989760771528
+    },
+    {
+      "train_step": 5841,
+      "loss": 0.39417449636532575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39417449636532575
+    },
+    {
+      "train_step": 5842,
+      "loss": 0.3688347081495137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3688347081495137
+    },
+    {
+      "train_step": 5843,
+      "loss": 0.48316343707344145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48316343707344145
+    },
+    {
+      "train_step": 5844,
+      "loss": 0.18089381871424032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.18089381871424032
+    },
+    {
+      "train_step": 5845,
+      "loss": 0.551425235730161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.551425235730161
+    },
+    {
+      "train_step": 5846,
+      "loss": 0.5058236564098142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5058236564098142
+    },
+    {
+      "train_step": 5847,
+      "loss": 0.3550748149951502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3550748149951502
+    },
+    {
+      "train_step": 5848,
+      "loss": 0.556953766921744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.556953766921744
+    },
+    {
+      "train_step": 5849,
+      "loss": 0.34387103672496466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34387103672496466
+    },
+    {
+      "train_step": 5850,
+      "loss": 0.3380486559506214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3380486559506214
+    },
+    {
+      "train_step": 5851,
+      "loss": 0.36657718325789007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36657718325789007
+    },
+    {
+      "train_step": 5852,
+      "loss": 0.4138120203302045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4138120203302045
+    },
+    {
+      "train_step": 5853,
+      "loss": 0.3897727596798886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3897727596798886
+    },
+    {
+      "train_step": 5854,
+      "loss": 0.3648239316710311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3648239316710311
+    },
+    {
+      "train_step": 5855,
+      "loss": 0.371975821331876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.371975821331876
+    },
+    {
+      "train_step": 5856,
+      "loss": 0.11339067918818944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.11339067918818944
+    },
+    {
+      "train_step": 5857,
+      "loss": 0.2735121741776937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2735121741776937
+    },
+    {
+      "train_step": 5858,
+      "loss": 0.20115546708737464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.20115546708737464
+    },
+    {
+      "train_step": 5859,
+      "loss": 0.45215184427213345,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45215184427213345
+    },
+    {
+      "train_step": 5860,
+      "loss": 0.2107787019978536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2107787019978536
+    },
+    {
+      "train_step": 5861,
+      "loss": 0.22463553087991875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.22463553087991875
+    },
+    {
+      "train_step": 5862,
+      "loss": 0.5259393749144102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5259393749144102
+    },
+    {
+      "train_step": 5863,
+      "loss": 0.315983486719923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.315983486719923
+    },
+    {
+      "train_step": 5864,
+      "loss": 0.2464335814734882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2464335814734882
+    },
+    {
+      "train_step": 5865,
+      "loss": 0.4854266508215875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4854266508215875
+    },
+    {
+      "train_step": 5866,
+      "loss": 0.27715691975098655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.27715691975098655
+    },
+    {
+      "train_step": 5867,
+      "loss": 0.21471296538657428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.21471296538657428
+    },
+    {
+      "train_step": 5868,
+      "loss": 0.48631751692372294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48631751692372294
+    },
+    {
+      "train_step": 5869,
+      "loss": 0.5118090982854382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5118090982854382
+    },
+    {
+      "train_step": 5870,
+      "loss": 0.5030869730783262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5030869730783262
+    },
+    {
+      "train_step": 5871,
+      "loss": 0.40683792487021964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40683792487021964
+    },
+    {
+      "train_step": 5872,
+      "loss": 0.5044550825600349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5044550825600349
+    },
+    {
+      "train_step": 5873,
+      "loss": 0.45026325356638697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45026325356638697
+    },
+    {
+      "train_step": 5874,
+      "loss": 0.3762370511092296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3762370511092296
+    },
+    {
+      "train_step": 5875,
+      "loss": 0.47755422947246895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47755422947246895
+    },
+    {
+      "train_step": 5876,
+      "loss": 0.5333835929060661,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5333835929060661
+    },
+    {
+      "train_step": 5877,
+      "loss": 0.410526499910145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.410526499910145
+    },
+    {
+      "train_step": 5878,
+      "loss": 0.5858783753587588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5858783753587588
+    },
+    {
+      "train_step": 5879,
+      "loss": 0.519944433125233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.519944433125233
+    },
+    {
+      "train_step": 5880,
+      "loss": 0.6731565959649473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6731565959649473
+    },
+    {
+      "train_step": 5881,
+      "loss": 0.5784433737314096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5784433737314096
+    },
+    {
+      "train_step": 5882,
+      "loss": 0.5712931859468028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5712931859468028
+    },
+    {
+      "train_step": 5883,
+      "loss": 0.5626333651004791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5626333651004791
+    },
+    {
+      "train_step": 5884,
+      "loss": 0.522510565571187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.522510565571187
+    },
+    {
+      "train_step": 5885,
+      "loss": 0.47635979880816415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47635979880816415
+    },
+    {
+      "train_step": 5886,
+      "loss": 0.542202674615973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.542202674615973
+    },
+    {
+      "train_step": 5887,
+      "loss": 0.5518411565524868,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5518411565524868
+    },
+    {
+      "train_step": 5888,
+      "loss": 0.581143516259801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.581143516259801
+    },
+    {
+      "train_step": 5889,
+      "loss": 0.66026278550995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66026278550995
+    },
+    {
+      "train_step": 5890,
+      "loss": 0.7013642670953806,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7013642670953806
+    },
+    {
+      "train_step": 5891,
+      "loss": 0.48883679266644026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48883679266644026
+    },
+    {
+      "train_step": 5892,
+      "loss": 0.5387439754634997,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5387439754634997
+    },
+    {
+      "train_step": 5893,
+      "loss": 0.618023619734946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.618023619734946
+    },
+    {
+      "train_step": 5894,
+      "loss": 0.5058900021998699,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5058900021998699
+    },
+    {
+      "train_step": 5895,
+      "loss": 0.4409177457090052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4409177457090052
+    },
+    {
+      "train_step": 5896,
+      "loss": 0.45958368101032365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45958368101032365
+    },
+    {
+      "train_step": 5897,
+      "loss": 0.3373120420937236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3373120420937236
+    },
+    {
+      "train_step": 5898,
+      "loss": 0.4494827234996302,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4494827234996302
+    },
+    {
+      "train_step": 5899,
+      "loss": 0.5582974732598803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5582974732598803
+    },
+    {
+      "train_step": 5900,
+      "loss": 0.5896019373986678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5896019373986678
+    },
+    {
+      "train_step": 5901,
+      "loss": 0.4830544604730047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4830544604730047
+    },
+    {
+      "train_step": 5902,
+      "loss": 0.41578613918309704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41578613918309704
+    },
+    {
+      "train_step": 5903,
+      "loss": 0.3940955826179619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3940955826179619
+    },
+    {
+      "train_step": 5904,
+      "loss": 0.4904574148250177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4904574148250177
+    },
+    {
+      "train_step": 5905,
+      "loss": 0.6597119839117551,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6597119839117551
+    },
+    {
+      "train_step": 5906,
+      "loss": 0.6485464189030853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6485464189030853
+    },
+    {
+      "train_step": 5907,
+      "loss": 0.33049611641808374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33049611641808374
+    },
+    {
+      "train_step": 5908,
+      "loss": 0.6982527190633219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6982527190633219
+    },
+    {
+      "train_step": 5909,
+      "loss": 0.6301580702493841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6301580702493841
+    },
+    {
+      "train_step": 5910,
+      "loss": 0.6519195050508971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6519195050508971
+    },
+    {
+      "train_step": 5911,
+      "loss": 0.4457363273035363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4457363273035363
+    },
+    {
+      "train_step": 5912,
+      "loss": 0.3978376412104641,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3978376412104641
+    },
+    {
+      "train_step": 5913,
+      "loss": 0.6743339573873383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6743339573873383
+    },
+    {
+      "train_step": 5914,
+      "loss": 0.4435655089816438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4435655089816438
+    },
+    {
+      "train_step": 5915,
+      "loss": 0.320060389503275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.320060389503275
+    },
+    {
+      "train_step": 5916,
+      "loss": 0.6786874133091352,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6786874133091352
+    },
+    {
+      "train_step": 5917,
+      "loss": 0.658425866872449,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.658425866872449
+    },
+    {
+      "train_step": 5918,
+      "loss": 0.5199996468317359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5199996468317359
+    },
+    {
+      "train_step": 5919,
+      "loss": 0.42405257913291344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42405257913291344
+    },
+    {
+      "train_step": 5920,
+      "loss": 0.6240255650168729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6240255650168729
+    },
+    {
+      "train_step": 5921,
+      "loss": 0.5848878875993844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5848878875993844
+    },
+    {
+      "train_step": 5922,
+      "loss": 0.5553332414397476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5553332414397476
+    },
+    {
+      "train_step": 5923,
+      "loss": 0.4399188046524549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4399188046524549
+    },
+    {
+      "train_step": 5924,
+      "loss": 0.698481938103535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.698481938103535
+    },
+    {
+      "train_step": 5925,
+      "loss": 0.5820663949875364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5820663949875364
+    },
+    {
+      "train_step": 5926,
+      "loss": 0.7531718579195252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7531718579195252
+    },
+    {
+      "train_step": 5927,
+      "loss": 0.5704666523589782,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5704666523589782
+    },
+    {
+      "train_step": 5928,
+      "loss": 0.4651052159802405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4651052159802405
+    },
+    {
+      "train_step": 5929,
+      "loss": 0.3930161721980674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3930161721980674
+    },
+    {
+      "train_step": 5930,
+      "loss": 0.5806631935882455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5806631935882455
+    },
+    {
+      "train_step": 5931,
+      "loss": 0.48963696044716815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48963696044716815
+    },
+    {
+      "train_step": 5932,
+      "loss": 0.6174652671089428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6174652671089428
+    },
+    {
+      "train_step": 5933,
+      "loss": 0.4942290470661929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4942290470661929
+    },
+    {
+      "train_step": 5934,
+      "loss": 0.5032000358555492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5032000358555492
+    },
+    {
+      "train_step": 5935,
+      "loss": 0.574069373569883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.574069373569883
+    },
+    {
+      "train_step": 5936,
+      "loss": 0.4906032369679754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4906032369679754
+    },
+    {
+      "train_step": 5937,
+      "loss": 0.5680851156586901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5680851156586901
+    },
+    {
+      "train_step": 5938,
+      "loss": 0.4784449004558365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4784449004558365
+    },
+    {
+      "train_step": 5939,
+      "loss": 0.6009466687671737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6009466687671737
+    },
+    {
+      "train_step": 5940,
+      "loss": 0.4321221815442171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4321221815442171
+    },
+    {
+      "train_step": 5941,
+      "loss": 0.558567933196679,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.558567933196679
+    },
+    {
+      "train_step": 5942,
+      "loss": 0.7051865616280781,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7051865616280781
+    },
+    {
+      "train_step": 5943,
+      "loss": 0.6000549249300527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6000549249300527
+    },
+    {
+      "train_step": 5944,
+      "loss": 0.5187493443650882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5187493443650882
+    },
+    {
+      "train_step": 5945,
+      "loss": 0.32559700140841735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32559700140841735
+    },
+    {
+      "train_step": 5946,
+      "loss": 0.5032345329862521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5032345329862521
+    },
+    {
+      "train_step": 5947,
+      "loss": 0.35464289751038164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35464289751038164
+    },
+    {
+      "train_step": 5948,
+      "loss": 0.5621089869286754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5621089869286754
+    },
+    {
+      "train_step": 5949,
+      "loss": 0.511887363222356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.511887363222356
+    },
+    {
+      "train_step": 5950,
+      "loss": 0.46676105353389674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46676105353389674
+    },
+    {
+      "train_step": 5951,
+      "loss": 0.3651661615960802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3651661615960802
+    },
+    {
+      "train_step": 5952,
+      "loss": 0.4370506534760974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4370506534760974
+    },
+    {
+      "train_step": 5953,
+      "loss": 0.6596221219616297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6596221219616297
+    },
+    {
+      "train_step": 5954,
+      "loss": 0.6473077468900346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6473077468900346
+    },
+    {
+      "train_step": 5955,
+      "loss": 0.5302022008931451,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5302022008931451
+    },
+    {
+      "train_step": 5956,
+      "loss": 0.5083561628984861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5083561628984861
+    },
+    {
+      "train_step": 5957,
+      "loss": 0.49196964737965865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49196964737965865
+    },
+    {
+      "train_step": 5958,
+      "loss": 0.4047119757695562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4047119757695562
+    },
+    {
+      "train_step": 5959,
+      "loss": 0.7595082578383769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7595082578383769
+    },
+    {
+      "train_step": 5960,
+      "loss": 0.4798599251668815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4798599251668815
+    },
+    {
+      "train_step": 5961,
+      "loss": 0.6051256250956047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6051256250956047
+    },
+    {
+      "train_step": 5962,
+      "loss": 0.608232623997478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.608232623997478
+    },
+    {
+      "train_step": 5963,
+      "loss": 0.586224521588087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.586224521588087
+    },
+    {
+      "train_step": 5964,
+      "loss": 0.4582071652232448,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4582071652232448
+    },
+    {
+      "train_step": 5965,
+      "loss": 0.5850101343821831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5850101343821831
+    },
+    {
+      "train_step": 5966,
+      "loss": 0.636434879452196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.636434879452196
+    },
+    {
+      "train_step": 5967,
+      "loss": 0.5499557857874849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5499557857874849
+    },
+    {
+      "train_step": 5968,
+      "loss": 0.48126564436015296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48126564436015296
+    },
+    {
+      "train_step": 5969,
+      "loss": 0.6479702457011798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6479702457011798
+    },
+    {
+      "train_step": 5970,
+      "loss": 0.570853829795051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.570853829795051
+    },
+    {
+      "train_step": 5971,
+      "loss": 0.37818312941451293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37818312941451293
+    },
+    {
+      "train_step": 5972,
+      "loss": 0.3922033725480719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3922033725480719
+    },
+    {
+      "train_step": 5973,
+      "loss": 0.6072379561148047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6072379561148047
+    },
+    {
+      "train_step": 5974,
+      "loss": 0.467466955444212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.467466955444212
+    },
+    {
+      "train_step": 5975,
+      "loss": 0.6242070719959906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6242070719959906
+    },
+    {
+      "train_step": 5976,
+      "loss": 0.5585680430345967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5585680430345967
+    },
+    {
+      "train_step": 5977,
+      "loss": 0.36072294176242903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36072294176242903
+    },
+    {
+      "train_step": 5978,
+      "loss": 0.5810948962212333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5810948962212333
+    },
+    {
+      "train_step": 5979,
+      "loss": 0.589117702392161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.589117702392161
+    },
+    {
+      "train_step": 5980,
+      "loss": 0.6232693016236888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6232693016236888
+    },
+    {
+      "train_step": 5981,
+      "loss": 0.4137098401796623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4137098401796623
+    },
+    {
+      "train_step": 5982,
+      "loss": 0.4922290785743257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4922290785743257
+    },
+    {
+      "train_step": 5983,
+      "loss": 0.5367114276436378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5367114276436378
+    },
+    {
+      "train_step": 5984,
+      "loss": 0.594204982013945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.594204982013945
+    },
+    {
+      "train_step": 5985,
+      "loss": 0.4850573027058422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4850573027058422
+    },
+    {
+      "train_step": 5986,
+      "loss": 0.8398064497693543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8398064497693543
+    },
+    {
+      "train_step": 5987,
+      "loss": 0.5710569782127772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5710569782127772
+    },
+    {
+      "train_step": 5988,
+      "loss": 0.4748645238177099,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4748645238177099
+    },
+    {
+      "train_step": 5989,
+      "loss": 0.41607451013435015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41607451013435015
+    },
+    {
+      "train_step": 5990,
+      "loss": 0.5316325298270825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5316325298270825
+    },
+    {
+      "train_step": 5991,
+      "loss": 0.49723032597049843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49723032597049843
+    },
+    {
+      "train_step": 5992,
+      "loss": 0.39155813506685805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39155813506685805
+    },
+    {
+      "train_step": 5993,
+      "loss": 0.5472741885768143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5472741885768143
+    },
+    {
+      "train_step": 5994,
+      "loss": 0.47820329123472644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47820329123472644
+    },
+    {
+      "train_step": 5995,
+      "loss": 0.6177395023622306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6177395023622306
+    },
+    {
+      "train_step": 5996,
+      "loss": 0.5069535130019158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5069535130019158
+    },
+    {
+      "train_step": 5997,
+      "loss": 0.6575363376906682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6575363376906682
+    },
+    {
+      "train_step": 5998,
+      "loss": 0.6601576120640856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6601576120640856
+    },
+    {
+      "train_step": 5999,
+      "loss": 0.3361999679999214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3361999679999214
+    },
+    {
+      "train_step": 6000,
+      "loss": 0.4505646946617974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4505646946617974
+    },
+    {
+      "train_step": 6001,
+      "loss": 0.6193262780584052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6193262780584052
+    },
+    {
+      "train_step": 6002,
+      "loss": 0.60402654047127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.60402654047127
+    },
+    {
+      "train_step": 6003,
+      "loss": 0.6722067333891655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6722067333891655
+    },
+    {
+      "train_step": 6004,
+      "loss": 0.3566787944038934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3566787944038934
+    },
+    {
+      "train_step": 6005,
+      "loss": 0.6838836417292712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6838836417292712
+    },
+    {
+      "train_step": 6006,
+      "loss": 0.5623815327091536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5623815327091536
+    },
+    {
+      "train_step": 6007,
+      "loss": 0.7093655055415804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7093655055415804
+    },
+    {
+      "train_step": 6008,
+      "loss": 0.5962420857449647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962420857449647
+    },
+    {
+      "train_step": 6009,
+      "loss": 0.45801700331964623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45801700331964623
+    },
+    {
+      "train_step": 6010,
+      "loss": 0.4797199107969111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4797199107969111
+    },
+    {
+      "train_step": 6011,
+      "loss": 0.44685386174659225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44685386174659225
+    },
+    {
+      "train_step": 6012,
+      "loss": 0.6478476489234588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6478476489234588
+    },
+    {
+      "train_step": 6013,
+      "loss": 0.5793021899086965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5793021899086965
+    },
+    {
+      "train_step": 6014,
+      "loss": 0.4047821599563718,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4047821599563718
+    },
+    {
+      "train_step": 6015,
+      "loss": 0.4170061707770812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4170061707770812
+    },
+    {
+      "train_step": 6016,
+      "loss": 0.3851204854435401,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3851204854435401
+    },
+    {
+      "train_step": 6017,
+      "loss": 0.5623878651193326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5623878651193326
+    },
+    {
+      "train_step": 6018,
+      "loss": 0.43927642116363835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43927642116363835
+    },
+    {
+      "train_step": 6019,
+      "loss": 0.6371539472431539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6371539472431539
+    },
+    {
+      "train_step": 6020,
+      "loss": 0.5311727844863428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5311727844863428
+    },
+    {
+      "train_step": 6021,
+      "loss": 0.6033483632722142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6033483632722142
+    },
+    {
+      "train_step": 6022,
+      "loss": 0.5741637880983784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5741637880983784
+    },
+    {
+      "train_step": 6023,
+      "loss": 0.34302908836435764,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34302908836435764
+    },
+    {
+      "train_step": 6024,
+      "loss": 0.6215350610430496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6215350610430496
+    },
+    {
+      "train_step": 6025,
+      "loss": 0.3954960209706446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3954960209706446
+    },
+    {
+      "train_step": 6026,
+      "loss": 0.7024301475421263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7024301475421263
+    },
+    {
+      "train_step": 6027,
+      "loss": 0.5915462966666523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5915462966666523
+    },
+    {
+      "train_step": 6028,
+      "loss": 0.7224725524563544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7224725524563544
+    },
+    {
+      "train_step": 6029,
+      "loss": 0.49370617071068784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49370617071068784
+    },
+    {
+      "train_step": 6030,
+      "loss": 0.6062152504773111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6062152504773111
+    },
+    {
+      "train_step": 6031,
+      "loss": 0.465834777181395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.465834777181395
+    },
+    {
+      "train_step": 6032,
+      "loss": 0.44916746329744583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44916746329744583
+    },
+    {
+      "train_step": 6033,
+      "loss": 0.7377427592123482,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7377427592123482
+    },
+    {
+      "train_step": 6034,
+      "loss": 0.4291803745663049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4291803745663049
+    },
+    {
+      "train_step": 6035,
+      "loss": 0.49414493663190234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49414493663190234
+    },
+    {
+      "train_step": 6036,
+      "loss": 0.48559800057791125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48559800057791125
+    },
+    {
+      "train_step": 6037,
+      "loss": 0.4562565971177684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4562565971177684
+    },
+    {
+      "train_step": 6038,
+      "loss": 0.4656423142142988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4656423142142988
+    },
+    {
+      "train_step": 6039,
+      "loss": 0.3837180024295498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3837180024295498
+    },
+    {
+      "train_step": 6040,
+      "loss": 0.45497873630019464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45497873630019464
+    },
+    {
+      "train_step": 6041,
+      "loss": 0.6402721420291532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6402721420291532
+    },
+    {
+      "train_step": 6042,
+      "loss": 0.33390201259642066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33390201259642066
+    },
+    {
+      "train_step": 6043,
+      "loss": 0.4983530180661573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4983530180661573
+    },
+    {
+      "train_step": 6044,
+      "loss": 0.39369137982742564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39369137982742564
+    },
+    {
+      "train_step": 6045,
+      "loss": 0.5849869509528985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5849869509528985
+    },
+    {
+      "train_step": 6046,
+      "loss": 0.476344097231428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.476344097231428
+    },
+    {
+      "train_step": 6047,
+      "loss": 0.5459643006399384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5459643006399384
+    },
+    {
+      "train_step": 6048,
+      "loss": 0.6510871107052435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6510871107052435
+    },
+    {
+      "train_step": 6049,
+      "loss": 0.618625260299923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.618625260299923
+    },
+    {
+      "train_step": 6050,
+      "loss": 0.4737036602636596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4737036602636596
+    },
+    {
+      "train_step": 6051,
+      "loss": 0.5697468011329649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5697468011329649
+    },
+    {
+      "train_step": 6052,
+      "loss": 0.5235542004557221,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5235542004557221
+    },
+    {
+      "train_step": 6053,
+      "loss": 0.5797041871047941,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5797041871047941
+    },
+    {
+      "train_step": 6054,
+      "loss": 0.44297484849713886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44297484849713886
+    },
+    {
+      "train_step": 6055,
+      "loss": 0.4677131996980171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4677131996980171
+    },
+    {
+      "train_step": 6056,
+      "loss": 0.5165359699563283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5165359699563283
+    },
+    {
+      "train_step": 6057,
+      "loss": 0.4213052644432256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4213052644432256
+    },
+    {
+      "train_step": 6058,
+      "loss": 0.4955940214746683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4955940214746683
+    },
+    {
+      "train_step": 6059,
+      "loss": 0.4341124773365938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4341124773365938
+    },
+    {
+      "train_step": 6060,
+      "loss": 0.5263014994732357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5263014994732357
+    },
+    {
+      "train_step": 6061,
+      "loss": 0.4854079687712932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4854079687712932
+    },
+    {
+      "train_step": 6062,
+      "loss": 0.5140734271112595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5140734271112595
+    },
+    {
+      "train_step": 6063,
+      "loss": 0.6709354357824474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6709354357824474
+    },
+    {
+      "train_step": 6064,
+      "loss": 0.5898803219212385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5898803219212385
+    },
+    {
+      "train_step": 6065,
+      "loss": 0.47569549971441805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47569549971441805
+    },
+    {
+      "train_step": 6066,
+      "loss": 0.4672559341817152,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4672559341817152
+    },
+    {
+      "train_step": 6067,
+      "loss": 0.6933730245654414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6933730245654414
+    },
+    {
+      "train_step": 6068,
+      "loss": 0.6412072628738446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6412072628738446
+    },
+    {
+      "train_step": 6069,
+      "loss": 0.5972671536158721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5972671536158721
+    },
+    {
+      "train_step": 6070,
+      "loss": 0.6596492680501725,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6596492680501725
+    },
+    {
+      "train_step": 6071,
+      "loss": 0.4986324733437129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4986324733437129
+    },
+    {
+      "train_step": 6072,
+      "loss": 0.4734099860763215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4734099860763215
+    },
+    {
+      "train_step": 6073,
+      "loss": 0.44468552578835896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44468552578835896
+    },
+    {
+      "train_step": 6074,
+      "loss": 0.4789620495885538,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4789620495885538
+    },
+    {
+      "train_step": 6075,
+      "loss": 0.4302505993075709,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4302505993075709
+    },
+    {
+      "train_step": 6076,
+      "loss": 0.5133562947161625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5133562947161625
+    },
+    {
+      "train_step": 6077,
+      "loss": 0.5442328308585673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5442328308585673
+    },
+    {
+      "train_step": 6078,
+      "loss": 0.6938829419937642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6938829419937642
+    },
+    {
+      "train_step": 6079,
+      "loss": 0.43271835892015903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43271835892015903
+    },
+    {
+      "train_step": 6080,
+      "loss": 0.542659916163258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.542659916163258
+    },
+    {
+      "train_step": 6081,
+      "loss": 0.6303832213390114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6303832213390114
+    },
+    {
+      "train_step": 6082,
+      "loss": 0.4911261431568522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4911261431568522
+    },
+    {
+      "train_step": 6083,
+      "loss": 0.47444769524512515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47444769524512515
+    },
+    {
+      "train_step": 6084,
+      "loss": 0.5806101402568358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5806101402568358
+    },
+    {
+      "train_step": 6085,
+      "loss": 0.7141783673656589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7141783673656589
+    },
+    {
+      "train_step": 6086,
+      "loss": 0.7672374544397991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7672374544397991
+    },
+    {
+      "train_step": 6087,
+      "loss": 0.4122622265035851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4122622265035851
+    },
+    {
+      "train_step": 6088,
+      "loss": 0.49792730568423565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49792730568423565
+    },
+    {
+      "train_step": 6089,
+      "loss": 0.453110751975689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.453110751975689
+    },
+    {
+      "train_step": 6090,
+      "loss": 0.5187030986394481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5187030986394481
+    },
+    {
+      "train_step": 6091,
+      "loss": 0.5914622875098516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5914622875098516
+    },
+    {
+      "train_step": 6092,
+      "loss": 0.565096840374783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.565096840374783
+    },
+    {
+      "train_step": 6093,
+      "loss": 0.6906711570178892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6906711570178892
+    },
+    {
+      "train_step": 6094,
+      "loss": 0.4094547869192586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4094547869192586
+    },
+    {
+      "train_step": 6095,
+      "loss": 0.589191383572002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.589191383572002
+    },
+    {
+      "train_step": 6096,
+      "loss": 0.4454745465949411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4454745465949411
+    },
+    {
+      "train_step": 6097,
+      "loss": 0.6743556620584513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6743556620584513
+    },
+    {
+      "train_step": 6098,
+      "loss": 0.6048964844544148,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6048964844544148
+    },
+    {
+      "train_step": 6099,
+      "loss": 0.7897789519308904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7897789519308904
+    },
+    {
+      "train_step": 6100,
+      "loss": 0.48596354961955424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48596354961955424
+    },
+    {
+      "train_step": 6101,
+      "loss": 0.37247233341236907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37247233341236907
+    },
+    {
+      "train_step": 6102,
+      "loss": 0.6110576518325502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6110576518325502
+    },
+    {
+      "train_step": 6103,
+      "loss": 0.6134293910403733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6134293910403733
+    },
+    {
+      "train_step": 6104,
+      "loss": 0.460956993370191,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.460956993370191
+    },
+    {
+      "train_step": 6105,
+      "loss": 0.561890811529725,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.561890811529725
+    },
+    {
+      "train_step": 6106,
+      "loss": 0.8406416534254727,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8406416534254727
+    },
+    {
+      "train_step": 6107,
+      "loss": 0.4590883790709318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4590883790709318
+    },
+    {
+      "train_step": 6108,
+      "loss": 0.5287903879557427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5287903879557427
+    },
+    {
+      "train_step": 6109,
+      "loss": 0.6105633499157698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6105633499157698
+    },
+    {
+      "train_step": 6110,
+      "loss": 0.7527346533551112,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7527346533551112
+    },
+    {
+      "train_step": 6111,
+      "loss": 0.47156105210968663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47156105210968663
+    },
+    {
+      "train_step": 6112,
+      "loss": 0.6288260920524401,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6288260920524401
+    },
+    {
+      "train_step": 6113,
+      "loss": 0.6745241440021239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6745241440021239
+    },
+    {
+      "train_step": 6114,
+      "loss": 0.559571228348905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.559571228348905
+    },
+    {
+      "train_step": 6115,
+      "loss": 0.827088753330573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.827088753330573
+    },
+    {
+      "train_step": 6116,
+      "loss": 0.639151693923583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.639151693923583
+    },
+    {
+      "train_step": 6117,
+      "loss": 0.6060055788423395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6060055788423395
+    },
+    {
+      "train_step": 6118,
+      "loss": 0.7313597043297563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7313597043297563
+    },
+    {
+      "train_step": 6119,
+      "loss": 0.489505308031839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.489505308031839
+    },
+    {
+      "train_step": 6120,
+      "loss": 0.4323633023822682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4323633023822682
+    },
+    {
+      "train_step": 6121,
+      "loss": 0.39949948412479835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39949948412479835
+    },
+    {
+      "train_step": 6122,
+      "loss": 0.48217269306223626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48217269306223626
+    },
+    {
+      "train_step": 6123,
+      "loss": 0.6688398327291177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6688398327291177
+    },
+    {
+      "train_step": 6124,
+      "loss": 0.6383830279820093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6383830279820093
+    },
+    {
+      "train_step": 6125,
+      "loss": 0.5998413812184514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5998413812184514
+    },
+    {
+      "train_step": 6126,
+      "loss": 0.6212286671007418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6212286671007418
+    },
+    {
+      "train_step": 6127,
+      "loss": 0.42853777818539013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42853777818539013
+    },
+    {
+      "train_step": 6128,
+      "loss": 0.4951363485461498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4951363485461498
+    },
+    {
+      "train_step": 6129,
+      "loss": 0.6680918033189513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6680918033189513
+    },
+    {
+      "train_step": 6130,
+      "loss": 0.4091742566293355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4091742566293355
+    },
+    {
+      "train_step": 6131,
+      "loss": 0.6395655026218139,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6395655026218139
+    },
+    {
+      "train_step": 6132,
+      "loss": 0.42659146282850297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42659146282850297
+    },
+    {
+      "train_step": 6133,
+      "loss": 0.44694712566864947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44694712566864947
+    },
+    {
+      "train_step": 6134,
+      "loss": 0.45674026056156813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45674026056156813
+    },
+    {
+      "train_step": 6135,
+      "loss": 0.6611594441802808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6611594441802808
+    },
+    {
+      "train_step": 6136,
+      "loss": 0.5033348442162495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5033348442162495
+    },
+    {
+      "train_step": 6137,
+      "loss": 0.8159437377614023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8159437377614023
+    },
+    {
+      "train_step": 6138,
+      "loss": 0.7013408941337645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7013408941337645
+    },
+    {
+      "train_step": 6139,
+      "loss": 0.4676722586669305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4676722586669305
+    },
+    {
+      "train_step": 6140,
+      "loss": 0.5257823044126357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5257823044126357
+    },
+    {
+      "train_step": 6141,
+      "loss": 0.525570015480085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.525570015480085
+    },
+    {
+      "train_step": 6142,
+      "loss": 0.4333610515376283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4333610515376283
+    },
+    {
+      "train_step": 6143,
+      "loss": 0.6373746272833017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6373746272833017
+    },
+    {
+      "train_step": 6144,
+      "loss": 0.47469967379298084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47469967379298084
+    },
+    {
+      "train_step": 6145,
+      "loss": 0.36773308644434227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36773308644434227
+    },
+    {
+      "train_step": 6146,
+      "loss": 0.4473057515736243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4473057515736243
+    },
+    {
+      "train_step": 6147,
+      "loss": 0.7099100477216456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7099100477216456
+    },
+    {
+      "train_step": 6148,
+      "loss": 0.5848247629420575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5848247629420575
+    },
+    {
+      "train_step": 6149,
+      "loss": 0.5109490160955746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5109490160955746
+    },
+    {
+      "train_step": 6150,
+      "loss": 0.7145890210413477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7145890210413477
+    },
+    {
+      "train_step": 6151,
+      "loss": 0.6316737135240869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6316737135240869
+    },
+    {
+      "train_step": 6152,
+      "loss": 0.49228088941480025,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49228088941480025
+    },
+    {
+      "train_step": 6153,
+      "loss": 0.5907451821326672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5907451821326672
+    },
+    {
+      "train_step": 6154,
+      "loss": 0.6180065314888613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6180065314888613
+    },
+    {
+      "train_step": 6155,
+      "loss": 0.5284707618228704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5284707618228704
+    },
+    {
+      "train_step": 6156,
+      "loss": 0.6004030721925272,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6004030721925272
+    },
+    {
+      "train_step": 6157,
+      "loss": 0.4636298269442627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4636298269442627
+    },
+    {
+      "train_step": 6158,
+      "loss": 0.5443653689293938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5443653689293938
+    },
+    {
+      "train_step": 6159,
+      "loss": 0.5975527652023741,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5975527652023741
+    },
+    {
+      "train_step": 6160,
+      "loss": 0.4168354839072459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4168354839072459
+    },
+    {
+      "train_step": 6161,
+      "loss": 0.5966663661796566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5966663661796566
+    },
+    {
+      "train_step": 6162,
+      "loss": 0.6034249031619393,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6034249031619393
+    },
+    {
+      "train_step": 6163,
+      "loss": 0.40869025612560156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40869025612560156
+    },
+    {
+      "train_step": 6164,
+      "loss": 0.459694654515899,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.459694654515899
+    },
+    {
+      "train_step": 6165,
+      "loss": 0.335459911049184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.335459911049184
+    },
+    {
+      "train_step": 6166,
+      "loss": 0.45504622820770874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45504622820770874
+    },
+    {
+      "train_step": 6167,
+      "loss": 0.6400974980331513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6400974980331513
+    },
+    {
+      "train_step": 6168,
+      "loss": 0.85516631280299,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.85516631280299
+    },
+    {
+      "train_step": 6169,
+      "loss": 0.6748596814692194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6748596814692194
+    },
+    {
+      "train_step": 6170,
+      "loss": 0.480797880378241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.480797880378241
+    },
+    {
+      "train_step": 6171,
+      "loss": 0.4981856272233562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4981856272233562
+    },
+    {
+      "train_step": 6172,
+      "loss": 0.5067669161554791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5067669161554791
+    },
+    {
+      "train_step": 6173,
+      "loss": 0.4445018016290523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4445018016290523
+    },
+    {
+      "train_step": 6174,
+      "loss": 0.4901354981256288,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4901354981256288
+    },
+    {
+      "train_step": 6175,
+      "loss": 0.6776069890447343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6776069890447343
+    },
+    {
+      "train_step": 6176,
+      "loss": 0.6202346654297889,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6202346654297889
+    },
+    {
+      "train_step": 6177,
+      "loss": 0.5947839233740263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5947839233740263
+    },
+    {
+      "train_step": 6178,
+      "loss": 0.8631824312032761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8631824312032761
+    },
+    {
+      "train_step": 6179,
+      "loss": 0.5019801482550424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5019801482550424
+    },
+    {
+      "train_step": 6180,
+      "loss": 0.49519329205412954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49519329205412954
+    },
+    {
+      "train_step": 6181,
+      "loss": 0.5247409649354634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5247409649354634
+    },
+    {
+      "train_step": 6182,
+      "loss": 0.351589670957477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.351589670957477
+    },
+    {
+      "train_step": 6183,
+      "loss": 0.6302192710768495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6302192710768495
+    },
+    {
+      "train_step": 6184,
+      "loss": 0.7000940479581534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7000940479581534
+    },
+    {
+      "train_step": 6185,
+      "loss": 0.6905303656328905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6905303656328905
+    },
+    {
+      "train_step": 6186,
+      "loss": 0.6695188373380828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6695188373380828
+    },
+    {
+      "train_step": 6187,
+      "loss": 0.53074095820502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.53074095820502
+    },
+    {
+      "train_step": 6188,
+      "loss": 0.44913987664471516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44913987664471516
+    },
+    {
+      "train_step": 6189,
+      "loss": 0.7459248147405061,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7459248147405061
+    },
+    {
+      "train_step": 6190,
+      "loss": 0.627883582185684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.627883582185684
+    },
+    {
+      "train_step": 6191,
+      "loss": 0.5975993320908033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5975993320908033
+    },
+    {
+      "train_step": 6192,
+      "loss": 0.6622311710971477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6622311710971477
+    },
+    {
+      "train_step": 6193,
+      "loss": 0.35953850559016914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35953850559016914
+    },
+    {
+      "train_step": 6194,
+      "loss": 0.443202760596771,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.443202760596771
+    },
+    {
+      "train_step": 6195,
+      "loss": 0.4920799913568878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4920799913568878
+    },
+    {
+      "train_step": 6196,
+      "loss": 0.45184964539363554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45184964539363554
+    },
+    {
+      "train_step": 6197,
+      "loss": 0.5452131492168109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5452131492168109
+    },
+    {
+      "train_step": 6198,
+      "loss": 0.4827929912422545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4827929912422545
+    },
+    {
+      "train_step": 6199,
+      "loss": 0.42003798370116824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42003798370116824
+    },
+    {
+      "train_step": 6200,
+      "loss": 0.3837882388364988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3837882388364988
+    },
+    {
+      "train_step": 6201,
+      "loss": 0.47883002548027975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47883002548027975
+    },
+    {
+      "train_step": 6202,
+      "loss": 0.4023160937130243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4023160937130243
+    },
+    {
+      "train_step": 6203,
+      "loss": 0.5997975712279773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5997975712279773
+    },
+    {
+      "train_step": 6204,
+      "loss": 0.31794137905620995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31794137905620995
+    },
+    {
+      "train_step": 6205,
+      "loss": 0.7449308627551161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7449308627551161
+    },
+    {
+      "train_step": 6206,
+      "loss": 0.5304754505333491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5304754505333491
+    },
+    {
+      "train_step": 6207,
+      "loss": 0.5906597529764142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5906597529764142
+    },
+    {
+      "train_step": 6208,
+      "loss": 0.6188306279426108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6188306279426108
+    },
+    {
+      "train_step": 6209,
+      "loss": 0.5848440309290543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5848440309290543
+    },
+    {
+      "train_step": 6210,
+      "loss": 0.37280467705569836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37280467705569836
+    },
+    {
+      "train_step": 6211,
+      "loss": 0.43109162747927354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43109162747927354
+    },
+    {
+      "train_step": 6212,
+      "loss": 0.2770787416921131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2770787416921131
+    },
+    {
+      "train_step": 6213,
+      "loss": 0.5153058970929222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5153058970929222
+    },
+    {
+      "train_step": 6214,
+      "loss": 0.515426304399084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.515426304399084
+    },
+    {
+      "train_step": 6215,
+      "loss": 0.3690872773276009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3690872773276009
+    },
+    {
+      "train_step": 6216,
+      "loss": 0.4319388684909942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4319388684909942
+    },
+    {
+      "train_step": 6217,
+      "loss": 0.44672714362946375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44672714362946375
+    },
+    {
+      "train_step": 6218,
+      "loss": 0.731605387050209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.731605387050209
+    },
+    {
+      "train_step": 6219,
+      "loss": 0.5036471774594801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5036471774594801
+    },
+    {
+      "train_step": 6220,
+      "loss": 0.5713014716679142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5713014716679142
+    },
+    {
+      "train_step": 6221,
+      "loss": 0.6078759963774555,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6078759963774555
+    },
+    {
+      "train_step": 6222,
+      "loss": 0.5170528755942367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5170528755942367
+    },
+    {
+      "train_step": 6223,
+      "loss": 0.4506830967891475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4506830967891475
+    },
+    {
+      "train_step": 6224,
+      "loss": 0.425821980986134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.425821980986134
+    },
+    {
+      "train_step": 6225,
+      "loss": 0.5570799762386519,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5570799762386519
+    },
+    {
+      "train_step": 6226,
+      "loss": 0.5829088435447286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5829088435447286
+    },
+    {
+      "train_step": 6227,
+      "loss": 0.521329415062192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.521329415062192
+    },
+    {
+      "train_step": 6228,
+      "loss": 0.3146140336964057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3146140336964057
+    },
+    {
+      "train_step": 6229,
+      "loss": 0.4667833843553102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4667833843553102
+    },
+    {
+      "train_step": 6230,
+      "loss": 0.5226025240028253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5226025240028253
+    },
+    {
+      "train_step": 6231,
+      "loss": 0.42909111915171666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42909111915171666
+    },
+    {
+      "train_step": 6232,
+      "loss": 0.49785642762342863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49785642762342863
+    },
+    {
+      "train_step": 6233,
+      "loss": 0.5589444279017315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5589444279017315
+    },
+    {
+      "train_step": 6234,
+      "loss": 0.5631857748741879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5631857748741879
+    },
+    {
+      "train_step": 6235,
+      "loss": 0.5573133735652662,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5573133735652662
+    },
+    {
+      "train_step": 6236,
+      "loss": 0.4868342710770471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4868342710770471
+    },
+    {
+      "train_step": 6237,
+      "loss": 0.38909419560701214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38909419560701214
+    },
+    {
+      "train_step": 6238,
+      "loss": 0.531442011776789,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.531442011776789
+    },
+    {
+      "train_step": 6239,
+      "loss": 0.5405453193620181,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5405453193620181
+    },
+    {
+      "train_step": 6240,
+      "loss": 0.4681490569399965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4681490569399965
+    },
+    {
+      "train_step": 6241,
+      "loss": 0.6236698965929606,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6236698965929606
+    },
+    {
+      "train_step": 6242,
+      "loss": 0.7327717720033473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7327717720033473
+    },
+    {
+      "train_step": 6243,
+      "loss": 0.6719263278759442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6719263278759442
+    },
+    {
+      "train_step": 6244,
+      "loss": 0.6333144886361041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6333144886361041
+    },
+    {
+      "train_step": 6245,
+      "loss": 0.41783296600320446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41783296600320446
+    },
+    {
+      "train_step": 6246,
+      "loss": 0.571166599767095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.571166599767095
+    },
+    {
+      "train_step": 6247,
+      "loss": 0.5208914886237521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5208914886237521
+    },
+    {
+      "train_step": 6248,
+      "loss": 0.4858966336087235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4858966336087235
+    },
+    {
+      "train_step": 6249,
+      "loss": 0.4940253798551754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4940253798551754
+    },
+    {
+      "train_step": 6250,
+      "loss": 0.441833202285445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.441833202285445
+    },
+    {
+      "train_step": 6251,
+      "loss": 0.42190706215237395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42190706215237395
+    },
+    {
+      "train_step": 6252,
+      "loss": 0.599618221460716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.599618221460716
+    },
+    {
+      "train_step": 6253,
+      "loss": 0.39181395399991953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39181395399991953
+    },
+    {
+      "train_step": 6254,
+      "loss": 0.62628373506389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.62628373506389
+    },
+    {
+      "train_step": 6255,
+      "loss": 0.6155992200197454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6155992200197454
+    },
+    {
+      "train_step": 6256,
+      "loss": 0.38748523126374573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38748523126374573
+    },
+    {
+      "train_step": 6257,
+      "loss": 0.5715081182125058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5715081182125058
+    },
+    {
+      "train_step": 6258,
+      "loss": 0.44157672626586875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44157672626586875
+    },
+    {
+      "train_step": 6259,
+      "loss": 0.4585161526028714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4585161526028714
+    },
+    {
+      "train_step": 6260,
+      "loss": 0.6279944133889,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6279944133889
+    },
+    {
+      "train_step": 6261,
+      "loss": 0.6135783400266607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6135783400266607
+    },
+    {
+      "train_step": 6262,
+      "loss": 0.5339849717115808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5339849717115808
+    },
+    {
+      "train_step": 6263,
+      "loss": 0.40308732714206863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40308732714206863
+    },
+    {
+      "train_step": 6264,
+      "loss": 0.7927953681636417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7927953681636417
+    },
+    {
+      "train_step": 6265,
+      "loss": 0.7507277002484747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7507277002484747
+    },
+    {
+      "train_step": 6266,
+      "loss": 0.66260395880486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66260395880486
+    },
+    {
+      "train_step": 6267,
+      "loss": 0.6210434383529562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6210434383529562
+    },
+    {
+      "train_step": 6268,
+      "loss": 0.5711356906130864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5711356906130864
+    },
+    {
+      "train_step": 6269,
+      "loss": 0.5117045740929588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5117045740929588
+    },
+    {
+      "train_step": 6270,
+      "loss": 0.6505923477274729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6505923477274729
+    },
+    {
+      "train_step": 6271,
+      "loss": 0.5176125443463264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5176125443463264
+    },
+    {
+      "train_step": 6272,
+      "loss": 0.5521310031218399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5521310031218399
+    },
+    {
+      "train_step": 6273,
+      "loss": 0.5460986793797532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5460986793797532
+    },
+    {
+      "train_step": 6274,
+      "loss": 0.4804856187071027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4804856187071027
+    },
+    {
+      "train_step": 6275,
+      "loss": 0.5618790544752874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5618790544752874
+    },
+    {
+      "train_step": 6276,
+      "loss": 0.6020923416099282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6020923416099282
+    },
+    {
+      "train_step": 6277,
+      "loss": 0.4627917054816913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4627917054816913
+    },
+    {
+      "train_step": 6278,
+      "loss": 0.6147783352004859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6147783352004859
+    },
+    {
+      "train_step": 6279,
+      "loss": 0.5725875189916625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5725875189916625
+    },
+    {
+      "train_step": 6280,
+      "loss": 0.5150172330947512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5150172330947512
+    },
+    {
+      "train_step": 6281,
+      "loss": 0.47251587893000524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47251587893000524
+    },
+    {
+      "train_step": 6282,
+      "loss": 0.42697559646325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42697559646325
+    },
+    {
+      "train_step": 6283,
+      "loss": 0.6518651362835043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6518651362835043
+    },
+    {
+      "train_step": 6284,
+      "loss": 0.5121301770281251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5121301770281251
+    },
+    {
+      "train_step": 6285,
+      "loss": 0.5449869081229388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5449869081229388
+    },
+    {
+      "train_step": 6286,
+      "loss": 0.3760603437972871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3760603437972871
+    },
+    {
+      "train_step": 6287,
+      "loss": 0.3745407680560943,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3745407680560943
+    },
+    {
+      "train_step": 6288,
+      "loss": 0.40006499192056405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40006499192056405
+    },
+    {
+      "train_step": 6289,
+      "loss": 0.45608073377614966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45608073377614966
+    },
+    {
+      "train_step": 6290,
+      "loss": 0.6597268440237067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6597268440237067
+    },
+    {
+      "train_step": 6291,
+      "loss": 0.5918155061124645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5918155061124645
+    },
+    {
+      "train_step": 6292,
+      "loss": 0.6210487821520522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6210487821520522
+    },
+    {
+      "train_step": 6293,
+      "loss": 0.746622787272335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.746622787272335
+    },
+    {
+      "train_step": 6294,
+      "loss": 0.5976117430529237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5976117430529237
+    },
+    {
+      "train_step": 6295,
+      "loss": 0.6926046088217426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6926046088217426
+    },
+    {
+      "train_step": 6296,
+      "loss": 0.5267210519513239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5267210519513239
+    },
+    {
+      "train_step": 6297,
+      "loss": 0.533630809221911,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.533630809221911
+    },
+    {
+      "train_step": 6298,
+      "loss": 0.5284726412250209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5284726412250209
+    },
+    {
+      "train_step": 6299,
+      "loss": 0.7153831739000506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7153831739000506
+    },
+    {
+      "train_step": 6300,
+      "loss": 0.5973615300175051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5973615300175051
+    },
+    {
+      "train_step": 6301,
+      "loss": 0.5531650289835808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5531650289835808
+    },
+    {
+      "train_step": 6302,
+      "loss": 0.48306289928741597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48306289928741597
+    },
+    {
+      "train_step": 6303,
+      "loss": 0.5571433652580247,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5571433652580247
+    },
+    {
+      "train_step": 6304,
+      "loss": 0.46989932090129827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46989932090129827
+    },
+    {
+      "train_step": 6305,
+      "loss": 0.45693059292466603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45693059292466603
+    },
+    {
+      "train_step": 6306,
+      "loss": 0.32659576997674794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32659576997674794
+    },
+    {
+      "train_step": 6307,
+      "loss": 0.5896238715535603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5896238715535603
+    },
+    {
+      "train_step": 6308,
+      "loss": 0.46127454199334844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46127454199334844
+    },
+    {
+      "train_step": 6309,
+      "loss": 0.5862937057218435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5862937057218435
+    },
+    {
+      "train_step": 6310,
+      "loss": 0.6188288213856542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6188288213856542
+    },
+    {
+      "train_step": 6311,
+      "loss": 0.3857400691715743,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3857400691715743
+    },
+    {
+      "train_step": 6312,
+      "loss": 0.5694655817739601,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5694655817739601
+    },
+    {
+      "train_step": 6313,
+      "loss": 0.6485493014441986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6485493014441986
+    },
+    {
+      "train_step": 6314,
+      "loss": 0.6301077902686344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6301077902686344
+    },
+    {
+      "train_step": 6315,
+      "loss": 0.45640241048279523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45640241048279523
+    },
+    {
+      "train_step": 6316,
+      "loss": 0.6303871845076805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6303871845076805
+    },
+    {
+      "train_step": 6317,
+      "loss": 0.637611655040073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.637611655040073
+    },
+    {
+      "train_step": 6318,
+      "loss": 0.49058405914779385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49058405914779385
+    },
+    {
+      "train_step": 6319,
+      "loss": 0.45768693680153605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45768693680153605
+    },
+    {
+      "train_step": 6320,
+      "loss": 0.33321015837909146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33321015837909146
+    },
+    {
+      "train_step": 6321,
+      "loss": 0.5941363144773075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5941363144773075
+    },
+    {
+      "train_step": 6322,
+      "loss": 0.700658313281183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.700658313281183
+    },
+    {
+      "train_step": 6323,
+      "loss": 0.41083235440986493,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41083235440986493
+    },
+    {
+      "train_step": 6324,
+      "loss": 0.35374925260423556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35374925260423556
+    },
+    {
+      "train_step": 6325,
+      "loss": 0.535363049773132,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.535363049773132
+    },
+    {
+      "train_step": 6326,
+      "loss": 0.5531143891944118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5531143891944118
+    },
+    {
+      "train_step": 6327,
+      "loss": 0.31972260680872866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31972260680872866
+    },
+    {
+      "train_step": 6328,
+      "loss": 0.4639552752429511,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4639552752429511
+    },
+    {
+      "train_step": 6329,
+      "loss": 0.5965304631442563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5965304631442563
+    },
+    {
+      "train_step": 6330,
+      "loss": 0.49879743458895487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49879743458895487
+    },
+    {
+      "train_step": 6331,
+      "loss": 0.5817763832178895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5817763832178895
+    },
+    {
+      "train_step": 6332,
+      "loss": 0.7232516109139715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7232516109139715
+    },
+    {
+      "train_step": 6333,
+      "loss": 0.35390801208099076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35390801208099076
+    },
+    {
+      "train_step": 6334,
+      "loss": 0.46924069378246264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46924069378246264
+    },
+    {
+      "train_step": 6335,
+      "loss": 0.7157788574593837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7157788574593837
+    },
+    {
+      "train_step": 6336,
+      "loss": 0.5940612135039367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5940612135039367
+    },
+    {
+      "train_step": 6337,
+      "loss": 0.4506735007360888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4506735007360888
+    },
+    {
+      "train_step": 6338,
+      "loss": 0.704973145666402,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.704973145666402
+    },
+    {
+      "train_step": 6339,
+      "loss": 0.6420612677811672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6420612677811672
+    },
+    {
+      "train_step": 6340,
+      "loss": 0.6111164887278223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6111164887278223
+    },
+    {
+      "train_step": 6341,
+      "loss": 0.38651173333352384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38651173333352384
+    },
+    {
+      "train_step": 6342,
+      "loss": 0.532430941666259,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.532430941666259
+    },
+    {
+      "train_step": 6343,
+      "loss": 0.5687848627890069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5687848627890069
+    },
+    {
+      "train_step": 6344,
+      "loss": 0.6867557033877644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6867557033877644
+    },
+    {
+      "train_step": 6345,
+      "loss": 0.4426780034971295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4426780034971295
+    },
+    {
+      "train_step": 6346,
+      "loss": 0.5131841667272875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5131841667272875
+    },
+    {
+      "train_step": 6347,
+      "loss": 0.38645797573951673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38645797573951673
+    },
+    {
+      "train_step": 6348,
+      "loss": 0.8056647515891061,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8056647515891061
+    },
+    {
+      "train_step": 6349,
+      "loss": 0.6505183291735973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6505183291735973
+    },
+    {
+      "train_step": 6350,
+      "loss": 0.656976905424951,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.656976905424951
+    },
+    {
+      "train_step": 6351,
+      "loss": 0.610373481143646,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.610373481143646
+    },
+    {
+      "train_step": 6352,
+      "loss": 0.6336559062405317,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6336559062405317
+    },
+    {
+      "train_step": 6353,
+      "loss": 0.46953025565904544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46953025565904544
+    },
+    {
+      "train_step": 6354,
+      "loss": 0.44933974542113686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44933974542113686
+    },
+    {
+      "train_step": 6355,
+      "loss": 0.5917880251104332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5917880251104332
+    },
+    {
+      "train_step": 6356,
+      "loss": 0.4974731940528766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4974731940528766
+    },
+    {
+      "train_step": 6357,
+      "loss": 0.5898800951142018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5898800951142018
+    },
+    {
+      "train_step": 6358,
+      "loss": 0.5206677987477957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5206677987477957
+    },
+    {
+      "train_step": 6359,
+      "loss": 0.47055347150109567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47055347150109567
+    },
+    {
+      "train_step": 6360,
+      "loss": 0.6060392057525477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6060392057525477
+    },
+    {
+      "train_step": 6361,
+      "loss": 0.6657705222840586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6657705222840586
+    },
+    {
+      "train_step": 6362,
+      "loss": 0.5941595885283518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5941595885283518
+    },
+    {
+      "train_step": 6363,
+      "loss": 0.6557509643553833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6557509643553833
+    },
+    {
+      "train_step": 6364,
+      "loss": 0.7692723252506573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7692723252506573
+    },
+    {
+      "train_step": 6365,
+      "loss": 0.5600700414378508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5600700414378508
+    },
+    {
+      "train_step": 6366,
+      "loss": 0.5386324062354825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5386324062354825
+    },
+    {
+      "train_step": 6367,
+      "loss": 0.5103761667962734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5103761667962734
+    },
+    {
+      "train_step": 6368,
+      "loss": 0.4523665613620201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4523665613620201
+    },
+    {
+      "train_step": 6369,
+      "loss": 0.5389022751592261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5389022751592261
+    },
+    {
+      "train_step": 6370,
+      "loss": 0.662415830709588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.662415830709588
+    },
+    {
+      "train_step": 6371,
+      "loss": 0.7063220714224281,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7063220714224281
+    },
+    {
+      "train_step": 6372,
+      "loss": 0.5776917234751143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5776917234751143
+    },
+    {
+      "train_step": 6373,
+      "loss": 0.565198609979867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.565198609979867
+    },
+    {
+      "train_step": 6374,
+      "loss": 0.38310884057696293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38310884057696293
+    },
+    {
+      "train_step": 6375,
+      "loss": 0.47506621188877657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47506621188877657
+    },
+    {
+      "train_step": 6376,
+      "loss": 0.5806762589633212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5806762589633212
+    },
+    {
+      "train_step": 6377,
+      "loss": 0.5242834041630234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5242834041630234
+    },
+    {
+      "train_step": 6378,
+      "loss": 0.6288616585793827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6288616585793827
+    },
+    {
+      "train_step": 6379,
+      "loss": 0.39166286117496185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39166286117496185
+    },
+    {
+      "train_step": 6380,
+      "loss": 0.604289047547816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.604289047547816
+    },
+    {
+      "train_step": 6381,
+      "loss": 0.481621598302693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.481621598302693
+    },
+    {
+      "train_step": 6382,
+      "loss": 0.7091102570677051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7091102570677051
+    },
+    {
+      "train_step": 6383,
+      "loss": 0.5952029901868301,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5952029901868301
+    },
+    {
+      "train_step": 6384,
+      "loss": 0.572810060513717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.572810060513717
+    },
+    {
+      "train_step": 6385,
+      "loss": 0.4956843881755715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4956843881755715
+    },
+    {
+      "train_step": 6386,
+      "loss": 0.6510175643507173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6510175643507173
+    },
+    {
+      "train_step": 6387,
+      "loss": 0.5442031200496397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5442031200496397
+    },
+    {
+      "train_step": 6388,
+      "loss": 0.3148330078278221,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3148330078278221
+    },
+    {
+      "train_step": 6389,
+      "loss": 0.5865293310527937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5865293310527937
+    },
+    {
+      "train_step": 6390,
+      "loss": 0.5723864125845051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5723864125845051
+    },
+    {
+      "train_step": 6391,
+      "loss": 0.4194169358380275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4194169358380275
+    },
+    {
+      "train_step": 6392,
+      "loss": 0.48421926031702933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48421926031702933
+    },
+    {
+      "train_step": 6393,
+      "loss": 0.559644806854437,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.559644806854437
+    },
+    {
+      "train_step": 6394,
+      "loss": 0.5920501242402892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5920501242402892
+    },
+    {
+      "train_step": 6395,
+      "loss": 0.4998682919754765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4998682919754765
+    },
+    {
+      "train_step": 6396,
+      "loss": 0.7400183085763613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7400183085763613
+    },
+    {
+      "train_step": 6397,
+      "loss": 0.595086100582185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.595086100582185
+    },
+    {
+      "train_step": 6398,
+      "loss": 0.32196316410314824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32196316410314824
+    },
+    {
+      "train_step": 6399,
+      "loss": 0.4554422230361271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4554422230361271
+    },
+    {
+      "train_step": 6400,
+      "loss": 0.536428742262526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.536428742262526
+    },
+    {
+      "train_step": 6401,
+      "loss": 0.5022484537948346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5022484537948346
+    },
+    {
+      "train_step": 6402,
+      "loss": 0.7633337727914861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7633337727914861
+    },
+    {
+      "train_step": 6403,
+      "loss": 0.44254424601623055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44254424601623055
+    },
+    {
+      "train_step": 6404,
+      "loss": 0.5659677142209746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5659677142209746
+    },
+    {
+      "train_step": 6405,
+      "loss": 0.6275536245586892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6275536245586892
+    },
+    {
+      "train_step": 6406,
+      "loss": 0.4866745943052603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4866745943052603
+    },
+    {
+      "train_step": 6407,
+      "loss": 0.3128860713425441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3128860713425441
+    },
+    {
+      "train_step": 6408,
+      "loss": 0.5350433237451216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5350433237451216
+    },
+    {
+      "train_step": 6409,
+      "loss": 0.5797692116119589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5797692116119589
+    },
+    {
+      "train_step": 6410,
+      "loss": 0.7666545552520123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7666545552520123
+    },
+    {
+      "train_step": 6411,
+      "loss": 0.6119526463765713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6119526463765713
+    },
+    {
+      "train_step": 6412,
+      "loss": 0.3044786379924604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3044786379924604
+    },
+    {
+      "train_step": 6413,
+      "loss": 0.4549511867042184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4549511867042184
+    },
+    {
+      "train_step": 6414,
+      "loss": 0.5375951828038977,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5375951828038977
+    },
+    {
+      "train_step": 6415,
+      "loss": 0.5379304873599635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5379304873599635
+    },
+    {
+      "train_step": 6416,
+      "loss": 0.38200792532917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38200792532917
+    },
+    {
+      "train_step": 6417,
+      "loss": 0.7190757538874126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7190757538874126
+    },
+    {
+      "train_step": 6418,
+      "loss": 0.4964878810350561,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4964878810350561
+    },
+    {
+      "train_step": 6419,
+      "loss": 0.7794745685845591,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7794745685845591
+    },
+    {
+      "train_step": 6420,
+      "loss": 0.6106361100776132,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6106361100776132
+    },
+    {
+      "train_step": 6421,
+      "loss": 0.45375318701498624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45375318701498624
+    },
+    {
+      "train_step": 6422,
+      "loss": 0.6665326631336181,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6665326631336181
+    },
+    {
+      "train_step": 6423,
+      "loss": 0.5619263273329388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5619263273329388
+    },
+    {
+      "train_step": 6424,
+      "loss": 0.6425054967374213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6425054967374213
+    },
+    {
+      "train_step": 6425,
+      "loss": 0.5860747685974184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5860747685974184
+    },
+    {
+      "train_step": 6426,
+      "loss": 0.5410488748029852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5410488748029852
+    },
+    {
+      "train_step": 6427,
+      "loss": 0.49241979863051666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49241979863051666
+    },
+    {
+      "train_step": 6428,
+      "loss": 0.5953632943232815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5953632943232815
+    },
+    {
+      "train_step": 6429,
+      "loss": 0.556805198770024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.556805198770024
+    },
+    {
+      "train_step": 6430,
+      "loss": 0.5010347092793805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5010347092793805
+    },
+    {
+      "train_step": 6431,
+      "loss": 0.6826978454919936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6826978454919936
+    },
+    {
+      "train_step": 6432,
+      "loss": 0.46716652716737034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46716652716737034
+    },
+    {
+      "train_step": 6433,
+      "loss": 0.43484585542003334,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43484585542003334
+    },
+    {
+      "train_step": 6434,
+      "loss": 0.7918240493381463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7918240493381463
+    },
+    {
+      "train_step": 6435,
+      "loss": 0.4578755493775861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4578755493775861
+    },
+    {
+      "train_step": 6436,
+      "loss": 0.4410409899525422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4410409899525422
+    },
+    {
+      "train_step": 6437,
+      "loss": 0.7382302008292361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7382302008292361
+    },
+    {
+      "train_step": 6438,
+      "loss": 0.5789440916581629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5789440916581629
+    },
+    {
+      "train_step": 6439,
+      "loss": 0.5306663207201349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5306663207201349
+    },
+    {
+      "train_step": 6440,
+      "loss": 0.44183311180160817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44183311180160817
+    },
+    {
+      "train_step": 6441,
+      "loss": 0.39955383954213713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39955383954213713
+    },
+    {
+      "train_step": 6442,
+      "loss": 0.46919956688369574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46919956688369574
+    },
+    {
+      "train_step": 6443,
+      "loss": 0.46744049030782353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46744049030782353
+    },
+    {
+      "train_step": 6444,
+      "loss": 0.4525106766335757,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4525106766335757
+    },
+    {
+      "train_step": 6445,
+      "loss": 0.6823327235358759,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6823327235358759
+    },
+    {
+      "train_step": 6446,
+      "loss": 0.7052601454537278,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7052601454537278
+    },
+    {
+      "train_step": 6447,
+      "loss": 0.6756587904885617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6756587904885617
+    },
+    {
+      "train_step": 6448,
+      "loss": 0.7537382808632123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7537382808632123
+    },
+    {
+      "train_step": 6449,
+      "loss": 0.5933112646024175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5933112646024175
+    },
+    {
+      "train_step": 6450,
+      "loss": 0.5121669034082921,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5121669034082921
+    },
+    {
+      "train_step": 6451,
+      "loss": 0.29799631112803404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.29799631112803404
+    },
+    {
+      "train_step": 6452,
+      "loss": 0.5800839895308603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5800839895308603
+    },
+    {
+      "train_step": 6453,
+      "loss": 0.5103153756767271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5103153756767271
+    },
+    {
+      "train_step": 6454,
+      "loss": 0.7355699634910975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7355699634910975
+    },
+    {
+      "train_step": 6455,
+      "loss": 0.35485346841102516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35485346841102516
+    },
+    {
+      "train_step": 6456,
+      "loss": 0.516414970360792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.516414970360792
+    },
+    {
+      "train_step": 6457,
+      "loss": 0.4236058704742498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4236058704742498
+    },
+    {
+      "train_step": 6458,
+      "loss": 0.6436416666962813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6436416666962813
+    },
+    {
+      "train_step": 6459,
+      "loss": 0.5699698035165842,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5699698035165842
+    },
+    {
+      "train_step": 6460,
+      "loss": 0.561151618985891,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.561151618985891
+    },
+    {
+      "train_step": 6461,
+      "loss": 0.45996434977045797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45996434977045797
+    },
+    {
+      "train_step": 6462,
+      "loss": 0.4051863196124477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4051863196124477
+    },
+    {
+      "train_step": 6463,
+      "loss": 0.50680674921131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.50680674921131
+    },
+    {
+      "train_step": 6464,
+      "loss": 0.5380676638790521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5380676638790521
+    },
+    {
+      "train_step": 6465,
+      "loss": 0.6321607997767049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6321607997767049
+    },
+    {
+      "train_step": 6466,
+      "loss": 0.8129552447371452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8129552447371452
+    },
+    {
+      "train_step": 6467,
+      "loss": 0.4357272415109838,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4357272415109838
+    },
+    {
+      "train_step": 6468,
+      "loss": 0.417213040937684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.417213040937684
+    },
+    {
+      "train_step": 6469,
+      "loss": 0.6424330776508813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6424330776508813
+    },
+    {
+      "train_step": 6470,
+      "loss": 0.5249803957901732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5249803957901732
+    },
+    {
+      "train_step": 6471,
+      "loss": 0.5711803668092839,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5711803668092839
+    },
+    {
+      "train_step": 6472,
+      "loss": 0.5331668711225934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5331668711225934
+    },
+    {
+      "train_step": 6473,
+      "loss": 0.5934944168619137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5934944168619137
+    },
+    {
+      "train_step": 6474,
+      "loss": 0.505289245056284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.505289245056284
+    },
+    {
+      "train_step": 6475,
+      "loss": 0.3751908659607307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3751908659607307
+    },
+    {
+      "train_step": 6476,
+      "loss": 0.41178684376528824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41178684376528824
+    },
+    {
+      "train_step": 6477,
+      "loss": 0.7242239876701965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7242239876701965
+    },
+    {
+      "train_step": 6478,
+      "loss": 0.6562655606493543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6562655606493543
+    },
+    {
+      "train_step": 6479,
+      "loss": 0.6126412312594647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6126412312594647
+    },
+    {
+      "train_step": 6480,
+      "loss": 0.5165915975376018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5165915975376018
+    },
+    {
+      "train_step": 6481,
+      "loss": 0.46689765167046093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46689765167046093
+    },
+    {
+      "train_step": 6482,
+      "loss": 0.7219535470148384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7219535470148384
+    },
+    {
+      "train_step": 6483,
+      "loss": 0.7167327137382851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7167327137382851
+    },
+    {
+      "train_step": 6484,
+      "loss": 0.5779380757546634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5779380757546634
+    },
+    {
+      "train_step": 6485,
+      "loss": 0.5340151443585611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5340151443585611
+    },
+    {
+      "train_step": 6486,
+      "loss": 0.5678653852356899,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5678653852356899
+    },
+    {
+      "train_step": 6487,
+      "loss": 0.35715517897881416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35715517897881416
+    },
+    {
+      "train_step": 6488,
+      "loss": 0.3570895655898996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3570895655898996
+    },
+    {
+      "train_step": 6489,
+      "loss": 0.316960402101821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.316960402101821
+    },
+    {
+      "train_step": 6490,
+      "loss": 0.5280868829102988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5280868829102988
+    },
+    {
+      "train_step": 6491,
+      "loss": 0.45036226968635157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45036226968635157
+    },
+    {
+      "train_step": 6492,
+      "loss": 0.552017218107717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.552017218107717
+    },
+    {
+      "train_step": 6493,
+      "loss": 0.8147230825490929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8147230825490929
+    },
+    {
+      "train_step": 6494,
+      "loss": 0.4077652490282423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4077652490282423
+    },
+    {
+      "train_step": 6495,
+      "loss": 0.502496048150918,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.502496048150918
+    },
+    {
+      "train_step": 6496,
+      "loss": 0.6329483299260292,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6329483299260292
+    },
+    {
+      "train_step": 6497,
+      "loss": 0.7703570070644227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7703570070644227
+    },
+    {
+      "train_step": 6498,
+      "loss": 0.5882602896047285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5882602896047285
+    },
+    {
+      "train_step": 6499,
+      "loss": 0.5097510483890767,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5097510483890767
+    },
+    {
+      "train_step": 6500,
+      "loss": 0.6161661264586098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6161661264586098
+    },
+    {
+      "train_step": 6501,
+      "loss": 0.6074604212039255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6074604212039255
+    },
+    {
+      "train_step": 6502,
+      "loss": 0.42956685869508227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42956685869508227
+    },
+    {
+      "train_step": 6503,
+      "loss": 0.5589100734456881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5589100734456881
+    },
+    {
+      "train_step": 6504,
+      "loss": 0.4108023025186268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4108023025186268
+    },
+    {
+      "train_step": 6505,
+      "loss": 0.5227930075992683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5227930075992683
+    },
+    {
+      "train_step": 6506,
+      "loss": 0.33458742747152975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33458742747152975
+    },
+    {
+      "train_step": 6507,
+      "loss": 0.4737517679626327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4737517679626327
+    },
+    {
+      "train_step": 6508,
+      "loss": 0.6942883626377307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6942883626377307
+    },
+    {
+      "train_step": 6509,
+      "loss": 0.7159017244597506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7159017244597506
+    },
+    {
+      "train_step": 6510,
+      "loss": 0.7056911604173088,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7056911604173088
+    },
+    {
+      "train_step": 6511,
+      "loss": 0.6627764706872127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6627764706872127
+    },
+    {
+      "train_step": 6512,
+      "loss": 0.4471671175281414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4471671175281414
+    },
+    {
+      "train_step": 6513,
+      "loss": 0.4684702803855528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4684702803855528
+    },
+    {
+      "train_step": 6514,
+      "loss": 0.5381775874248107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5381775874248107
+    },
+    {
+      "train_step": 6515,
+      "loss": 0.5358513538907251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5358513538907251
+    },
+    {
+      "train_step": 6516,
+      "loss": 0.7199785826118984,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7199785826118984
+    },
+    {
+      "train_step": 6517,
+      "loss": 0.28851200570505997,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.28851200570505997
+    },
+    {
+      "train_step": 6518,
+      "loss": 0.49822845431228846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49822845431228846
+    },
+    {
+      "train_step": 6519,
+      "loss": 0.6366912097829311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6366912097829311
+    },
+    {
+      "train_step": 6520,
+      "loss": 0.6572174263095107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6572174263095107
+    },
+    {
+      "train_step": 6521,
+      "loss": 0.4712310864355793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4712310864355793
+    },
+    {
+      "train_step": 6522,
+      "loss": 0.3939033029150798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3939033029150798
+    },
+    {
+      "train_step": 6523,
+      "loss": 0.37768898253589367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37768898253589367
+    },
+    {
+      "train_step": 6524,
+      "loss": 0.35885086652121984,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35885086652121984
+    },
+    {
+      "train_step": 6525,
+      "loss": 0.437108463807985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.437108463807985
+    },
+    {
+      "train_step": 6526,
+      "loss": 0.48084047213774656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48084047213774656
+    },
+    {
+      "train_step": 6527,
+      "loss": 0.5687056592547306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5687056592547306
+    },
+    {
+      "train_step": 6528,
+      "loss": 0.6641571686610154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6641571686610154
+    },
+    {
+      "train_step": 6529,
+      "loss": 0.36087156024166217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36087156024166217
+    },
+    {
+      "train_step": 6530,
+      "loss": 0.5785798730153903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5785798730153903
+    },
+    {
+      "train_step": 6531,
+      "loss": 0.48911622595621207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48911622595621207
+    },
+    {
+      "train_step": 6532,
+      "loss": 0.32738115191419265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32738115191419265
+    },
+    {
+      "train_step": 6533,
+      "loss": 0.5743526546241382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5743526546241382
+    },
+    {
+      "train_step": 6534,
+      "loss": 0.47585482072656093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47585482072656093
+    },
+    {
+      "train_step": 6535,
+      "loss": 0.7465416150688087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7465416150688087
+    },
+    {
+      "train_step": 6536,
+      "loss": 0.2868790597049787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2868790597049787
+    },
+    {
+      "train_step": 6537,
+      "loss": 0.5758151125360509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5758151125360509
+    },
+    {
+      "train_step": 6538,
+      "loss": 0.679167483237399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.679167483237399
+    },
+    {
+      "train_step": 6539,
+      "loss": 0.8373809474501285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8373809474501285
+    },
+    {
+      "train_step": 6540,
+      "loss": 0.6573535974750645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6573535974750645
+    },
+    {
+      "train_step": 6541,
+      "loss": 0.460564828242929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.460564828242929
+    },
+    {
+      "train_step": 6542,
+      "loss": 0.5972827850764403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5972827850764403
+    },
+    {
+      "train_step": 6543,
+      "loss": 0.3997976315818153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3997976315818153
+    },
+    {
+      "train_step": 6544,
+      "loss": 0.5026648268052225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5026648268052225
+    },
+    {
+      "train_step": 6545,
+      "loss": 0.42355334798645733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42355334798645733
+    },
+    {
+      "train_step": 6546,
+      "loss": 0.4554914657478121,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4554914657478121
+    },
+    {
+      "train_step": 6547,
+      "loss": 0.42857840364993693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42857840364993693
+    },
+    {
+      "train_step": 6548,
+      "loss": 0.7578845679601124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7578845679601124
+    },
+    {
+      "train_step": 6549,
+      "loss": 0.7310459379886582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7310459379886582
+    },
+    {
+      "train_step": 6550,
+      "loss": 0.4904576904234331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4904576904234331
+    },
+    {
+      "train_step": 6551,
+      "loss": 0.6484937539317521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6484937539317521
+    },
+    {
+      "train_step": 6552,
+      "loss": 0.6211831166150735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6211831166150735
+    },
+    {
+      "train_step": 6553,
+      "loss": 0.523953810565417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.523953810565417
+    },
+    {
+      "train_step": 6554,
+      "loss": 0.6810039073967094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6810039073967094
+    },
+    {
+      "train_step": 6555,
+      "loss": 0.6408056102470743,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6408056102470743
+    },
+    {
+      "train_step": 6556,
+      "loss": 0.8420241263487667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8420241263487667
+    },
+    {
+      "train_step": 6557,
+      "loss": 0.6608551656487565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6608551656487565
+    },
+    {
+      "train_step": 6558,
+      "loss": 0.5495979497896194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5495979497896194
+    },
+    {
+      "train_step": 6559,
+      "loss": 0.5781252848486834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5781252848486834
+    },
+    {
+      "train_step": 6560,
+      "loss": 0.5119291420926501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5119291420926501
+    },
+    {
+      "train_step": 6561,
+      "loss": 0.682001652437596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.682001652437596
+    },
+    {
+      "train_step": 6562,
+      "loss": 0.5532574907571887,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5532574907571887
+    },
+    {
+      "train_step": 6563,
+      "loss": 0.4666981636649916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4666981636649916
+    },
+    {
+      "train_step": 6564,
+      "loss": 0.42586225135989547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42586225135989547
+    },
+    {
+      "train_step": 6565,
+      "loss": 0.5035039491112646,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035039491112646
+    },
+    {
+      "train_step": 6566,
+      "loss": 0.5838169123579117,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5838169123579117
+    },
+    {
+      "train_step": 6567,
+      "loss": 0.4725978337011175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4725978337011175
+    },
+    {
+      "train_step": 6568,
+      "loss": 0.5074390378615715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5074390378615715
+    },
+    {
+      "train_step": 6569,
+      "loss": 0.4735046470084596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4735046470084596
+    },
+    {
+      "train_step": 6570,
+      "loss": 0.5341146596866507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5341146596866507
+    },
+    {
+      "train_step": 6571,
+      "loss": 0.38240186727948033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38240186727948033
+    },
+    {
+      "train_step": 6572,
+      "loss": 0.5553259081253565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5553259081253565
+    },
+    {
+      "train_step": 6573,
+      "loss": 0.4047081379702784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4047081379702784
+    },
+    {
+      "train_step": 6574,
+      "loss": 0.6055600778717424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6055600778717424
+    },
+    {
+      "train_step": 6575,
+      "loss": 0.5018389782404394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5018389782404394
+    },
+    {
+      "train_step": 6576,
+      "loss": 0.567095342428791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.567095342428791
+    },
+    {
+      "train_step": 6577,
+      "loss": 0.46586102270429375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46586102270429375
+    },
+    {
+      "train_step": 6578,
+      "loss": 0.57741293919151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.57741293919151
+    },
+    {
+      "train_step": 6579,
+      "loss": 0.6319772923593701,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6319772923593701
+    },
+    {
+      "train_step": 6580,
+      "loss": 0.5161704093244435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5161704093244435
+    },
+    {
+      "train_step": 6581,
+      "loss": 0.5871212780022501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5871212780022501
+    },
+    {
+      "train_step": 6582,
+      "loss": 0.43226994401362095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43226994401362095
+    },
+    {
+      "train_step": 6583,
+      "loss": 0.5153394787903071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5153394787903071
+    },
+    {
+      "train_step": 6584,
+      "loss": 0.5135288016813546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5135288016813546
+    },
+    {
+      "train_step": 6585,
+      "loss": 0.5841195595532095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5841195595532095
+    },
+    {
+      "train_step": 6586,
+      "loss": 0.5323840048374949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5323840048374949
+    },
+    {
+      "train_step": 6587,
+      "loss": 0.5635578719456803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5635578719456803
+    },
+    {
+      "train_step": 6588,
+      "loss": 0.7416490018328932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7416490018328932
+    },
+    {
+      "train_step": 6589,
+      "loss": 0.5848964574711958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5848964574711958
+    },
+    {
+      "train_step": 6590,
+      "loss": 0.579168320508073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.579168320508073
+    },
+    {
+      "train_step": 6591,
+      "loss": 0.45742176677727425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45742176677727425
+    },
+    {
+      "train_step": 6592,
+      "loss": 0.4333750508966854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4333750508966854
+    },
+    {
+      "train_step": 6593,
+      "loss": 0.5303675648555569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5303675648555569
+    },
+    {
+      "train_step": 6594,
+      "loss": 0.5222801340482339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5222801340482339
+    },
+    {
+      "train_step": 6595,
+      "loss": 0.4529305934838363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4529305934838363
+    },
+    {
+      "train_step": 6596,
+      "loss": 0.5342567787619504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5342567787619504
+    },
+    {
+      "train_step": 6597,
+      "loss": 0.3640477243469678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3640477243469678
+    },
+    {
+      "train_step": 6598,
+      "loss": 0.6296354128185817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6296354128185817
+    },
+    {
+      "train_step": 6599,
+      "loss": 0.48345188496433855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48345188496433855
+    },
+    {
+      "train_step": 6600,
+      "loss": 0.49797211071001696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49797211071001696
+    },
+    {
+      "train_step": 6601,
+      "loss": 0.5087026571547991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5087026571547991
+    },
+    {
+      "train_step": 6602,
+      "loss": 0.7825232338467949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7825232338467949
+    },
+    {
+      "train_step": 6603,
+      "loss": 0.6256524254166098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6256524254166098
+    },
+    {
+      "train_step": 6604,
+      "loss": 0.3925670336617089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3925670336617089
+    },
+    {
+      "train_step": 6605,
+      "loss": 0.6905209104067171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6905209104067171
+    },
+    {
+      "train_step": 6606,
+      "loss": 0.3704139115564517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3704139115564517
+    },
+    {
+      "train_step": 6607,
+      "loss": 0.6888274470901686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6888274470901686
+    },
+    {
+      "train_step": 6608,
+      "loss": 0.4743929973120784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4743929973120784
+    },
+    {
+      "train_step": 6609,
+      "loss": 0.4138311511799571,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4138311511799571
+    },
+    {
+      "train_step": 6610,
+      "loss": 0.5033799574041564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5033799574041564
+    },
+    {
+      "train_step": 6611,
+      "loss": 0.49389383232719203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49389383232719203
+    },
+    {
+      "train_step": 6612,
+      "loss": 0.49221213600202374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49221213600202374
+    },
+    {
+      "train_step": 6613,
+      "loss": 0.49538684762078283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49538684762078283
+    },
+    {
+      "train_step": 6614,
+      "loss": 0.5904468878943847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5904468878943847
+    },
+    {
+      "train_step": 6615,
+      "loss": 0.6562335955501676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6562335955501676
+    },
+    {
+      "train_step": 6616,
+      "loss": 0.6649834097280591,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6649834097280591
+    },
+    {
+      "train_step": 6617,
+      "loss": 0.5974967615608523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5974967615608523
+    },
+    {
+      "train_step": 6618,
+      "loss": 0.5775113975846079,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5775113975846079
+    },
+    {
+      "train_step": 6619,
+      "loss": 0.4566001426380344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4566001426380344
+    },
+    {
+      "train_step": 6620,
+      "loss": 0.4210232571282924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4210232571282924
+    },
+    {
+      "train_step": 6621,
+      "loss": 0.46583428715378516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46583428715378516
+    },
+    {
+      "train_step": 6622,
+      "loss": 0.7304973395217804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7304973395217804
+    },
+    {
+      "train_step": 6623,
+      "loss": 0.4686236498520725,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4686236498520725
+    },
+    {
+      "train_step": 6624,
+      "loss": 0.710201328082018,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.710201328082018
+    },
+    {
+      "train_step": 6625,
+      "loss": 0.5406976377145655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5406976377145655
+    },
+    {
+      "train_step": 6626,
+      "loss": 0.5726210473217951,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5726210473217951
+    },
+    {
+      "train_step": 6627,
+      "loss": 0.6300377992763425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6300377992763425
+    },
+    {
+      "train_step": 6628,
+      "loss": 0.4015810581561053,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4015810581561053
+    },
+    {
+      "train_step": 6629,
+      "loss": 0.6662327282796996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6662327282796996
+    },
+    {
+      "train_step": 6630,
+      "loss": 0.460027213715093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.460027213715093
+    },
+    {
+      "train_step": 6631,
+      "loss": 0.45355817463807124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45355817463807124
+    },
+    {
+      "train_step": 6632,
+      "loss": 0.4856030334527454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4856030334527454
+    },
+    {
+      "train_step": 6633,
+      "loss": 0.46270588496552,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46270588496552
+    },
+    {
+      "train_step": 6634,
+      "loss": 0.7991114735158114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7991114735158114
+    },
+    {
+      "train_step": 6635,
+      "loss": 0.5204082806909333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5204082806909333
+    },
+    {
+      "train_step": 6636,
+      "loss": 0.5490578206731984,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5490578206731984
+    },
+    {
+      "train_step": 6637,
+      "loss": 0.5355738348435586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5355738348435586
+    },
+    {
+      "train_step": 6638,
+      "loss": 0.5254229648800748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5254229648800748
+    },
+    {
+      "train_step": 6639,
+      "loss": 0.5039936392459239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5039936392459239
+    },
+    {
+      "train_step": 6640,
+      "loss": 0.8270005511990907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8270005511990907
+    },
+    {
+      "train_step": 6641,
+      "loss": 0.4591565884222668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4591565884222668
+    },
+    {
+      "train_step": 6642,
+      "loss": 0.5566043499885076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5566043499885076
+    },
+    {
+      "train_step": 6643,
+      "loss": 0.6506595876771166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6506595876771166
+    },
+    {
+      "train_step": 6644,
+      "loss": 0.5163735387142789,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5163735387142789
+    },
+    {
+      "train_step": 6645,
+      "loss": 0.4517756042225644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4517756042225644
+    },
+    {
+      "train_step": 6646,
+      "loss": 0.6635037856954441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6635037856954441
+    },
+    {
+      "train_step": 6647,
+      "loss": 0.5169615852657604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5169615852657604
+    },
+    {
+      "train_step": 6648,
+      "loss": 0.40803806437670526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40803806437670526
+    },
+    {
+      "train_step": 6649,
+      "loss": 0.6250852860935869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6250852860935869
+    },
+    {
+      "train_step": 6650,
+      "loss": 0.4439670862400602,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4439670862400602
+    },
+    {
+      "train_step": 6651,
+      "loss": 0.5253011046771123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5253011046771123
+    },
+    {
+      "train_step": 6652,
+      "loss": 0.5076346810941833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5076346810941833
+    },
+    {
+      "train_step": 6653,
+      "loss": 0.6647418232392304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6647418232392304
+    },
+    {
+      "train_step": 6654,
+      "loss": 0.6029031310500174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6029031310500174
+    },
+    {
+      "train_step": 6655,
+      "loss": 0.492128305870029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.492128305870029
+    },
+    {
+      "train_step": 6656,
+      "loss": 0.47284168402156573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47284168402156573
+    },
+    {
+      "train_step": 6657,
+      "loss": 0.4753628547392652,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4753628547392652
+    },
+    {
+      "train_step": 6658,
+      "loss": 0.473935123461642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.473935123461642
+    },
+    {
+      "train_step": 6659,
+      "loss": 0.4375347437070814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4375347437070814
+    },
+    {
+      "train_step": 6660,
+      "loss": 0.48532328272200703,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48532328272200703
+    },
+    {
+      "train_step": 6661,
+      "loss": 0.44093979918981985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44093979918981985
+    },
+    {
+      "train_step": 6662,
+      "loss": 0.5276703535212647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5276703535212647
+    },
+    {
+      "train_step": 6663,
+      "loss": 0.400001769964834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.400001769964834
+    },
+    {
+      "train_step": 6664,
+      "loss": 0.4176559316358916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4176559316358916
+    },
+    {
+      "train_step": 6665,
+      "loss": 0.6702423709928301,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6702423709928301
+    },
+    {
+      "train_step": 6666,
+      "loss": 0.5114966604984668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5114966604984668
+    },
+    {
+      "train_step": 6667,
+      "loss": 0.5022411768581788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5022411768581788
+    },
+    {
+      "train_step": 6668,
+      "loss": 0.46645769787271557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46645769787271557
+    },
+    {
+      "train_step": 6669,
+      "loss": 0.5178494709575723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5178494709575723
+    },
+    {
+      "train_step": 6670,
+      "loss": 0.39371242856872923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39371242856872923
+    },
+    {
+      "train_step": 6671,
+      "loss": 0.3885406222106702,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3885406222106702
+    },
+    {
+      "train_step": 6672,
+      "loss": 0.6410628044564721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6410628044564721
+    },
+    {
+      "train_step": 6673,
+      "loss": 0.6475297162356832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6475297162356832
+    },
+    {
+      "train_step": 6674,
+      "loss": 0.5585626966740861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5585626966740861
+    },
+    {
+      "train_step": 6675,
+      "loss": 0.5238860143033095,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5238860143033095
+    },
+    {
+      "train_step": 6676,
+      "loss": 0.5736175231899511,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5736175231899511
+    },
+    {
+      "train_step": 6677,
+      "loss": 0.5995483671347624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5995483671347624
+    },
+    {
+      "train_step": 6678,
+      "loss": 0.5590418963120272,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5590418963120272
+    },
+    {
+      "train_step": 6679,
+      "loss": 0.5138531986547821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5138531986547821
+    },
+    {
+      "train_step": 6680,
+      "loss": 0.4400285531237066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4400285531237066
+    },
+    {
+      "train_step": 6681,
+      "loss": 0.5214259635164606,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5214259635164606
+    },
+    {
+      "train_step": 6682,
+      "loss": 0.5846008657725336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5846008657725336
+    },
+    {
+      "train_step": 6683,
+      "loss": 0.5344057575357496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5344057575357496
+    },
+    {
+      "train_step": 6684,
+      "loss": 0.445840328234801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.445840328234801
+    },
+    {
+      "train_step": 6685,
+      "loss": 0.5399847198640061,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5399847198640061
+    },
+    {
+      "train_step": 6686,
+      "loss": 0.5967990818549374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5967990818549374
+    },
+    {
+      "train_step": 6687,
+      "loss": 0.43878160397839017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43878160397839017
+    },
+    {
+      "train_step": 6688,
+      "loss": 0.7140210799271297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7140210799271297
+    },
+    {
+      "train_step": 6689,
+      "loss": 0.4760908227549624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4760908227549624
+    },
+    {
+      "train_step": 6690,
+      "loss": 0.4187721267516506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4187721267516506
+    },
+    {
+      "train_step": 6691,
+      "loss": 0.5493243573230574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5493243573230574
+    },
+    {
+      "train_step": 6692,
+      "loss": 0.4302821869629315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4302821869629315
+    },
+    {
+      "train_step": 6693,
+      "loss": 0.6488250711738386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6488250711738386
+    },
+    {
+      "train_step": 6694,
+      "loss": 0.5316371208706577,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5316371208706577
+    },
+    {
+      "train_step": 6695,
+      "loss": 0.5797551617868975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5797551617868975
+    },
+    {
+      "train_step": 6696,
+      "loss": 0.5534954101395794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5534954101395794
+    },
+    {
+      "train_step": 6697,
+      "loss": 0.5926101275508865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5926101275508865
+    },
+    {
+      "train_step": 6698,
+      "loss": 0.48335674017483665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48335674017483665
+    },
+    {
+      "train_step": 6699,
+      "loss": 0.6212730027256222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6212730027256222
+    },
+    {
+      "train_step": 6700,
+      "loss": 0.609745806357015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.609745806357015
+    },
+    {
+      "train_step": 6701,
+      "loss": 0.4670244110421946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4670244110421946
+    },
+    {
+      "train_step": 6702,
+      "loss": 0.713430099223996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.713430099223996
+    },
+    {
+      "train_step": 6703,
+      "loss": 0.6760590180895911,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6760590180895911
+    },
+    {
+      "train_step": 6704,
+      "loss": 0.4266724472357784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4266724472357784
+    },
+    {
+      "train_step": 6705,
+      "loss": 0.5389028266403982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5389028266403982
+    },
+    {
+      "train_step": 6706,
+      "loss": 0.6043458578516507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6043458578516507
+    },
+    {
+      "train_step": 6707,
+      "loss": 0.6018444955578527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6018444955578527
+    },
+    {
+      "train_step": 6708,
+      "loss": 0.640562831019658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.640562831019658
+    },
+    {
+      "train_step": 6709,
+      "loss": 0.4382595661567332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4382595661567332
+    },
+    {
+      "train_step": 6710,
+      "loss": 0.5585263852936339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5585263852936339
+    },
+    {
+      "train_step": 6711,
+      "loss": 0.7004566945157289,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7004566945157289
+    },
+    {
+      "train_step": 6712,
+      "loss": 0.5969147919338114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5969147919338114
+    },
+    {
+      "train_step": 6713,
+      "loss": 0.48465435903396475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48465435903396475
+    },
+    {
+      "train_step": 6714,
+      "loss": 0.4064479039616479,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4064479039616479
+    },
+    {
+      "train_step": 6715,
+      "loss": 0.35820044915433796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35820044915433796
+    },
+    {
+      "train_step": 6716,
+      "loss": 0.6068575531382692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6068575531382692
+    },
+    {
+      "train_step": 6717,
+      "loss": 0.5834643821141405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5834643821141405
+    },
+    {
+      "train_step": 6718,
+      "loss": 0.6549419527819513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6549419527819513
+    },
+    {
+      "train_step": 6719,
+      "loss": 0.7024581282276066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7024581282276066
+    },
+    {
+      "train_step": 6720,
+      "loss": 0.570238911406497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.570238911406497
+    },
+    {
+      "train_step": 6721,
+      "loss": 0.556717416855642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.556717416855642
+    },
+    {
+      "train_step": 6722,
+      "loss": 0.5487808398844201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5487808398844201
+    },
+    {
+      "train_step": 6723,
+      "loss": 0.6072189404308406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6072189404308406
+    },
+    {
+      "train_step": 6724,
+      "loss": 0.6314157259247153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6314157259247153
+    },
+    {
+      "train_step": 6725,
+      "loss": 0.6712785784699067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6712785784699067
+    },
+    {
+      "train_step": 6726,
+      "loss": 0.6648598381000201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6648598381000201
+    },
+    {
+      "train_step": 6727,
+      "loss": 0.6095537589888064,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6095537589888064
+    },
+    {
+      "train_step": 6728,
+      "loss": 0.5908487045435048,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5908487045435048
+    },
+    {
+      "train_step": 6729,
+      "loss": 0.6773020331889799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6773020331889799
+    },
+    {
+      "train_step": 6730,
+      "loss": 0.6637206232634595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6637206232634595
+    },
+    {
+      "train_step": 6731,
+      "loss": 0.4878286291678738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4878286291678738
+    },
+    {
+      "train_step": 6732,
+      "loss": 0.4686785596306959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4686785596306959
+    },
+    {
+      "train_step": 6733,
+      "loss": 0.7535271963990229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7535271963990229
+    },
+    {
+      "train_step": 6734,
+      "loss": 0.592400439488236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.592400439488236
+    },
+    {
+      "train_step": 6735,
+      "loss": 0.4484502109508069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4484502109508069
+    },
+    {
+      "train_step": 6736,
+      "loss": 0.5188177064374518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5188177064374518
+    },
+    {
+      "train_step": 6737,
+      "loss": 0.41543988233220525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41543988233220525
+    },
+    {
+      "train_step": 6738,
+      "loss": 0.4136552595831744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4136552595831744
+    },
+    {
+      "train_step": 6739,
+      "loss": 0.5587253561352623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5587253561352623
+    },
+    {
+      "train_step": 6740,
+      "loss": 0.624008487418341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.624008487418341
+    },
+    {
+      "train_step": 6741,
+      "loss": 0.4725096185889113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4725096185889113
+    },
+    {
+      "train_step": 6742,
+      "loss": 0.5839743725930633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5839743725930633
+    },
+    {
+      "train_step": 6743,
+      "loss": 0.4715673441493611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4715673441493611
+    },
+    {
+      "train_step": 6744,
+      "loss": 0.5276423611946409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5276423611946409
+    },
+    {
+      "train_step": 6745,
+      "loss": 0.5039903080647796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5039903080647796
+    },
+    {
+      "train_step": 6746,
+      "loss": 0.6166077115926663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6166077115926663
+    },
+    {
+      "train_step": 6747,
+      "loss": 0.44119880669368333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44119880669368333
+    },
+    {
+      "train_step": 6748,
+      "loss": 0.5990474933611031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5990474933611031
+    },
+    {
+      "train_step": 6749,
+      "loss": 0.41126799413509035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41126799413509035
+    },
+    {
+      "train_step": 6750,
+      "loss": 0.7282818387139558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7282818387139558
+    },
+    {
+      "train_step": 6751,
+      "loss": 0.5169511651224922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5169511651224922
+    },
+    {
+      "train_step": 6752,
+      "loss": 0.6263677371417021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6263677371417021
+    },
+    {
+      "train_step": 6753,
+      "loss": 0.5633995932438123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5633995932438123
+    },
+    {
+      "train_step": 6754,
+      "loss": 0.7346023822564108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7346023822564108
+    },
+    {
+      "train_step": 6755,
+      "loss": 0.5955456336667397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5955456336667397
+    },
+    {
+      "train_step": 6756,
+      "loss": 0.697541819336933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.697541819336933
+    },
+    {
+      "train_step": 6757,
+      "loss": 0.43725970705307327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43725970705307327
+    },
+    {
+      "train_step": 6758,
+      "loss": 0.4493126696657619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4493126696657619
+    },
+    {
+      "train_step": 6759,
+      "loss": 0.7440704902075878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7440704902075878
+    },
+    {
+      "train_step": 6760,
+      "loss": 0.496430549233146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.496430549233146
+    },
+    {
+      "train_step": 6761,
+      "loss": 0.6233051575530038,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6233051575530038
+    },
+    {
+      "train_step": 6762,
+      "loss": 0.3772168653602192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3772168653602192
+    },
+    {
+      "train_step": 6763,
+      "loss": 0.4182902681494829,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4182902681494829
+    },
+    {
+      "train_step": 6764,
+      "loss": 0.5127755053477202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5127755053477202
+    },
+    {
+      "train_step": 6765,
+      "loss": 0.44166500494444133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44166500494444133
+    },
+    {
+      "train_step": 6766,
+      "loss": 0.6703029802112203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6703029802112203
+    },
+    {
+      "train_step": 6767,
+      "loss": 0.48953015602341554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48953015602341554
+    },
+    {
+      "train_step": 6768,
+      "loss": 0.5445628513787659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5445628513787659
+    },
+    {
+      "train_step": 6769,
+      "loss": 0.6016364851375738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6016364851375738
+    },
+    {
+      "train_step": 6770,
+      "loss": 0.4609805569897683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4609805569897683
+    },
+    {
+      "train_step": 6771,
+      "loss": 0.6374482575542803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6374482575542803
+    },
+    {
+      "train_step": 6772,
+      "loss": 0.5000000975201934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5000000975201934
+    },
+    {
+      "train_step": 6773,
+      "loss": 0.5094728469858616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5094728469858616
+    },
+    {
+      "train_step": 6774,
+      "loss": 0.4365198652710871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4365198652710871
+    },
+    {
+      "train_step": 6775,
+      "loss": 0.5992589749886589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5992589749886589
+    },
+    {
+      "train_step": 6776,
+      "loss": 0.4637365674081568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4637365674081568
+    },
+    {
+      "train_step": 6777,
+      "loss": 0.4036264991133489,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4036264991133489
+    },
+    {
+      "train_step": 6778,
+      "loss": 0.39794454166247323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39794454166247323
+    },
+    {
+      "train_step": 6779,
+      "loss": 0.697166898652801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.697166898652801
+    },
+    {
+      "train_step": 6780,
+      "loss": 0.5051024692960279,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5051024692960279
+    },
+    {
+      "train_step": 6781,
+      "loss": 0.5877578503351275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5877578503351275
+    },
+    {
+      "train_step": 6782,
+      "loss": 0.538753886063458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.538753886063458
+    },
+    {
+      "train_step": 6783,
+      "loss": 0.651049637644975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.651049637644975
+    },
+    {
+      "train_step": 6784,
+      "loss": 0.7260955626179898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7260955626179898
+    },
+    {
+      "train_step": 6785,
+      "loss": 0.4896182239362766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4896182239362766
+    },
+    {
+      "train_step": 6786,
+      "loss": 0.4901040101570645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4901040101570645
+    },
+    {
+      "train_step": 6787,
+      "loss": 0.609823585075348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.609823585075348
+    },
+    {
+      "train_step": 6788,
+      "loss": 0.6173572605110712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6173572605110712
+    },
+    {
+      "train_step": 6789,
+      "loss": 0.4691505425340869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4691505425340869
+    },
+    {
+      "train_step": 6790,
+      "loss": 0.564436037770954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.564436037770954
+    },
+    {
+      "train_step": 6791,
+      "loss": 0.5139482615651166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5139482615651166
+    },
+    {
+      "train_step": 6792,
+      "loss": 0.7856988373959305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7856988373959305
+    },
+    {
+      "train_step": 6793,
+      "loss": 0.5270634685002263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5270634685002263
+    },
+    {
+      "train_step": 6794,
+      "loss": 0.6305598286521834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6305598286521834
+    },
+    {
+      "train_step": 6795,
+      "loss": 0.6777296589693197,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6777296589693197
+    },
+    {
+      "train_step": 6796,
+      "loss": 0.5403551424558752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5403551424558752
+    },
+    {
+      "train_step": 6797,
+      "loss": 0.6057641627629273,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6057641627629273
+    },
+    {
+      "train_step": 6798,
+      "loss": 0.5213071624212706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5213071624212706
+    },
+    {
+      "train_step": 6799,
+      "loss": 0.7733374566146898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7733374566146898
+    },
+    {
+      "train_step": 6800,
+      "loss": 0.5508011056322417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5508011056322417
+    },
+    {
+      "train_step": 6801,
+      "loss": 0.41829366973068705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41829366973068705
+    },
+    {
+      "train_step": 6802,
+      "loss": 0.5890020995818828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5890020995818828
+    },
+    {
+      "train_step": 6803,
+      "loss": 0.4663879817613859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4663879817613859
+    },
+    {
+      "train_step": 6804,
+      "loss": 0.3781986552252126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3781986552252126
+    },
+    {
+      "train_step": 6805,
+      "loss": 0.5711637597485185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5711637597485185
+    },
+    {
+      "train_step": 6806,
+      "loss": 0.36541167259534746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36541167259534746
+    },
+    {
+      "train_step": 6807,
+      "loss": 0.6408221143137434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6408221143137434
+    },
+    {
+      "train_step": 6808,
+      "loss": 0.40959327949722113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40959327949722113
+    },
+    {
+      "train_step": 6809,
+      "loss": 0.3848843731601628,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3848843731601628
+    },
+    {
+      "train_step": 6810,
+      "loss": 0.4070496065302474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4070496065302474
+    },
+    {
+      "train_step": 6811,
+      "loss": 0.6456418413214222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6456418413214222
+    },
+    {
+      "train_step": 6812,
+      "loss": 0.4642398816206958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4642398816206958
+    },
+    {
+      "train_step": 6813,
+      "loss": 0.5094567754821885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5094567754821885
+    },
+    {
+      "train_step": 6814,
+      "loss": 0.588538311990763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.588538311990763
+    },
+    {
+      "train_step": 6815,
+      "loss": 0.5729539064456166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5729539064456166
+    },
+    {
+      "train_step": 6816,
+      "loss": 0.6146710595300173,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6146710595300173
+    },
+    {
+      "train_step": 6817,
+      "loss": 0.6221031043819967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6221031043819967
+    },
+    {
+      "train_step": 6818,
+      "loss": 0.6118851433345068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6118851433345068
+    },
+    {
+      "train_step": 6819,
+      "loss": 0.5673786988256838,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5673786988256838
+    },
+    {
+      "train_step": 6820,
+      "loss": 0.6384683539285817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6384683539285817
+    },
+    {
+      "train_step": 6821,
+      "loss": 0.5423535506700246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5423535506700246
+    },
+    {
+      "train_step": 6822,
+      "loss": 0.6229739015297148,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6229739015297148
+    },
+    {
+      "train_step": 6823,
+      "loss": 0.6344443976169819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6344443976169819
+    },
+    {
+      "train_step": 6824,
+      "loss": 0.5993560508997734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5993560508997734
+    },
+    {
+      "train_step": 6825,
+      "loss": 0.6077093764703528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6077093764703528
+    },
+    {
+      "train_step": 6826,
+      "loss": 0.6818781763292344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6818781763292344
+    },
+    {
+      "train_step": 6827,
+      "loss": 0.5875988303319251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5875988303319251
+    },
+    {
+      "train_step": 6828,
+      "loss": 0.6672188112386616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6672188112386616
+    },
+    {
+      "train_step": 6829,
+      "loss": 0.5687197844022556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5687197844022556
+    },
+    {
+      "train_step": 6830,
+      "loss": 0.8129747999182896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8129747999182896
+    },
+    {
+      "train_step": 6831,
+      "loss": 0.7298915539821069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7298915539821069
+    },
+    {
+      "train_step": 6832,
+      "loss": 0.6184020192762356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6184020192762356
+    },
+    {
+      "train_step": 6833,
+      "loss": 0.4407406851971855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4407406851971855
+    },
+    {
+      "train_step": 6834,
+      "loss": 0.7214796056131467,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7214796056131467
+    },
+    {
+      "train_step": 6835,
+      "loss": 0.6299357685619692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6299357685619692
+    },
+    {
+      "train_step": 6836,
+      "loss": 0.49774203426547997,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49774203426547997
+    },
+    {
+      "train_step": 6837,
+      "loss": 0.481073591607191,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.481073591607191
+    },
+    {
+      "train_step": 6838,
+      "loss": 0.4755099302948189,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4755099302948189
+    },
+    {
+      "train_step": 6839,
+      "loss": 0.35508002677624934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35508002677624934
+    },
+    {
+      "train_step": 6840,
+      "loss": 0.439834895855849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.439834895855849
+    },
+    {
+      "train_step": 6841,
+      "loss": 0.47292083115772876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47292083115772876
+    },
+    {
+      "train_step": 6842,
+      "loss": 0.7184143689307667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7184143689307667
+    },
+    {
+      "train_step": 6843,
+      "loss": 0.5441929189409487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441929189409487
+    },
+    {
+      "train_step": 6844,
+      "loss": 0.6954358117233387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6954358117233387
+    },
+    {
+      "train_step": 6845,
+      "loss": 0.4378513931450849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4378513931450849
+    },
+    {
+      "train_step": 6846,
+      "loss": 0.5983278584968879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5983278584968879
+    },
+    {
+      "train_step": 6847,
+      "loss": 0.5429885743690791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5429885743690791
+    },
+    {
+      "train_step": 6848,
+      "loss": 0.5546239985476389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5546239985476389
+    },
+    {
+      "train_step": 6849,
+      "loss": 0.5351215191007093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5351215191007093
+    },
+    {
+      "train_step": 6850,
+      "loss": 0.625686034262191,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.625686034262191
+    },
+    {
+      "train_step": 6851,
+      "loss": 0.6619510659952813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6619510659952813
+    },
+    {
+      "train_step": 6852,
+      "loss": 0.4755377856592131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4755377856592131
+    },
+    {
+      "train_step": 6853,
+      "loss": 0.38571570738336874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38571570738336874
+    },
+    {
+      "train_step": 6854,
+      "loss": 0.5426630856107398,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5426630856107398
+    },
+    {
+      "train_step": 6855,
+      "loss": 0.4210971013013192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4210971013013192
+    },
+    {
+      "train_step": 6856,
+      "loss": 0.5076932824810546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5076932824810546
+    },
+    {
+      "train_step": 6857,
+      "loss": 0.4057689237557609,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4057689237557609
+    },
+    {
+      "train_step": 6858,
+      "loss": 0.6616694856409308,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6616694856409308
+    },
+    {
+      "train_step": 6859,
+      "loss": 0.6297937241775218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6297937241775218
+    },
+    {
+      "train_step": 6860,
+      "loss": 0.5741048086151788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5741048086151788
+    },
+    {
+      "train_step": 6861,
+      "loss": 0.5390545564980339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5390545564980339
+    },
+    {
+      "train_step": 6862,
+      "loss": 0.40105497723671113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40105497723671113
+    },
+    {
+      "train_step": 6863,
+      "loss": 0.6752381269476323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6752381269476323
+    },
+    {
+      "train_step": 6864,
+      "loss": 0.6259359465607749,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6259359465607749
+    },
+    {
+      "train_step": 6865,
+      "loss": 0.7528386566813063,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7528386566813063
+    },
+    {
+      "train_step": 6866,
+      "loss": 0.5223265689045606,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5223265689045606
+    },
+    {
+      "train_step": 6867,
+      "loss": 0.6654353146299153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6654353146299153
+    },
+    {
+      "train_step": 6868,
+      "loss": 0.7906563365686832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7906563365686832
+    },
+    {
+      "train_step": 6869,
+      "loss": 0.73056382667135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.73056382667135
+    },
+    {
+      "train_step": 6870,
+      "loss": 0.5779552982017536,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5779552982017536
+    },
+    {
+      "train_step": 6871,
+      "loss": 0.5727498700648701,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5727498700648701
+    },
+    {
+      "train_step": 6872,
+      "loss": 0.5333851227733867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5333851227733867
+    },
+    {
+      "train_step": 6873,
+      "loss": 0.5033524448721488,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5033524448721488
+    },
+    {
+      "train_step": 6874,
+      "loss": 0.7601470893164232,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7601470893164232
+    },
+    {
+      "train_step": 6875,
+      "loss": 0.46962403446779866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46962403446779866
+    },
+    {
+      "train_step": 6876,
+      "loss": 0.6698480891172424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6698480891172424
+    },
+    {
+      "train_step": 6877,
+      "loss": 0.34682685764065824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34682685764065824
+    },
+    {
+      "train_step": 6878,
+      "loss": 0.46754269915750546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46754269915750546
+    },
+    {
+      "train_step": 6879,
+      "loss": 0.5912974417014422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5912974417014422
+    },
+    {
+      "train_step": 6880,
+      "loss": 0.3806610298551104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3806610298551104
+    },
+    {
+      "train_step": 6881,
+      "loss": 0.6964932454760498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6964932454760498
+    },
+    {
+      "train_step": 6882,
+      "loss": 0.7467574963201217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7467574963201217
+    },
+    {
+      "train_step": 6883,
+      "loss": 0.48348997437401803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48348997437401803
+    },
+    {
+      "train_step": 6884,
+      "loss": 0.6633519418337512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6633519418337512
+    },
+    {
+      "train_step": 6885,
+      "loss": 0.6889963205176365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6889963205176365
+    },
+    {
+      "train_step": 6886,
+      "loss": 0.5313496893205324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5313496893205324
+    },
+    {
+      "train_step": 6887,
+      "loss": 0.5968604565350469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5968604565350469
+    },
+    {
+      "train_step": 6888,
+      "loss": 0.48862031510794457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48862031510794457
+    },
+    {
+      "train_step": 6889,
+      "loss": 0.4333497849597622,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4333497849597622
+    },
+    {
+      "train_step": 6890,
+      "loss": 0.41915504465798226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41915504465798226
+    },
+    {
+      "train_step": 6891,
+      "loss": 0.6024858453726122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6024858453726122
+    },
+    {
+      "train_step": 6892,
+      "loss": 0.7242746870197739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7242746870197739
+    },
+    {
+      "train_step": 6893,
+      "loss": 0.6973863710829232,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6973863710829232
+    },
+    {
+      "train_step": 6894,
+      "loss": 0.443539726713592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.443539726713592
+    },
+    {
+      "train_step": 6895,
+      "loss": 0.6611389512773724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6611389512773724
+    },
+    {
+      "train_step": 6896,
+      "loss": 0.3198931865985106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3198931865985106
+    },
+    {
+      "train_step": 6897,
+      "loss": 0.6916783248937618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6916783248937618
+    },
+    {
+      "train_step": 6898,
+      "loss": 0.4191592128849623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4191592128849623
+    },
+    {
+      "train_step": 6899,
+      "loss": 0.7749502530651905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7749502530651905
+    },
+    {
+      "train_step": 6900,
+      "loss": 0.6311181894357478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6311181894357478
+    },
+    {
+      "train_step": 6901,
+      "loss": 0.48995160268926485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48995160268926485
+    },
+    {
+      "train_step": 6902,
+      "loss": 0.5708710630488085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5708710630488085
+    },
+    {
+      "train_step": 6903,
+      "loss": 0.7092389036181264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7092389036181264
+    },
+    {
+      "train_step": 6904,
+      "loss": 0.6186902870864043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6186902870864043
+    },
+    {
+      "train_step": 6905,
+      "loss": 0.4201554238405294,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4201554238405294
+    },
+    {
+      "train_step": 6906,
+      "loss": 0.4411597371786211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4411597371786211
+    },
+    {
+      "train_step": 6907,
+      "loss": 0.4266313944074799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4266313944074799
+    },
+    {
+      "train_step": 6908,
+      "loss": 0.5148567485720766,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5148567485720766
+    },
+    {
+      "train_step": 6909,
+      "loss": 0.4841037488580512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4841037488580512
+    },
+    {
+      "train_step": 6910,
+      "loss": 0.539325790694102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.539325790694102
+    },
+    {
+      "train_step": 6911,
+      "loss": 0.5098744356344347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5098744356344347
+    },
+    {
+      "train_step": 6912,
+      "loss": 0.6092265636188622,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6092265636188622
+    },
+    {
+      "train_step": 6913,
+      "loss": 0.5199714824688847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5199714824688847
+    },
+    {
+      "train_step": 6914,
+      "loss": 0.4818068824565143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4818068824565143
+    },
+    {
+      "train_step": 6915,
+      "loss": 0.607760512111346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.607760512111346
+    },
+    {
+      "train_step": 6916,
+      "loss": 0.7290335549733982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7290335549733982
+    },
+    {
+      "train_step": 6917,
+      "loss": 0.3955896643188181,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3955896643188181
+    },
+    {
+      "train_step": 6918,
+      "loss": 0.5900131573454495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5900131573454495
+    },
+    {
+      "train_step": 6919,
+      "loss": 0.62533257125971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.62533257125971
+    },
+    {
+      "train_step": 6920,
+      "loss": 0.7007023832064307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7007023832064307
+    },
+    {
+      "train_step": 6921,
+      "loss": 0.3759506022025101,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3759506022025101
+    },
+    {
+      "train_step": 6922,
+      "loss": 0.6848575415397392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6848575415397392
+    },
+    {
+      "train_step": 6923,
+      "loss": 0.49778229361769355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49778229361769355
+    },
+    {
+      "train_step": 6924,
+      "loss": 0.4431863731602362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4431863731602362
+    },
+    {
+      "train_step": 6925,
+      "loss": 0.4607000110238707,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4607000110238707
+    },
+    {
+      "train_step": 6926,
+      "loss": 0.38639940577479637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38639940577479637
+    },
+    {
+      "train_step": 6927,
+      "loss": 0.5418104964241455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5418104964241455
+    },
+    {
+      "train_step": 6928,
+      "loss": 0.510364476597205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.510364476597205
+    },
+    {
+      "train_step": 6929,
+      "loss": 0.5917580173165617,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5917580173165617
+    },
+    {
+      "train_step": 6930,
+      "loss": 0.4745713233791493,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4745713233791493
+    },
+    {
+      "train_step": 6931,
+      "loss": 0.607861714692583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.607861714692583
+    },
+    {
+      "train_step": 6932,
+      "loss": 0.7522657219229508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7522657219229508
+    },
+    {
+      "train_step": 6933,
+      "loss": 0.5369994956862866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5369994956862866
+    },
+    {
+      "train_step": 6934,
+      "loss": 0.582170524143771,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.582170524143771
+    },
+    {
+      "train_step": 6935,
+      "loss": 0.45554019035659493,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45554019035659493
+    },
+    {
+      "train_step": 6936,
+      "loss": 0.4024012985810475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4024012985810475
+    },
+    {
+      "train_step": 6937,
+      "loss": 0.5991863125974028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5991863125974028
+    },
+    {
+      "train_step": 6938,
+      "loss": 0.7581902458529273,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7581902458529273
+    },
+    {
+      "train_step": 6939,
+      "loss": 0.5140868990781368,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5140868990781368
+    },
+    {
+      "train_step": 6940,
+      "loss": 0.6425700458913567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6425700458913567
+    },
+    {
+      "train_step": 6941,
+      "loss": 0.5869393673710935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5869393673710935
+    },
+    {
+      "train_step": 6942,
+      "loss": 0.6506774188421042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6506774188421042
+    },
+    {
+      "train_step": 6943,
+      "loss": 0.6635559244178683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6635559244178683
+    },
+    {
+      "train_step": 6944,
+      "loss": 0.6935484208576199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6935484208576199
+    },
+    {
+      "train_step": 6945,
+      "loss": 0.6253121844630942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6253121844630942
+    },
+    {
+      "train_step": 6946,
+      "loss": 0.4886177494390267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4886177494390267
+    },
+    {
+      "train_step": 6947,
+      "loss": 0.7270167956470606,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7270167956470606
+    },
+    {
+      "train_step": 6948,
+      "loss": 0.6874845387284046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6874845387284046
+    },
+    {
+      "train_step": 6949,
+      "loss": 0.6374539607808736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6374539607808736
+    },
+    {
+      "train_step": 6950,
+      "loss": 0.44038740566214474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44038740566214474
+    },
+    {
+      "train_step": 6951,
+      "loss": 0.7013049808632145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7013049808632145
+    },
+    {
+      "train_step": 6952,
+      "loss": 0.6631293191596489,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6631293191596489
+    },
+    {
+      "train_step": 6953,
+      "loss": 0.5796800687797691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5796800687797691
+    },
+    {
+      "train_step": 6954,
+      "loss": 0.3571144437722515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3571144437722515
+    },
+    {
+      "train_step": 6955,
+      "loss": 0.34752522485163506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34752522485163506
+    },
+    {
+      "train_step": 6956,
+      "loss": 0.500706987335019,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.500706987335019
+    },
+    {
+      "train_step": 6957,
+      "loss": 0.7218424679390685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7218424679390685
+    },
+    {
+      "train_step": 6958,
+      "loss": 0.6548015701439961,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6548015701439961
+    },
+    {
+      "train_step": 6959,
+      "loss": 0.6230683741480976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6230683741480976
+    },
+    {
+      "train_step": 6960,
+      "loss": 0.419864946688354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.419864946688354
+    },
+    {
+      "train_step": 6961,
+      "loss": 0.6832676333188495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6832676333188495
+    },
+    {
+      "train_step": 6962,
+      "loss": 0.5948668179110569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5948668179110569
+    },
+    {
+      "train_step": 6963,
+      "loss": 0.6933004218995712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6933004218995712
+    },
+    {
+      "train_step": 6964,
+      "loss": 0.5329655279825827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5329655279825827
+    },
+    {
+      "train_step": 6965,
+      "loss": 0.7014790983438265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7014790983438265
+    },
+    {
+      "train_step": 6966,
+      "loss": 0.7775680817703731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7775680817703731
+    },
+    {
+      "train_step": 6967,
+      "loss": 0.6059362450720368,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6059362450720368
+    },
+    {
+      "train_step": 6968,
+      "loss": 0.45000569195829243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45000569195829243
+    },
+    {
+      "train_step": 6969,
+      "loss": 0.5285606708718165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5285606708718165
+    },
+    {
+      "train_step": 6970,
+      "loss": 0.5308404916319751,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5308404916319751
+    },
+    {
+      "train_step": 6971,
+      "loss": 0.7176226918790826,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7176226918790826
+    },
+    {
+      "train_step": 6972,
+      "loss": 0.6005249729814792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6005249729814792
+    },
+    {
+      "train_step": 6973,
+      "loss": 0.6167520023846865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6167520023846865
+    },
+    {
+      "train_step": 6974,
+      "loss": 0.6637330396765284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6637330396765284
+    },
+    {
+      "train_step": 6975,
+      "loss": 0.8832207231506664,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8832207231506664
+    },
+    {
+      "train_step": 6976,
+      "loss": 0.6069917659944603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6069917659944603
+    },
+    {
+      "train_step": 6977,
+      "loss": 1.0953030156800836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 1.0953030156800836
+    },
+    {
+      "train_step": 6978,
+      "loss": 0.27998446928066295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.27998446928066295
+    },
+    {
+      "train_step": 6979,
+      "loss": 0.47956988216727486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47956988216727486
+    },
+    {
+      "train_step": 6980,
+      "loss": 0.6589318920415577,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6589318920415577
+    },
+    {
+      "train_step": 6981,
+      "loss": 0.5689403691254165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5689403691254165
+    },
+    {
+      "train_step": 6982,
+      "loss": 0.6771429073616907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6771429073616907
+    },
+    {
+      "train_step": 6983,
+      "loss": 0.46142237388137813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46142237388137813
+    },
+    {
+      "train_step": 6984,
+      "loss": 0.542008904767641,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.542008904767641
+    },
+    {
+      "train_step": 6985,
+      "loss": 0.671032179704315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.671032179704315
+    },
+    {
+      "train_step": 6986,
+      "loss": 0.36302608044833234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36302608044833234
+    },
+    {
+      "train_step": 6987,
+      "loss": 0.6446356609903134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6446356609903134
+    },
+    {
+      "train_step": 6988,
+      "loss": 0.42318236579117774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42318236579117774
+    },
+    {
+      "train_step": 6989,
+      "loss": 0.407792745797651,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.407792745797651
+    },
+    {
+      "train_step": 6990,
+      "loss": 0.7386284935565762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7386284935565762
+    },
+    {
+      "train_step": 6991,
+      "loss": 0.6338764605653989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6338764605653989
+    },
+    {
+      "train_step": 6992,
+      "loss": 0.39087378142024953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39087378142024953
+    },
+    {
+      "train_step": 6993,
+      "loss": 0.5976043850808683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5976043850808683
+    },
+    {
+      "train_step": 6994,
+      "loss": 0.6508492171185007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6508492171185007
+    },
+    {
+      "train_step": 6995,
+      "loss": 0.687347385871896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.687347385871896
+    },
+    {
+      "train_step": 6996,
+      "loss": 0.4564725796397201,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4564725796397201
+    },
+    {
+      "train_step": 6997,
+      "loss": 0.45211079990652847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45211079990652847
+    },
+    {
+      "train_step": 6998,
+      "loss": 0.638556948620694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.638556948620694
+    },
+    {
+      "train_step": 6999,
+      "loss": 0.764206426334478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.764206426334478
+    },
+    {
+      "train_step": 7000,
+      "loss": 0.6159768537343706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6159768537343706
+    },
+    {
+      "train_step": 7001,
+      "loss": 0.5063939514285175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5063939514285175
+    },
+    {
+      "train_step": 7002,
+      "loss": 0.4160152698569515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4160152698569515
+    },
+    {
+      "train_step": 7003,
+      "loss": 0.7257889140990301,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7257889140990301
+    },
+    {
+      "train_step": 7004,
+      "loss": 0.5276898204539546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5276898204539546
+    },
+    {
+      "train_step": 7005,
+      "loss": 0.4714729075268328,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4714729075268328
+    },
+    {
+      "train_step": 7006,
+      "loss": 0.49821846037596773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49821846037596773
+    },
+    {
+      "train_step": 7007,
+      "loss": 0.9073293056538976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9073293056538976
+    },
+    {
+      "train_step": 7008,
+      "loss": 0.7650733679728344,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7650733679728344
+    },
+    {
+      "train_step": 7009,
+      "loss": 0.6099947826760652,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6099947826760652
+    },
+    {
+      "train_step": 7010,
+      "loss": 0.48173126431961394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48173126431961394
+    },
+    {
+      "train_step": 7011,
+      "loss": 0.7607835977518783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7607835977518783
+    },
+    {
+      "train_step": 7012,
+      "loss": 0.9076981441059231,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9076981441059231
+    },
+    {
+      "train_step": 7013,
+      "loss": 0.5056811940689541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5056811940689541
+    },
+    {
+      "train_step": 7014,
+      "loss": 0.4826110840610939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4826110840610939
+    },
+    {
+      "train_step": 7015,
+      "loss": 0.718672482285972,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.718672482285972
+    },
+    {
+      "train_step": 7016,
+      "loss": 0.5173506394241429,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5173506394241429
+    },
+    {
+      "train_step": 7017,
+      "loss": 0.5996309735376347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5996309735376347
+    },
+    {
+      "train_step": 7018,
+      "loss": 0.4203499842109151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4203499842109151
+    },
+    {
+      "train_step": 7019,
+      "loss": 0.5195726762789736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5195726762789736
+    },
+    {
+      "train_step": 7020,
+      "loss": 0.5071043320610378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5071043320610378
+    },
+    {
+      "train_step": 7021,
+      "loss": 0.6351067196477262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6351067196477262
+    },
+    {
+      "train_step": 7022,
+      "loss": 0.5745276173565896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5745276173565896
+    },
+    {
+      "train_step": 7023,
+      "loss": 0.4757658802737549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4757658802737549
+    },
+    {
+      "train_step": 7024,
+      "loss": 0.5376684344486395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5376684344486395
+    },
+    {
+      "train_step": 7025,
+      "loss": 0.642302034604298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.642302034604298
+    },
+    {
+      "train_step": 7026,
+      "loss": 0.38526330962393146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38526330962393146
+    },
+    {
+      "train_step": 7027,
+      "loss": 0.48601169742195416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48601169742195416
+    },
+    {
+      "train_step": 7028,
+      "loss": 0.5202178914811676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5202178914811676
+    },
+    {
+      "train_step": 7029,
+      "loss": 0.5302399650714976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5302399650714976
+    },
+    {
+      "train_step": 7030,
+      "loss": 0.6394040806095007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6394040806095007
+    },
+    {
+      "train_step": 7031,
+      "loss": 0.3847617418159123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3847617418159123
+    },
+    {
+      "train_step": 7032,
+      "loss": 0.5629451068252194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5629451068252194
+    },
+    {
+      "train_step": 7033,
+      "loss": 0.5708661035378911,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5708661035378911
+    },
+    {
+      "train_step": 7034,
+      "loss": 0.5208765064176644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5208765064176644
+    },
+    {
+      "train_step": 7035,
+      "loss": 0.640387124914263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.640387124914263
+    },
+    {
+      "train_step": 7036,
+      "loss": 0.6904177430795229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6904177430795229
+    },
+    {
+      "train_step": 7037,
+      "loss": 0.70830634566437,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.70830634566437
+    },
+    {
+      "train_step": 7038,
+      "loss": 0.6531161927316778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6531161927316778
+    },
+    {
+      "train_step": 7039,
+      "loss": 0.484571359647336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.484571359647336
+    },
+    {
+      "train_step": 7040,
+      "loss": 0.6930192577102606,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6930192577102606
+    },
+    {
+      "train_step": 7041,
+      "loss": 0.5427564822620022,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5427564822620022
+    },
+    {
+      "train_step": 7042,
+      "loss": 0.49771778705014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49771778705014
+    },
+    {
+      "train_step": 7043,
+      "loss": 0.5453390601321471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5453390601321471
+    },
+    {
+      "train_step": 7044,
+      "loss": 0.44747549883326665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44747549883326665
+    },
+    {
+      "train_step": 7045,
+      "loss": 0.5802793709730298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5802793709730298
+    },
+    {
+      "train_step": 7046,
+      "loss": 0.6223088409531589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6223088409531589
+    },
+    {
+      "train_step": 7047,
+      "loss": 0.39096112412723316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39096112412723316
+    },
+    {
+      "train_step": 7048,
+      "loss": 0.35310988562699636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35310988562699636
+    },
+    {
+      "train_step": 7049,
+      "loss": 0.33062130209230733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33062130209230733
+    },
+    {
+      "train_step": 7050,
+      "loss": 0.4614547698272798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4614547698272798
+    },
+    {
+      "train_step": 7051,
+      "loss": 0.551114070425256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.551114070425256
+    },
+    {
+      "train_step": 7052,
+      "loss": 0.6806747112621903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6806747112621903
+    },
+    {
+      "train_step": 7053,
+      "loss": 0.6342687261836761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6342687261836761
+    },
+    {
+      "train_step": 7054,
+      "loss": 0.7955308614710388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7955308614710388
+    },
+    {
+      "train_step": 7055,
+      "loss": 0.5549179134169411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5549179134169411
+    },
+    {
+      "train_step": 7056,
+      "loss": 0.6570139401858525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6570139401858525
+    },
+    {
+      "train_step": 7057,
+      "loss": 0.5962935844375902,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962935844375902
+    },
+    {
+      "train_step": 7058,
+      "loss": 0.7560915740413701,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7560915740413701
+    },
+    {
+      "train_step": 7059,
+      "loss": 0.5329350393747495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5329350393747495
+    },
+    {
+      "train_step": 7060,
+      "loss": 0.5403557390376318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5403557390376318
+    },
+    {
+      "train_step": 7061,
+      "loss": 0.7624466902667015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7624466902667015
+    },
+    {
+      "train_step": 7062,
+      "loss": 0.4630804899675816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4630804899675816
+    },
+    {
+      "train_step": 7063,
+      "loss": 0.4537088649941467,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4537088649941467
+    },
+    {
+      "train_step": 7064,
+      "loss": 0.4689045255484019,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4689045255484019
+    },
+    {
+      "train_step": 7065,
+      "loss": 0.5834384511997808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5834384511997808
+    },
+    {
+      "train_step": 7066,
+      "loss": 0.611975817565336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.611975817565336
+    },
+    {
+      "train_step": 7067,
+      "loss": 0.5753325299531715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5753325299531715
+    },
+    {
+      "train_step": 7068,
+      "loss": 0.6823747874650636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6823747874650636
+    },
+    {
+      "train_step": 7069,
+      "loss": 0.7554706320988215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7554706320988215
+    },
+    {
+      "train_step": 7070,
+      "loss": 0.6561971380787085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6561971380787085
+    },
+    {
+      "train_step": 7071,
+      "loss": 0.5173383194993693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5173383194993693
+    },
+    {
+      "train_step": 7072,
+      "loss": 0.7472162068680933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7472162068680933
+    },
+    {
+      "train_step": 7073,
+      "loss": 0.6998797478805614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6998797478805614
+    },
+    {
+      "train_step": 7074,
+      "loss": 0.581073219235502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.581073219235502
+    },
+    {
+      "train_step": 7075,
+      "loss": 0.5949915010477718,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5949915010477718
+    },
+    {
+      "train_step": 7076,
+      "loss": 0.5117936982885178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5117936982885178
+    },
+    {
+      "train_step": 7077,
+      "loss": 0.6807627661973027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6807627661973027
+    },
+    {
+      "train_step": 7078,
+      "loss": 0.6175158032510011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6175158032510011
+    },
+    {
+      "train_step": 7079,
+      "loss": 0.5220358870946602,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5220358870946602
+    },
+    {
+      "train_step": 7080,
+      "loss": 0.6166871346090712,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6166871346090712
+    },
+    {
+      "train_step": 7081,
+      "loss": 0.5748535655111956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5748535655111956
+    },
+    {
+      "train_step": 7082,
+      "loss": 0.6981399460967908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6981399460967908
+    },
+    {
+      "train_step": 7083,
+      "loss": 0.692491460833084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.692491460833084
+    },
+    {
+      "train_step": 7084,
+      "loss": 0.48133258242323324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48133258242323324
+    },
+    {
+      "train_step": 7085,
+      "loss": 0.5773020791947552,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5773020791947552
+    },
+    {
+      "train_step": 7086,
+      "loss": 0.5962355848937937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962355848937937
+    },
+    {
+      "train_step": 7087,
+      "loss": 0.649336961752117,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.649336961752117
+    },
+    {
+      "train_step": 7088,
+      "loss": 0.4247898223495901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4247898223495901
+    },
+    {
+      "train_step": 7089,
+      "loss": 0.7486255858793275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7486255858793275
+    },
+    {
+      "train_step": 7090,
+      "loss": 0.6611997951890263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6611997951890263
+    },
+    {
+      "train_step": 7091,
+      "loss": 0.5163214353326017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5163214353326017
+    },
+    {
+      "train_step": 7092,
+      "loss": 0.41929687860923354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41929687860923354
+    },
+    {
+      "train_step": 7093,
+      "loss": 0.5807795864655966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5807795864655966
+    },
+    {
+      "train_step": 7094,
+      "loss": 0.6698485716584432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6698485716584432
+    },
+    {
+      "train_step": 7095,
+      "loss": 0.5416734199930068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5416734199930068
+    },
+    {
+      "train_step": 7096,
+      "loss": 0.5893727321343372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5893727321343372
+    },
+    {
+      "train_step": 7097,
+      "loss": 0.6092596577892652,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6092596577892652
+    },
+    {
+      "train_step": 7098,
+      "loss": 0.5608437386534969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5608437386534969
+    },
+    {
+      "train_step": 7099,
+      "loss": 0.7006754276511155,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7006754276511155
+    },
+    {
+      "train_step": 7100,
+      "loss": 0.5795269233924806,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5795269233924806
+    },
+    {
+      "train_step": 7101,
+      "loss": 0.6598898111569969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6598898111569969
+    },
+    {
+      "train_step": 7102,
+      "loss": 0.387849088127105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.387849088127105
+    },
+    {
+      "train_step": 7103,
+      "loss": 0.4183176841351062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4183176841351062
+    },
+    {
+      "train_step": 7104,
+      "loss": 0.60188373639817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.60188373639817
+    },
+    {
+      "train_step": 7105,
+      "loss": 0.28827657105943116,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.28827657105943116
+    },
+    {
+      "train_step": 7106,
+      "loss": 0.41462713909256843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41462713909256843
+    },
+    {
+      "train_step": 7107,
+      "loss": 0.6590221631471274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6590221631471274
+    },
+    {
+      "train_step": 7108,
+      "loss": 0.6695202434237341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6695202434237341
+    },
+    {
+      "train_step": 7109,
+      "loss": 0.7028646199193889,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7028646199193889
+    },
+    {
+      "train_step": 7110,
+      "loss": 0.7804765426460408,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7804765426460408
+    },
+    {
+      "train_step": 7111,
+      "loss": 0.4568671092325026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4568671092325026
+    },
+    {
+      "train_step": 7112,
+      "loss": 0.7275255909602888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7275255909602888
+    },
+    {
+      "train_step": 7113,
+      "loss": 0.6763199374761153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6763199374761153
+    },
+    {
+      "train_step": 7114,
+      "loss": 0.8204437382221542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8204437382221542
+    },
+    {
+      "train_step": 7115,
+      "loss": 0.5274940983932355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5274940983932355
+    },
+    {
+      "train_step": 7116,
+      "loss": 0.5513270843576129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5513270843576129
+    },
+    {
+      "train_step": 7117,
+      "loss": 0.47072057817778007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47072057817778007
+    },
+    {
+      "train_step": 7118,
+      "loss": 0.6067834868003822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6067834868003822
+    },
+    {
+      "train_step": 7119,
+      "loss": 0.4190390676878605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4190390676878605
+    },
+    {
+      "train_step": 7120,
+      "loss": 0.5693261966391212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5693261966391212
+    },
+    {
+      "train_step": 7121,
+      "loss": 0.546099359465266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.546099359465266
+    },
+    {
+      "train_step": 7122,
+      "loss": 0.52574414143168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.52574414143168
+    },
+    {
+      "train_step": 7123,
+      "loss": 0.41502949003694234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41502949003694234
+    },
+    {
+      "train_step": 7124,
+      "loss": 0.36572040262141275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36572040262141275
+    },
+    {
+      "train_step": 7125,
+      "loss": 0.3851447136183637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3851447136183637
+    },
+    {
+      "train_step": 7126,
+      "loss": 0.5784375461746886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5784375461746886
+    },
+    {
+      "train_step": 7127,
+      "loss": 0.4068999951434521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4068999951434521
+    },
+    {
+      "train_step": 7128,
+      "loss": 0.6686534245714547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6686534245714547
+    },
+    {
+      "train_step": 7129,
+      "loss": 0.5225205500970146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5225205500970146
+    },
+    {
+      "train_step": 7130,
+      "loss": 0.5653150867811405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5653150867811405
+    },
+    {
+      "train_step": 7131,
+      "loss": 0.6774331982958085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6774331982958085
+    },
+    {
+      "train_step": 7132,
+      "loss": 0.5381287323389272,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5381287323389272
+    },
+    {
+      "train_step": 7133,
+      "loss": 0.6115706693342389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6115706693342389
+    },
+    {
+      "train_step": 7134,
+      "loss": 0.5761305102802757,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5761305102802757
+    },
+    {
+      "train_step": 7135,
+      "loss": 0.44437651628036573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44437651628036573
+    },
+    {
+      "train_step": 7136,
+      "loss": 0.5736972498942046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5736972498942046
+    },
+    {
+      "train_step": 7137,
+      "loss": 0.47101806749311653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47101806749311653
+    },
+    {
+      "train_step": 7138,
+      "loss": 0.5677385439753826,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5677385439753826
+    },
+    {
+      "train_step": 7139,
+      "loss": 0.7512376838902034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7512376838902034
+    },
+    {
+      "train_step": 7140,
+      "loss": 0.5005097479005547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5005097479005547
+    },
+    {
+      "train_step": 7141,
+      "loss": 0.4239332907130014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4239332907130014
+    },
+    {
+      "train_step": 7142,
+      "loss": 0.7116004051399287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7116004051399287
+    },
+    {
+      "train_step": 7143,
+      "loss": 0.786950885629416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.786950885629416
+    },
+    {
+      "train_step": 7144,
+      "loss": 0.5955261154676543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5955261154676543
+    },
+    {
+      "train_step": 7145,
+      "loss": 0.4080420929570138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4080420929570138
+    },
+    {
+      "train_step": 7146,
+      "loss": 0.6875639438358361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6875639438358361
+    },
+    {
+      "train_step": 7147,
+      "loss": 0.4852222223633428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4852222223633428
+    },
+    {
+      "train_step": 7148,
+      "loss": 0.5449811382449338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5449811382449338
+    },
+    {
+      "train_step": 7149,
+      "loss": 0.7160488292636554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7160488292636554
+    },
+    {
+      "train_step": 7150,
+      "loss": 0.49517653485651913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49517653485651913
+    },
+    {
+      "train_step": 7151,
+      "loss": 0.5435562267348066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5435562267348066
+    },
+    {
+      "train_step": 7152,
+      "loss": 0.43414073175463724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43414073175463724
+    },
+    {
+      "train_step": 7153,
+      "loss": 0.6926247006381022,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6926247006381022
+    },
+    {
+      "train_step": 7154,
+      "loss": 0.6333464995054116,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6333464995054116
+    },
+    {
+      "train_step": 7155,
+      "loss": 0.7116817976048585,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7116817976048585
+    },
+    {
+      "train_step": 7156,
+      "loss": 0.3966112034646875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3966112034646875
+    },
+    {
+      "train_step": 7157,
+      "loss": 0.6189520212695871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6189520212695871
+    },
+    {
+      "train_step": 7158,
+      "loss": 0.3825859998740508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3825859998740508
+    },
+    {
+      "train_step": 7159,
+      "loss": 0.5848038396756685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5848038396756685
+    },
+    {
+      "train_step": 7160,
+      "loss": 0.6718588803489401,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6718588803489401
+    },
+    {
+      "train_step": 7161,
+      "loss": 0.6722183551440578,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6722183551440578
+    },
+    {
+      "train_step": 7162,
+      "loss": 0.3500767413328106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3500767413328106
+    },
+    {
+      "train_step": 7163,
+      "loss": 0.6288283520952547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6288283520952547
+    },
+    {
+      "train_step": 7164,
+      "loss": 0.542977560082154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.542977560082154
+    },
+    {
+      "train_step": 7165,
+      "loss": 0.5999612305835216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5999612305835216
+    },
+    {
+      "train_step": 7166,
+      "loss": 0.5975094874617445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5975094874617445
+    },
+    {
+      "train_step": 7167,
+      "loss": 0.5221193654420417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5221193654420417
+    },
+    {
+      "train_step": 7168,
+      "loss": 0.6943589756568862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6943589756568862
+    },
+    {
+      "train_step": 7169,
+      "loss": 0.500745320006563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.500745320006563
+    },
+    {
+      "train_step": 7170,
+      "loss": 0.5248303794341472,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5248303794341472
+    },
+    {
+      "train_step": 7171,
+      "loss": 0.511397329874975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.511397329874975
+    },
+    {
+      "train_step": 7172,
+      "loss": 0.5113924121211935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5113924121211935
+    },
+    {
+      "train_step": 7173,
+      "loss": 0.3582222200772035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3582222200772035
+    },
+    {
+      "train_step": 7174,
+      "loss": 0.5734180232602409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5734180232602409
+    },
+    {
+      "train_step": 7175,
+      "loss": 0.5770614088275131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5770614088275131
+    },
+    {
+      "train_step": 7176,
+      "loss": 0.6924511090181369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6924511090181369
+    },
+    {
+      "train_step": 7177,
+      "loss": 0.5466175561291094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5466175561291094
+    },
+    {
+      "train_step": 7178,
+      "loss": 0.6849135616796157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6849135616796157
+    },
+    {
+      "train_step": 7179,
+      "loss": 0.6565659045592109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6565659045592109
+    },
+    {
+      "train_step": 7180,
+      "loss": 0.4498321078131417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4498321078131417
+    },
+    {
+      "train_step": 7181,
+      "loss": 0.5769390139858261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5769390139858261
+    },
+    {
+      "train_step": 7182,
+      "loss": 0.49762868535016525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49762868535016525
+    },
+    {
+      "train_step": 7183,
+      "loss": 0.5027442237913419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5027442237913419
+    },
+    {
+      "train_step": 7184,
+      "loss": 0.48469542830992585,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48469542830992585
+    },
+    {
+      "train_step": 7185,
+      "loss": 0.7338659531412721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7338659531412721
+    },
+    {
+      "train_step": 7186,
+      "loss": 0.6098917780471982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6098917780471982
+    },
+    {
+      "train_step": 7187,
+      "loss": 0.6288304393306211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6288304393306211
+    },
+    {
+      "train_step": 7188,
+      "loss": 0.5027903494484212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5027903494484212
+    },
+    {
+      "train_step": 7189,
+      "loss": 1.0145784822166588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 1.0145784822166588
+    },
+    {
+      "train_step": 7190,
+      "loss": 0.5589746110515756,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5589746110515756
+    },
+    {
+      "train_step": 7191,
+      "loss": 0.6307743675885823,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6307743675885823
+    },
+    {
+      "train_step": 7192,
+      "loss": 0.6319340189314414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6319340189314414
+    },
+    {
+      "train_step": 7193,
+      "loss": 0.5012501896219663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5012501896219663
+    },
+    {
+      "train_step": 7194,
+      "loss": 0.6878118585750881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6878118585750881
+    },
+    {
+      "train_step": 7195,
+      "loss": 0.6562048136051891,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6562048136051891
+    },
+    {
+      "train_step": 7196,
+      "loss": 0.771068892897076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.771068892897076
+    },
+    {
+      "train_step": 7197,
+      "loss": 0.563536924640541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.563536924640541
+    },
+    {
+      "train_step": 7198,
+      "loss": 0.6541659673216684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6541659673216684
+    },
+    {
+      "train_step": 7199,
+      "loss": 0.43317793365710217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43317793365710217
+    },
+    {
+      "train_step": 7200,
+      "loss": 0.6019189350113642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6019189350113642
+    },
+    {
+      "train_step": 7201,
+      "loss": 0.5155617065448946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5155617065448946
+    },
+    {
+      "train_step": 7202,
+      "loss": 0.4864761514705208,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4864761514705208
+    },
+    {
+      "train_step": 7203,
+      "loss": 0.5442387864214642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5442387864214642
+    },
+    {
+      "train_step": 7204,
+      "loss": 0.5042198557786908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5042198557786908
+    },
+    {
+      "train_step": 7205,
+      "loss": 0.5073391918576876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5073391918576876
+    },
+    {
+      "train_step": 7206,
+      "loss": 0.7178279721932755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7178279721932755
+    },
+    {
+      "train_step": 7207,
+      "loss": 0.5354629894752047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5354629894752047
+    },
+    {
+      "train_step": 7208,
+      "loss": 0.7931099958346195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7931099958346195
+    },
+    {
+      "train_step": 7209,
+      "loss": 0.4354728453031012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4354728453031012
+    },
+    {
+      "train_step": 7210,
+      "loss": 0.3727465589847467,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3727465589847467
+    },
+    {
+      "train_step": 7211,
+      "loss": 0.3570136824562301,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3570136824562301
+    },
+    {
+      "train_step": 7212,
+      "loss": 0.6160211518231618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6160211518231618
+    },
+    {
+      "train_step": 7213,
+      "loss": 0.5583121192679844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5583121192679844
+    },
+    {
+      "train_step": 7214,
+      "loss": 0.5446247342748612,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5446247342748612
+    },
+    {
+      "train_step": 7215,
+      "loss": 0.5803086027676778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5803086027676778
+    },
+    {
+      "train_step": 7216,
+      "loss": 0.3652748728798958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3652748728798958
+    },
+    {
+      "train_step": 7217,
+      "loss": 0.5627006041776093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5627006041776093
+    },
+    {
+      "train_step": 7218,
+      "loss": 0.39554488497892826,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39554488497892826
+    },
+    {
+      "train_step": 7219,
+      "loss": 0.59593883952413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.59593883952413
+    },
+    {
+      "train_step": 7220,
+      "loss": 0.5638857518340326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5638857518340326
+    },
+    {
+      "train_step": 7221,
+      "loss": 0.3191595219309973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3191595219309973
+    },
+    {
+      "train_step": 7222,
+      "loss": 0.6502202712991031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6502202712991031
+    },
+    {
+      "train_step": 7223,
+      "loss": 0.5348453098112688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5348453098112688
+    },
+    {
+      "train_step": 7224,
+      "loss": 0.47096078965777133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47096078965777133
+    },
+    {
+      "train_step": 7225,
+      "loss": 0.5465705993292088,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5465705993292088
+    },
+    {
+      "train_step": 7226,
+      "loss": 0.6810245715276992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6810245715276992
+    },
+    {
+      "train_step": 7227,
+      "loss": 0.4427761831133159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4427761831133159
+    },
+    {
+      "train_step": 7228,
+      "loss": 0.6085670292246941,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6085670292246941
+    },
+    {
+      "train_step": 7229,
+      "loss": 0.8251817466024862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8251817466024862
+    },
+    {
+      "train_step": 7230,
+      "loss": 0.3950214728668068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3950214728668068
+    },
+    {
+      "train_step": 7231,
+      "loss": 0.428519941762295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.428519941762295
+    },
+    {
+      "train_step": 7232,
+      "loss": 0.6254696941193243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6254696941193243
+    },
+    {
+      "train_step": 7233,
+      "loss": 0.5833591623464216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5833591623464216
+    },
+    {
+      "train_step": 7234,
+      "loss": 0.5356382436131568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5356382436131568
+    },
+    {
+      "train_step": 7235,
+      "loss": 0.7289521533051787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7289521533051787
+    },
+    {
+      "train_step": 7236,
+      "loss": 0.6714226454359215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6714226454359215
+    },
+    {
+      "train_step": 7237,
+      "loss": 0.5943436277234734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5943436277234734
+    },
+    {
+      "train_step": 7238,
+      "loss": 0.562565229438954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.562565229438954
+    },
+    {
+      "train_step": 7239,
+      "loss": 0.7100877223938188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7100877223938188
+    },
+    {
+      "train_step": 7240,
+      "loss": 0.7782654151371681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7782654151371681
+    },
+    {
+      "train_step": 7241,
+      "loss": 0.7077371152697024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7077371152697024
+    },
+    {
+      "train_step": 7242,
+      "loss": 0.42858838520202375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42858838520202375
+    },
+    {
+      "train_step": 7243,
+      "loss": 0.49781620820268296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49781620820268296
+    },
+    {
+      "train_step": 7244,
+      "loss": 0.6439324754725142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6439324754725142
+    },
+    {
+      "train_step": 7245,
+      "loss": 0.4359268224580304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4359268224580304
+    },
+    {
+      "train_step": 7246,
+      "loss": 0.6506111201524523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6506111201524523
+    },
+    {
+      "train_step": 7247,
+      "loss": 0.6094904973122832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6094904973122832
+    },
+    {
+      "train_step": 7248,
+      "loss": 0.48610772092079996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48610772092079996
+    },
+    {
+      "train_step": 7249,
+      "loss": 0.4005546436069887,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4005546436069887
+    },
+    {
+      "train_step": 7250,
+      "loss": 0.6723053604665121,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6723053604665121
+    },
+    {
+      "train_step": 7251,
+      "loss": 0.5379332676949851,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5379332676949851
+    },
+    {
+      "train_step": 7252,
+      "loss": 0.5253302425687472,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5253302425687472
+    },
+    {
+      "train_step": 7253,
+      "loss": 0.680564754010821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.680564754010821
+    },
+    {
+      "train_step": 7254,
+      "loss": 0.6030366449917954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6030366449917954
+    },
+    {
+      "train_step": 7255,
+      "loss": 0.41902169908025183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41902169908025183
+    },
+    {
+      "train_step": 7256,
+      "loss": 0.501071218885275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.501071218885275
+    },
+    {
+      "train_step": 7257,
+      "loss": 0.7178708070252423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7178708070252423
+    },
+    {
+      "train_step": 7258,
+      "loss": 0.6314404051491505,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6314404051491505
+    },
+    {
+      "train_step": 7259,
+      "loss": 0.48001698902017875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48001698902017875
+    },
+    {
+      "train_step": 7260,
+      "loss": 0.8851888205376021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8851888205376021
+    },
+    {
+      "train_step": 7261,
+      "loss": 0.471838660554694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.471838660554694
+    },
+    {
+      "train_step": 7262,
+      "loss": 0.39186785649072625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39186785649072625
+    },
+    {
+      "train_step": 7263,
+      "loss": 0.5109460111039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5109460111039
+    },
+    {
+      "train_step": 7264,
+      "loss": 0.4893344343325417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4893344343325417
+    },
+    {
+      "train_step": 7265,
+      "loss": 0.6777203742718954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6777203742718954
+    },
+    {
+      "train_step": 7266,
+      "loss": 0.47358119742204174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47358119742204174
+    },
+    {
+      "train_step": 7267,
+      "loss": 0.3434036309866141,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3434036309866141
+    },
+    {
+      "train_step": 7268,
+      "loss": 0.66522266728595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66522266728595
+    },
+    {
+      "train_step": 7269,
+      "loss": 0.5554251361463418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5554251361463418
+    },
+    {
+      "train_step": 7270,
+      "loss": 0.639618670316868,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.639618670316868
+    },
+    {
+      "train_step": 7271,
+      "loss": 0.7432982109595394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7432982109595394
+    },
+    {
+      "train_step": 7272,
+      "loss": 0.4528767970691596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4528767970691596
+    },
+    {
+      "train_step": 7273,
+      "loss": 0.6447997080942307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6447997080942307
+    },
+    {
+      "train_step": 7274,
+      "loss": 0.6286713273049399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6286713273049399
+    },
+    {
+      "train_step": 7275,
+      "loss": 0.4752610984288734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4752610984288734
+    },
+    {
+      "train_step": 7276,
+      "loss": 0.706471635981516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.706471635981516
+    },
+    {
+      "train_step": 7277,
+      "loss": 0.48198572986025195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48198572986025195
+    },
+    {
+      "train_step": 7278,
+      "loss": 0.4778460816378854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4778460816378854
+    },
+    {
+      "train_step": 7279,
+      "loss": 0.7712604516422015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7712604516422015
+    },
+    {
+      "train_step": 7280,
+      "loss": 0.6070901081316268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6070901081316268
+    },
+    {
+      "train_step": 7281,
+      "loss": 0.4899211517892176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4899211517892176
+    },
+    {
+      "train_step": 7282,
+      "loss": 0.5584713789616582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5584713789616582
+    },
+    {
+      "train_step": 7283,
+      "loss": 0.5547671271323441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5547671271323441
+    },
+    {
+      "train_step": 7284,
+      "loss": 0.5669160387963746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5669160387963746
+    },
+    {
+      "train_step": 7285,
+      "loss": 0.7212534708416032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7212534708416032
+    },
+    {
+      "train_step": 7286,
+      "loss": 0.7209993436277485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7209993436277485
+    },
+    {
+      "train_step": 7287,
+      "loss": 0.35458167407821645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35458167407821645
+    },
+    {
+      "train_step": 7288,
+      "loss": 0.7654342852105702,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7654342852105702
+    },
+    {
+      "train_step": 7289,
+      "loss": 0.31224086194854556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31224086194854556
+    },
+    {
+      "train_step": 7290,
+      "loss": 0.5413474537947597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5413474537947597
+    },
+    {
+      "train_step": 7291,
+      "loss": 0.5440400406802203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5440400406802203
+    },
+    {
+      "train_step": 7292,
+      "loss": 0.5544744307839752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5544744307839752
+    },
+    {
+      "train_step": 7293,
+      "loss": 0.4043169169390765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4043169169390765
+    },
+    {
+      "train_step": 7294,
+      "loss": 0.4159758736782913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4159758736782913
+    },
+    {
+      "train_step": 7295,
+      "loss": 0.6053042117515195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6053042117515195
+    },
+    {
+      "train_step": 7296,
+      "loss": 0.4354023294622098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4354023294622098
+    },
+    {
+      "train_step": 7297,
+      "loss": 0.5998524625449052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5998524625449052
+    },
+    {
+      "train_step": 7298,
+      "loss": 0.5433184830208643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5433184830208643
+    },
+    {
+      "train_step": 7299,
+      "loss": 0.5600595599360293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5600595599360293
+    },
+    {
+      "train_step": 7300,
+      "loss": 0.503196717458639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.503196717458639
+    },
+    {
+      "train_step": 7301,
+      "loss": 0.5492549470477828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5492549470477828
+    },
+    {
+      "train_step": 7302,
+      "loss": 0.39539503119492936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39539503119492936
+    },
+    {
+      "train_step": 7303,
+      "loss": 0.6781941152478024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6781941152478024
+    },
+    {
+      "train_step": 7304,
+      "loss": 0.6060187321804573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6060187321804573
+    },
+    {
+      "train_step": 7305,
+      "loss": 0.6296562760487507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6296562760487507
+    },
+    {
+      "train_step": 7306,
+      "loss": 0.3919498347897108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3919498347897108
+    },
+    {
+      "train_step": 7307,
+      "loss": 0.6867627823845939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6867627823845939
+    },
+    {
+      "train_step": 7308,
+      "loss": 0.5661695352839149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5661695352839149
+    },
+    {
+      "train_step": 7309,
+      "loss": 0.7333835474078534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7333835474078534
+    },
+    {
+      "train_step": 7310,
+      "loss": 0.5568335046535074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5568335046535074
+    },
+    {
+      "train_step": 7311,
+      "loss": 0.517193367137917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.517193367137917
+    },
+    {
+      "train_step": 7312,
+      "loss": 0.5183304410371568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5183304410371568
+    },
+    {
+      "train_step": 7313,
+      "loss": 0.6681695796926703,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6681695796926703
+    },
+    {
+      "train_step": 7314,
+      "loss": 0.5985599587007601,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5985599587007601
+    },
+    {
+      "train_step": 7315,
+      "loss": 0.5732712005701432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5732712005701432
+    },
+    {
+      "train_step": 7316,
+      "loss": 0.592972926979338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.592972926979338
+    },
+    {
+      "train_step": 7317,
+      "loss": 0.7371234991529343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7371234991529343
+    },
+    {
+      "train_step": 7318,
+      "loss": 0.4626859222764813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4626859222764813
+    },
+    {
+      "train_step": 7319,
+      "loss": 0.5727652611906564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5727652611906564
+    },
+    {
+      "train_step": 7320,
+      "loss": 0.6539114722660643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6539114722660643
+    },
+    {
+      "train_step": 7321,
+      "loss": 0.3821041584750812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3821041584750812
+    },
+    {
+      "train_step": 7322,
+      "loss": 0.573825516889109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.573825516889109
+    },
+    {
+      "train_step": 7323,
+      "loss": 0.4259295827943684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4259295827943684
+    },
+    {
+      "train_step": 7324,
+      "loss": 0.4087452737311229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4087452737311229
+    },
+    {
+      "train_step": 7325,
+      "loss": 0.5701536151451361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5701536151451361
+    },
+    {
+      "train_step": 7326,
+      "loss": 0.5765741190044579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5765741190044579
+    },
+    {
+      "train_step": 7327,
+      "loss": 0.6112089911222833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6112089911222833
+    },
+    {
+      "train_step": 7328,
+      "loss": 0.5165148263012009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5165148263012009
+    },
+    {
+      "train_step": 7329,
+      "loss": 0.6503326209042238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6503326209042238
+    },
+    {
+      "train_step": 7330,
+      "loss": 0.4321327393084688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4321327393084688
+    },
+    {
+      "train_step": 7331,
+      "loss": 0.5396866817049523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5396866817049523
+    },
+    {
+      "train_step": 7332,
+      "loss": 0.518321082859074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.518321082859074
+    },
+    {
+      "train_step": 7333,
+      "loss": 0.6302598454399584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6302598454399584
+    },
+    {
+      "train_step": 7334,
+      "loss": 0.5265642895828742,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5265642895828742
+    },
+    {
+      "train_step": 7335,
+      "loss": 0.586326963065726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.586326963065726
+    },
+    {
+      "train_step": 7336,
+      "loss": 0.294039493321507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.294039493321507
+    },
+    {
+      "train_step": 7337,
+      "loss": 0.49999760872713556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49999760872713556
+    },
+    {
+      "train_step": 7338,
+      "loss": 0.64223356895521,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.64223356895521
+    },
+    {
+      "train_step": 7339,
+      "loss": 0.6576510439228532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6576510439228532
+    },
+    {
+      "train_step": 7340,
+      "loss": 0.5416512333061496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5416512333061496
+    },
+    {
+      "train_step": 7341,
+      "loss": 0.5152951054748554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5152951054748554
+    },
+    {
+      "train_step": 7342,
+      "loss": 0.41067975613608293,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41067975613608293
+    },
+    {
+      "train_step": 7343,
+      "loss": 0.3422036093694465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3422036093694465
+    },
+    {
+      "train_step": 7344,
+      "loss": 0.679175698454049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.679175698454049
+    },
+    {
+      "train_step": 7345,
+      "loss": 0.3896197241595244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3896197241595244
+    },
+    {
+      "train_step": 7346,
+      "loss": 0.5118826656590599,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5118826656590599
+    },
+    {
+      "train_step": 7347,
+      "loss": 0.6058273029372637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6058273029372637
+    },
+    {
+      "train_step": 7348,
+      "loss": 0.4428739252680409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4428739252680409
+    },
+    {
+      "train_step": 7349,
+      "loss": 0.7309002687828804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7309002687828804
+    },
+    {
+      "train_step": 7350,
+      "loss": 0.676602625444245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.676602625444245
+    },
+    {
+      "train_step": 7351,
+      "loss": 0.6337721468105915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6337721468105915
+    },
+    {
+      "train_step": 7352,
+      "loss": 0.4878704458920917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4878704458920917
+    },
+    {
+      "train_step": 7353,
+      "loss": 0.46937087360361796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46937087360361796
+    },
+    {
+      "train_step": 7354,
+      "loss": 0.35940997746949566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35940997746949566
+    },
+    {
+      "train_step": 7355,
+      "loss": 0.5088431262526864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5088431262526864
+    },
+    {
+      "train_step": 7356,
+      "loss": 0.6688159259638884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6688159259638884
+    },
+    {
+      "train_step": 7357,
+      "loss": 0.5879508699649447,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5879508699649447
+    },
+    {
+      "train_step": 7358,
+      "loss": 0.6726992964180357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6726992964180357
+    },
+    {
+      "train_step": 7359,
+      "loss": 0.5094504372476134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5094504372476134
+    },
+    {
+      "train_step": 7360,
+      "loss": 0.6262766007769054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6262766007769054
+    },
+    {
+      "train_step": 7361,
+      "loss": 0.614097710053681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.614097710053681
+    },
+    {
+      "train_step": 7362,
+      "loss": 0.6289922697083459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6289922697083459
+    },
+    {
+      "train_step": 7363,
+      "loss": 0.6662013536575014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6662013536575014
+    },
+    {
+      "train_step": 7364,
+      "loss": 0.6510453745626347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6510453745626347
+    },
+    {
+      "train_step": 7365,
+      "loss": 0.7584449361756435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7584449361756435
+    },
+    {
+      "train_step": 7366,
+      "loss": 0.5364211117522549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5364211117522549
+    },
+    {
+      "train_step": 7367,
+      "loss": 0.56024870800399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.56024870800399
+    },
+    {
+      "train_step": 7368,
+      "loss": 0.571792977428099,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.571792977428099
+    },
+    {
+      "train_step": 7369,
+      "loss": 0.7201574550526924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7201574550526924
+    },
+    {
+      "train_step": 7370,
+      "loss": 0.7199080039813224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7199080039813224
+    },
+    {
+      "train_step": 7371,
+      "loss": 0.5736812375308353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5736812375308353
+    },
+    {
+      "train_step": 7372,
+      "loss": 0.39733571575458926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39733571575458926
+    },
+    {
+      "train_step": 7373,
+      "loss": 0.5314297003172007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5314297003172007
+    },
+    {
+      "train_step": 7374,
+      "loss": 0.6704099842178446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6704099842178446
+    },
+    {
+      "train_step": 7375,
+      "loss": 0.4968732495897843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4968732495897843
+    },
+    {
+      "train_step": 7376,
+      "loss": 0.5396453905242813,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5396453905242813
+    },
+    {
+      "train_step": 7377,
+      "loss": 0.49513517092746007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49513517092746007
+    },
+    {
+      "train_step": 7378,
+      "loss": 0.5786858094668034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5786858094668034
+    },
+    {
+      "train_step": 7379,
+      "loss": 0.49905615830210853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49905615830210853
+    },
+    {
+      "train_step": 7380,
+      "loss": 0.579278451557305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.579278451557305
+    },
+    {
+      "train_step": 7381,
+      "loss": 0.5398818152679375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5398818152679375
+    },
+    {
+      "train_step": 7382,
+      "loss": 0.5047044123134395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5047044123134395
+    },
+    {
+      "train_step": 7383,
+      "loss": 0.6133811530586245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6133811530586245
+    },
+    {
+      "train_step": 7384,
+      "loss": 0.8462482063507397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8462482063507397
+    },
+    {
+      "train_step": 7385,
+      "loss": 0.5716538285218401,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5716538285218401
+    },
+    {
+      "train_step": 7386,
+      "loss": 0.5249233011058475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5249233011058475
+    },
+    {
+      "train_step": 7387,
+      "loss": 0.5349154552036517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5349154552036517
+    },
+    {
+      "train_step": 7388,
+      "loss": 0.6688551940148926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6688551940148926
+    },
+    {
+      "train_step": 7389,
+      "loss": 0.5426559101161638,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5426559101161638
+    },
+    {
+      "train_step": 7390,
+      "loss": 0.4987304870101569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4987304870101569
+    },
+    {
+      "train_step": 7391,
+      "loss": 0.6332343807637034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6332343807637034
+    },
+    {
+      "train_step": 7392,
+      "loss": 0.6453589951482737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6453589951482737
+    },
+    {
+      "train_step": 7393,
+      "loss": 0.4556999059262704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4556999059262704
+    },
+    {
+      "train_step": 7394,
+      "loss": 0.6429710925329926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6429710925329926
+    },
+    {
+      "train_step": 7395,
+      "loss": 0.5583656465428135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5583656465428135
+    },
+    {
+      "train_step": 7396,
+      "loss": 0.5200658334030224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5200658334030224
+    },
+    {
+      "train_step": 7397,
+      "loss": 0.5721701922658214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5721701922658214
+    },
+    {
+      "train_step": 7398,
+      "loss": 0.568606808441686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.568606808441686
+    },
+    {
+      "train_step": 7399,
+      "loss": 0.8764189558946416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8764189558946416
+    },
+    {
+      "train_step": 7400,
+      "loss": 0.494902192698869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.494902192698869
+    },
+    {
+      "train_step": 7401,
+      "loss": 0.5264876498403677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5264876498403677
+    },
+    {
+      "train_step": 7402,
+      "loss": 0.5893046343838041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5893046343838041
+    },
+    {
+      "train_step": 7403,
+      "loss": 0.5141751034283571,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5141751034283571
+    },
+    {
+      "train_step": 7404,
+      "loss": 0.47155088933778416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47155088933778416
+    },
+    {
+      "train_step": 7405,
+      "loss": 0.5731130146095436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5731130146095436
+    },
+    {
+      "train_step": 7406,
+      "loss": 0.40108513585309846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40108513585309846
+    },
+    {
+      "train_step": 7407,
+      "loss": 0.6083554977186123,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6083554977186123
+    },
+    {
+      "train_step": 7408,
+      "loss": 0.5420312174508527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5420312174508527
+    },
+    {
+      "train_step": 7409,
+      "loss": 0.6647905671907409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6647905671907409
+    },
+    {
+      "train_step": 7410,
+      "loss": 0.3988920527812899,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3988920527812899
+    },
+    {
+      "train_step": 7411,
+      "loss": 0.49093269685929775,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49093269685929775
+    },
+    {
+      "train_step": 7412,
+      "loss": 0.5908010708861446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5908010708861446
+    },
+    {
+      "train_step": 7413,
+      "loss": 0.5211154386261639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5211154386261639
+    },
+    {
+      "train_step": 7414,
+      "loss": 0.5186375845130027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5186375845130027
+    },
+    {
+      "train_step": 7415,
+      "loss": 0.4769129854083745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4769129854083745
+    },
+    {
+      "train_step": 7416,
+      "loss": 0.5599556614547654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5599556614547654
+    },
+    {
+      "train_step": 7417,
+      "loss": 0.4906690331681888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4906690331681888
+    },
+    {
+      "train_step": 7418,
+      "loss": 0.44482808717807376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44482808717807376
+    },
+    {
+      "train_step": 7419,
+      "loss": 0.7418373745963287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7418373745963287
+    },
+    {
+      "train_step": 7420,
+      "loss": 0.5658794475355419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5658794475355419
+    },
+    {
+      "train_step": 7421,
+      "loss": 0.7248598995550187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7248598995550187
+    },
+    {
+      "train_step": 7422,
+      "loss": 0.544715237497755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.544715237497755
+    },
+    {
+      "train_step": 7423,
+      "loss": 0.6383151451153348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6383151451153348
+    },
+    {
+      "train_step": 7424,
+      "loss": 0.562304770396836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.562304770396836
+    },
+    {
+      "train_step": 7425,
+      "loss": 0.6007526025906882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6007526025906882
+    },
+    {
+      "train_step": 7426,
+      "loss": 0.6116202121210366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6116202121210366
+    },
+    {
+      "train_step": 7427,
+      "loss": 0.5193339725293573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5193339725293573
+    },
+    {
+      "train_step": 7428,
+      "loss": 0.48076608681797184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48076608681797184
+    },
+    {
+      "train_step": 7429,
+      "loss": 0.797114569702998,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.797114569702998
+    },
+    {
+      "train_step": 7430,
+      "loss": 0.3338839370528033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3338839370528033
+    },
+    {
+      "train_step": 7431,
+      "loss": 0.47512334980077864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47512334980077864
+    },
+    {
+      "train_step": 7432,
+      "loss": 0.5941450246257391,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5941450246257391
+    },
+    {
+      "train_step": 7433,
+      "loss": 0.5509619661270629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5509619661270629
+    },
+    {
+      "train_step": 7434,
+      "loss": 0.3882495589356763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3882495589356763
+    },
+    {
+      "train_step": 7435,
+      "loss": 0.3906633541783668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3906633541783668
+    },
+    {
+      "train_step": 7436,
+      "loss": 0.55631274237494,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.55631274237494
+    },
+    {
+      "train_step": 7437,
+      "loss": 0.47343530706337905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47343530706337905
+    },
+    {
+      "train_step": 7438,
+      "loss": 0.5304225866360728,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5304225866360728
+    },
+    {
+      "train_step": 7439,
+      "loss": 0.6514962174975134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6514962174975134
+    },
+    {
+      "train_step": 7440,
+      "loss": 0.5800917876450334,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5800917876450334
+    },
+    {
+      "train_step": 7441,
+      "loss": 0.5466335438861278,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5466335438861278
+    },
+    {
+      "train_step": 7442,
+      "loss": 0.5432748499685253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5432748499685253
+    },
+    {
+      "train_step": 7443,
+      "loss": 0.5441467005709133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441467005709133
+    },
+    {
+      "train_step": 7444,
+      "loss": 0.4342187160833459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4342187160833459
+    },
+    {
+      "train_step": 7445,
+      "loss": 0.3987796615144797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3987796615144797
+    },
+    {
+      "train_step": 7446,
+      "loss": 0.6499658395548562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6499658395548562
+    },
+    {
+      "train_step": 7447,
+      "loss": 0.6718504019587741,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6718504019587741
+    },
+    {
+      "train_step": 7448,
+      "loss": 0.49951909230170904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49951909230170904
+    },
+    {
+      "train_step": 7449,
+      "loss": 0.6425244892309516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6425244892309516
+    },
+    {
+      "train_step": 7450,
+      "loss": 0.5633159087811539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5633159087811539
+    },
+    {
+      "train_step": 7451,
+      "loss": 0.6038913092810709,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6038913092810709
+    },
+    {
+      "train_step": 7452,
+      "loss": 0.2792532523723014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2792532523723014
+    },
+    {
+      "train_step": 7453,
+      "loss": 0.5328496952417342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5328496952417342
+    },
+    {
+      "train_step": 7454,
+      "loss": 0.5982805442065388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5982805442065388
+    },
+    {
+      "train_step": 7455,
+      "loss": 0.3934075824200557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3934075824200557
+    },
+    {
+      "train_step": 7456,
+      "loss": 0.4501398369047881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4501398369047881
+    },
+    {
+      "train_step": 7457,
+      "loss": 0.45761073807442193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45761073807442193
+    },
+    {
+      "train_step": 7458,
+      "loss": 0.45696002278329495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45696002278329495
+    },
+    {
+      "train_step": 7459,
+      "loss": 0.6126701852587308,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6126701852587308
+    },
+    {
+      "train_step": 7460,
+      "loss": 0.5320681854274042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5320681854274042
+    },
+    {
+      "train_step": 7461,
+      "loss": 0.5743394924686702,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5743394924686702
+    },
+    {
+      "train_step": 7462,
+      "loss": 0.41222876490622645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41222876490622645
+    },
+    {
+      "train_step": 7463,
+      "loss": 0.5629107199212006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5629107199212006
+    },
+    {
+      "train_step": 7464,
+      "loss": 0.6103429712550701,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6103429712550701
+    },
+    {
+      "train_step": 7465,
+      "loss": 0.45041330079263514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45041330079263514
+    },
+    {
+      "train_step": 7466,
+      "loss": 0.5962543069924937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962543069924937
+    },
+    {
+      "train_step": 7467,
+      "loss": 0.4214319053486552,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4214319053486552
+    },
+    {
+      "train_step": 7468,
+      "loss": 0.3955004564634905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3955004564634905
+    },
+    {
+      "train_step": 7469,
+      "loss": 0.534171620101033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.534171620101033
+    },
+    {
+      "train_step": 7470,
+      "loss": 0.7292665343157518,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7292665343157518
+    },
+    {
+      "train_step": 7471,
+      "loss": 0.3791897266720836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3791897266720836
+    },
+    {
+      "train_step": 7472,
+      "loss": 0.5171850418633599,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5171850418633599
+    },
+    {
+      "train_step": 7473,
+      "loss": 0.5504605703524214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5504605703524214
+    },
+    {
+      "train_step": 7474,
+      "loss": 0.4318778546568744,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4318778546568744
+    },
+    {
+      "train_step": 7475,
+      "loss": 0.530799035530318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.530799035530318
+    },
+    {
+      "train_step": 7476,
+      "loss": 0.5160521079406054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5160521079406054
+    },
+    {
+      "train_step": 7477,
+      "loss": 0.6070275897863946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6070275897863946
+    },
+    {
+      "train_step": 7478,
+      "loss": 0.6689283433727273,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6689283433727273
+    },
+    {
+      "train_step": 7479,
+      "loss": 0.5523626987050209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5523626987050209
+    },
+    {
+      "train_step": 7480,
+      "loss": 0.5318200276982468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5318200276982468
+    },
+    {
+      "train_step": 7481,
+      "loss": 0.7095747254769071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7095747254769071
+    },
+    {
+      "train_step": 7482,
+      "loss": 0.4821207976480395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4821207976480395
+    },
+    {
+      "train_step": 7483,
+      "loss": 0.6360115842648798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6360115842648798
+    },
+    {
+      "train_step": 7484,
+      "loss": 0.5875844194670339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5875844194670339
+    },
+    {
+      "train_step": 7485,
+      "loss": 0.5170520586004398,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5170520586004398
+    },
+    {
+      "train_step": 7486,
+      "loss": 0.5054264369074161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5054264369074161
+    },
+    {
+      "train_step": 7487,
+      "loss": 0.5785554136237441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5785554136237441
+    },
+    {
+      "train_step": 7488,
+      "loss": 0.5466610574018604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5466610574018604
+    },
+    {
+      "train_step": 7489,
+      "loss": 0.5905744485942729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5905744485942729
+    },
+    {
+      "train_step": 7490,
+      "loss": 0.7246237584593485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7246237584593485
+    },
+    {
+      "train_step": 7491,
+      "loss": 0.4888809464625444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4888809464625444
+    },
+    {
+      "train_step": 7492,
+      "loss": 0.6576559650170644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6576559650170644
+    },
+    {
+      "train_step": 7493,
+      "loss": 0.697096560965738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.697096560965738
+    },
+    {
+      "train_step": 7494,
+      "loss": 0.6642237073616064,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6642237073616064
+    },
+    {
+      "train_step": 7495,
+      "loss": 0.6067153069434397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6067153069434397
+    },
+    {
+      "train_step": 7496,
+      "loss": 0.6528220893961332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6528220893961332
+    },
+    {
+      "train_step": 7497,
+      "loss": 0.6961004456067476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6961004456067476
+    },
+    {
+      "train_step": 7498,
+      "loss": 0.6579729218393411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6579729218393411
+    },
+    {
+      "train_step": 7499,
+      "loss": 0.7037676725826156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7037676725826156
+    },
+    {
+      "train_step": 7500,
+      "loss": 0.5875825547602831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5875825547602831
+    },
+    {
+      "train_step": 7501,
+      "loss": 0.6801586432517369,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6801586432517369
+    },
+    {
+      "train_step": 7502,
+      "loss": 0.45772756583210433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45772756583210433
+    },
+    {
+      "train_step": 7503,
+      "loss": 0.5110120337632305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5110120337632305
+    },
+    {
+      "train_step": 7504,
+      "loss": 0.46236233204992966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46236233204992966
+    },
+    {
+      "train_step": 7505,
+      "loss": 0.5330412096579603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5330412096579603
+    },
+    {
+      "train_step": 7506,
+      "loss": 0.5446592352246951,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5446592352246951
+    },
+    {
+      "train_step": 7507,
+      "loss": 0.48460722811996104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48460722811996104
+    },
+    {
+      "train_step": 7508,
+      "loss": 0.4517953763034675,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4517953763034675
+    },
+    {
+      "train_step": 7509,
+      "loss": 0.575551211886658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.575551211886658
+    },
+    {
+      "train_step": 7510,
+      "loss": 0.6096450578990738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6096450578990738
+    },
+    {
+      "train_step": 7511,
+      "loss": 0.428518285951291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.428518285951291
+    },
+    {
+      "train_step": 7512,
+      "loss": 0.5577267772926686,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5577267772926686
+    },
+    {
+      "train_step": 7513,
+      "loss": 0.6723092675366186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6723092675366186
+    },
+    {
+      "train_step": 7514,
+      "loss": 0.6150880215482522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6150880215482522
+    },
+    {
+      "train_step": 7515,
+      "loss": 0.5292492863475871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5292492863475871
+    },
+    {
+      "train_step": 7516,
+      "loss": 0.5197512217372421,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5197512217372421
+    },
+    {
+      "train_step": 7517,
+      "loss": 0.5470924030233103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5470924030233103
+    },
+    {
+      "train_step": 7518,
+      "loss": 0.5308143780030757,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5308143780030757
+    },
+    {
+      "train_step": 7519,
+      "loss": 0.4887502512918239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4887502512918239
+    },
+    {
+      "train_step": 7520,
+      "loss": 0.5932364007094302,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5932364007094302
+    },
+    {
+      "train_step": 7521,
+      "loss": 0.5103296557079948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5103296557079948
+    },
+    {
+      "train_step": 7522,
+      "loss": 0.685885679458158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.685885679458158
+    },
+    {
+      "train_step": 7523,
+      "loss": 0.4204631999163795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4204631999163795
+    },
+    {
+      "train_step": 7524,
+      "loss": 0.479042244748206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.479042244748206
+    },
+    {
+      "train_step": 7525,
+      "loss": 0.8998033458017087,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8998033458017087
+    },
+    {
+      "train_step": 7526,
+      "loss": 0.5244803442728314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5244803442728314
+    },
+    {
+      "train_step": 7527,
+      "loss": 0.6658807431674584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6658807431674584
+    },
+    {
+      "train_step": 7528,
+      "loss": 0.6485357637004985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6485357637004985
+    },
+    {
+      "train_step": 7529,
+      "loss": 0.6757626404674499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6757626404674499
+    },
+    {
+      "train_step": 7530,
+      "loss": 0.5591605674628742,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5591605674628742
+    },
+    {
+      "train_step": 7531,
+      "loss": 0.5573105726690337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5573105726690337
+    },
+    {
+      "train_step": 7532,
+      "loss": 0.5040961969142722,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5040961969142722
+    },
+    {
+      "train_step": 7533,
+      "loss": 0.683782510735213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.683782510735213
+    },
+    {
+      "train_step": 7534,
+      "loss": 0.3915680832108883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3915680832108883
+    },
+    {
+      "train_step": 7535,
+      "loss": 0.5737892339029516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5737892339029516
+    },
+    {
+      "train_step": 7536,
+      "loss": 0.35811891587205197,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35811891587205197
+    },
+    {
+      "train_step": 7537,
+      "loss": 0.5295058268246279,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5295058268246279
+    },
+    {
+      "train_step": 7538,
+      "loss": 0.6534247773453635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6534247773453635
+    },
+    {
+      "train_step": 7539,
+      "loss": 0.5155400959782684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5155400959782684
+    },
+    {
+      "train_step": 7540,
+      "loss": 0.4994974647652843,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4994974647652843
+    },
+    {
+      "train_step": 7541,
+      "loss": 0.4870871917935629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4870871917935629
+    },
+    {
+      "train_step": 7542,
+      "loss": 0.612869933132471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.612869933132471
+    },
+    {
+      "train_step": 7543,
+      "loss": 0.4453212185283261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4453212185283261
+    },
+    {
+      "train_step": 7544,
+      "loss": 0.6919533680248515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6919533680248515
+    },
+    {
+      "train_step": 7545,
+      "loss": 0.4117353138563916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4117353138563916
+    },
+    {
+      "train_step": 7546,
+      "loss": 0.4881916660633666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4881916660633666
+    },
+    {
+      "train_step": 7547,
+      "loss": 0.7589460329992425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7589460329992425
+    },
+    {
+      "train_step": 7548,
+      "loss": 0.4432039860438043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4432039860438043
+    },
+    {
+      "train_step": 7549,
+      "loss": 0.47631135983629713,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47631135983629713
+    },
+    {
+      "train_step": 7550,
+      "loss": 0.6015395016128403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6015395016128403
+    },
+    {
+      "train_step": 7551,
+      "loss": 0.5231357080135902,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5231357080135902
+    },
+    {
+      "train_step": 7552,
+      "loss": 0.5180463949032662,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5180463949032662
+    },
+    {
+      "train_step": 7553,
+      "loss": 0.5356161106911611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5356161106911611
+    },
+    {
+      "train_step": 7554,
+      "loss": 0.4918850559547372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4918850559547372
+    },
+    {
+      "train_step": 7555,
+      "loss": 0.6716166626576225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6716166626576225
+    },
+    {
+      "train_step": 7556,
+      "loss": 0.609222289583238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.609222289583238
+    },
+    {
+      "train_step": 7557,
+      "loss": 0.8510895712820228,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8510895712820228
+    },
+    {
+      "train_step": 7558,
+      "loss": 0.3989035590848673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3989035590848673
+    },
+    {
+      "train_step": 7559,
+      "loss": 0.6117002732729965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6117002732729965
+    },
+    {
+      "train_step": 7560,
+      "loss": 0.6313488048919476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6313488048919476
+    },
+    {
+      "train_step": 7561,
+      "loss": 0.695687597269243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.695687597269243
+    },
+    {
+      "train_step": 7562,
+      "loss": 0.6710930350998516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6710930350998516
+    },
+    {
+      "train_step": 7563,
+      "loss": 0.6822460311133994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6822460311133994
+    },
+    {
+      "train_step": 7564,
+      "loss": 0.4634479262398228,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4634479262398228
+    },
+    {
+      "train_step": 7565,
+      "loss": 0.49966173145740755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49966173145740755
+    },
+    {
+      "train_step": 7566,
+      "loss": 0.8606992788227827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8606992788227827
+    },
+    {
+      "train_step": 7567,
+      "loss": 0.4758315245625944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4758315245625944
+    },
+    {
+      "train_step": 7568,
+      "loss": 0.4500897522351466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4500897522351466
+    },
+    {
+      "train_step": 7569,
+      "loss": 0.5005441153809003,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5005441153809003
+    },
+    {
+      "train_step": 7570,
+      "loss": 0.42001972497318424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42001972497318424
+    },
+    {
+      "train_step": 7571,
+      "loss": 0.6813198163895673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6813198163895673
+    },
+    {
+      "train_step": 7572,
+      "loss": 0.6149584536199442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6149584536199442
+    },
+    {
+      "train_step": 7573,
+      "loss": 0.500245177734089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.500245177734089
+    },
+    {
+      "train_step": 7574,
+      "loss": 0.5055172890959425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5055172890959425
+    },
+    {
+      "train_step": 7575,
+      "loss": 0.5189426545222537,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5189426545222537
+    },
+    {
+      "train_step": 7576,
+      "loss": 0.41885626139787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41885626139787
+    },
+    {
+      "train_step": 7577,
+      "loss": 0.47356073631852524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47356073631852524
+    },
+    {
+      "train_step": 7578,
+      "loss": 0.7105114736908844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7105114736908844
+    },
+    {
+      "train_step": 7579,
+      "loss": 0.5164719488561255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5164719488561255
+    },
+    {
+      "train_step": 7580,
+      "loss": 0.7124534522224021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7124534522224021
+    },
+    {
+      "train_step": 7581,
+      "loss": 0.6646047063445006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6646047063445006
+    },
+    {
+      "train_step": 7582,
+      "loss": 0.7498130580370399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7498130580370399
+    },
+    {
+      "train_step": 7583,
+      "loss": 0.5230204475511356,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5230204475511356
+    },
+    {
+      "train_step": 7584,
+      "loss": 0.6786256301137795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6786256301137795
+    },
+    {
+      "train_step": 7585,
+      "loss": 0.4747877147650573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4747877147650573
+    },
+    {
+      "train_step": 7586,
+      "loss": 0.462648273044073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.462648273044073
+    },
+    {
+      "train_step": 7587,
+      "loss": 0.6482883243884725,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6482883243884725
+    },
+    {
+      "train_step": 7588,
+      "loss": 0.6368590732707821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6368590732707821
+    },
+    {
+      "train_step": 7589,
+      "loss": 0.46685182287753174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46685182287753174
+    },
+    {
+      "train_step": 7590,
+      "loss": 0.6327675045829606,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6327675045829606
+    },
+    {
+      "train_step": 7591,
+      "loss": 0.5504577173781046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5504577173781046
+    },
+    {
+      "train_step": 7592,
+      "loss": 0.44887163447802036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44887163447802036
+    },
+    {
+      "train_step": 7593,
+      "loss": 0.46245079942589457,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46245079942589457
+    },
+    {
+      "train_step": 7594,
+      "loss": 0.47345565911508547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47345565911508547
+    },
+    {
+      "train_step": 7595,
+      "loss": 0.6759248816388846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6759248816388846
+    },
+    {
+      "train_step": 7596,
+      "loss": 0.5800056057425615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5800056057425615
+    },
+    {
+      "train_step": 7597,
+      "loss": 0.42337876247669953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42337876247669953
+    },
+    {
+      "train_step": 7598,
+      "loss": 0.8815418002447284,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8815418002447284
+    },
+    {
+      "train_step": 7599,
+      "loss": 0.5246567770641509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5246567770641509
+    },
+    {
+      "train_step": 7600,
+      "loss": 0.6224258942344403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6224258942344403
+    },
+    {
+      "train_step": 7601,
+      "loss": 0.6350388028065365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6350388028065365
+    },
+    {
+      "train_step": 7602,
+      "loss": 0.7036643100688549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7036643100688549
+    },
+    {
+      "train_step": 7603,
+      "loss": 0.8349035127488547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8349035127488547
+    },
+    {
+      "train_step": 7604,
+      "loss": 0.5764309654341254,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5764309654341254
+    },
+    {
+      "train_step": 7605,
+      "loss": 0.5276966856475991,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5276966856475991
+    },
+    {
+      "train_step": 7606,
+      "loss": 0.35384058682980873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35384058682980873
+    },
+    {
+      "train_step": 7607,
+      "loss": 0.5334765563567102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5334765563567102
+    },
+    {
+      "train_step": 7608,
+      "loss": 0.6048443643296207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6048443643296207
+    },
+    {
+      "train_step": 7609,
+      "loss": 0.6638701174901338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6638701174901338
+    },
+    {
+      "train_step": 7610,
+      "loss": 0.6328438218889959,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6328438218889959
+    },
+    {
+      "train_step": 7611,
+      "loss": 0.6432192040593648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6432192040593648
+    },
+    {
+      "train_step": 7612,
+      "loss": 0.6595537905700674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6595537905700674
+    },
+    {
+      "train_step": 7613,
+      "loss": 0.6575261723526903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6575261723526903
+    },
+    {
+      "train_step": 7614,
+      "loss": 0.6696031548926379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6696031548926379
+    },
+    {
+      "train_step": 7615,
+      "loss": 0.5955047548886783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5955047548886783
+    },
+    {
+      "train_step": 7616,
+      "loss": 0.4968148955788489,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4968148955788489
+    },
+    {
+      "train_step": 7617,
+      "loss": 0.6041883270245922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6041883270245922
+    },
+    {
+      "train_step": 7618,
+      "loss": 0.6782616237015903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6782616237015903
+    },
+    {
+      "train_step": 7619,
+      "loss": 0.3542520514716988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3542520514716988
+    },
+    {
+      "train_step": 7620,
+      "loss": 0.47299277310299065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47299277310299065
+    },
+    {
+      "train_step": 7621,
+      "loss": 0.5047325845219027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5047325845219027
+    },
+    {
+      "train_step": 7622,
+      "loss": 0.5784485681756679,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5784485681756679
+    },
+    {
+      "train_step": 7623,
+      "loss": 0.41484389109170844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41484389109170844
+    },
+    {
+      "train_step": 7624,
+      "loss": 0.41439141172106114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41439141172106114
+    },
+    {
+      "train_step": 7625,
+      "loss": 0.41212044725802427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41212044725802427
+    },
+    {
+      "train_step": 7626,
+      "loss": 0.546355701620505,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.546355701620505
+    },
+    {
+      "train_step": 7627,
+      "loss": 0.60810099097818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.60810099097818
+    },
+    {
+      "train_step": 7628,
+      "loss": 0.5691197503041171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5691197503041171
+    },
+    {
+      "train_step": 7629,
+      "loss": 0.6488641141770785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6488641141770785
+    },
+    {
+      "train_step": 7630,
+      "loss": 0.7664715851255693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7664715851255693
+    },
+    {
+      "train_step": 7631,
+      "loss": 0.5722970666680796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5722970666680796
+    },
+    {
+      "train_step": 7632,
+      "loss": 0.6107336573612246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6107336573612246
+    },
+    {
+      "train_step": 7633,
+      "loss": 0.6013823799435141,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6013823799435141
+    },
+    {
+      "train_step": 7634,
+      "loss": 0.5668705127960797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5668705127960797
+    },
+    {
+      "train_step": 7635,
+      "loss": 0.5963000881117213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5963000881117213
+    },
+    {
+      "train_step": 7636,
+      "loss": 0.5928655465285154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5928655465285154
+    },
+    {
+      "train_step": 7637,
+      "loss": 0.6889860432925824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6889860432925824
+    },
+    {
+      "train_step": 7638,
+      "loss": 0.49084340166350654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49084340166350654
+    },
+    {
+      "train_step": 7639,
+      "loss": 0.8983970145209487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8983970145209487
+    },
+    {
+      "train_step": 7640,
+      "loss": 0.5802941699850852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5802941699850852
+    },
+    {
+      "train_step": 7641,
+      "loss": 0.7130956099249858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7130956099249858
+    },
+    {
+      "train_step": 7642,
+      "loss": 0.4690951344539501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4690951344539501
+    },
+    {
+      "train_step": 7643,
+      "loss": 0.6731907434419229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6731907434419229
+    },
+    {
+      "train_step": 7644,
+      "loss": 0.4304946899325508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4304946899325508
+    },
+    {
+      "train_step": 7645,
+      "loss": 0.6755561505064585,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6755561505064585
+    },
+    {
+      "train_step": 7646,
+      "loss": 0.44612431515588974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44612431515588974
+    },
+    {
+      "train_step": 7647,
+      "loss": 0.5355773133166557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5355773133166557
+    },
+    {
+      "train_step": 7648,
+      "loss": 0.39506227633676566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39506227633676566
+    },
+    {
+      "train_step": 7649,
+      "loss": 0.5369338866821314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5369338866821314
+    },
+    {
+      "train_step": 7650,
+      "loss": 0.5694873196976796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5694873196976796
+    },
+    {
+      "train_step": 7651,
+      "loss": 0.6846431398057116,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6846431398057116
+    },
+    {
+      "train_step": 7652,
+      "loss": 0.5006310131241248,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5006310131241248
+    },
+    {
+      "train_step": 7653,
+      "loss": 0.5906616773370809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5906616773370809
+    },
+    {
+      "train_step": 7654,
+      "loss": 0.6320061416726094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6320061416726094
+    },
+    {
+      "train_step": 7655,
+      "loss": 0.5373375372869519,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5373375372869519
+    },
+    {
+      "train_step": 7656,
+      "loss": 0.6351469163307377,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6351469163307377
+    },
+    {
+      "train_step": 7657,
+      "loss": 0.5509729723490265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5509729723490265
+    },
+    {
+      "train_step": 7658,
+      "loss": 0.5300611842017535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5300611842017535
+    },
+    {
+      "train_step": 7659,
+      "loss": 0.6170644759852558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6170644759852558
+    },
+    {
+      "train_step": 7660,
+      "loss": 0.6813710237944568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6813710237944568
+    },
+    {
+      "train_step": 7661,
+      "loss": 0.7261572232442295,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7261572232442295
+    },
+    {
+      "train_step": 7662,
+      "loss": 0.7502124561397092,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7502124561397092
+    },
+    {
+      "train_step": 7663,
+      "loss": 0.6799707879896405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6799707879896405
+    },
+    {
+      "train_step": 7664,
+      "loss": 0.6418544357838923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6418544357838923
+    },
+    {
+      "train_step": 7665,
+      "loss": 0.7140564085193833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7140564085193833
+    },
+    {
+      "train_step": 7666,
+      "loss": 0.31241904447968083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31241904447968083
+    },
+    {
+      "train_step": 7667,
+      "loss": 0.7133378884407203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7133378884407203
+    },
+    {
+      "train_step": 7668,
+      "loss": 0.5108287297218453,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5108287297218453
+    },
+    {
+      "train_step": 7669,
+      "loss": 0.5064428771936114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5064428771936114
+    },
+    {
+      "train_step": 7670,
+      "loss": 0.2918742613747995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2918742613747995
+    },
+    {
+      "train_step": 7671,
+      "loss": 0.6817415534101862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6817415534101862
+    },
+    {
+      "train_step": 7672,
+      "loss": 0.5839773366058642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5839773366058642
+    },
+    {
+      "train_step": 7673,
+      "loss": 0.5250592004228917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5250592004228917
+    },
+    {
+      "train_step": 7674,
+      "loss": 0.7287282412921475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7287282412921475
+    },
+    {
+      "train_step": 7675,
+      "loss": 0.38263868323077754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38263868323077754
+    },
+    {
+      "train_step": 7676,
+      "loss": 0.49929797654090063,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49929797654090063
+    },
+    {
+      "train_step": 7677,
+      "loss": 0.6182616626539968,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6182616626539968
+    },
+    {
+      "train_step": 7678,
+      "loss": 0.5276878542215131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5276878542215131
+    },
+    {
+      "train_step": 7679,
+      "loss": 0.47707418571656784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47707418571656784
+    },
+    {
+      "train_step": 7680,
+      "loss": 0.5180194577868136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5180194577868136
+    },
+    {
+      "train_step": 7681,
+      "loss": 0.3683067964258239,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3683067964258239
+    },
+    {
+      "train_step": 7682,
+      "loss": 0.5473450541707163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5473450541707163
+    },
+    {
+      "train_step": 7683,
+      "loss": 0.5589055545515554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5589055545515554
+    },
+    {
+      "train_step": 7684,
+      "loss": 0.786652702767499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.786652702767499
+    },
+    {
+      "train_step": 7685,
+      "loss": 0.5022575246889219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5022575246889219
+    },
+    {
+      "train_step": 7686,
+      "loss": 0.618797705091275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.618797705091275
+    },
+    {
+      "train_step": 7687,
+      "loss": 0.6299421051265118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6299421051265118
+    },
+    {
+      "train_step": 7688,
+      "loss": 0.655952732716225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.655952732716225
+    },
+    {
+      "train_step": 7689,
+      "loss": 0.5044707405557376,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5044707405557376
+    },
+    {
+      "train_step": 7690,
+      "loss": 0.7418122936317404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7418122936317404
+    },
+    {
+      "train_step": 7691,
+      "loss": 0.40227130708536035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40227130708536035
+    },
+    {
+      "train_step": 7692,
+      "loss": 0.5390856475273689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5390856475273689
+    },
+    {
+      "train_step": 7693,
+      "loss": 0.5188931803155452,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5188931803155452
+    },
+    {
+      "train_step": 7694,
+      "loss": 0.4479049087338415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4479049087338415
+    },
+    {
+      "train_step": 7695,
+      "loss": 0.5928907789118633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5928907789118633
+    },
+    {
+      "train_step": 7696,
+      "loss": 0.46806541596866663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46806541596866663
+    },
+    {
+      "train_step": 7697,
+      "loss": 0.5121789335124224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5121789335124224
+    },
+    {
+      "train_step": 7698,
+      "loss": 0.8331786192972492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8331786192972492
+    },
+    {
+      "train_step": 7699,
+      "loss": 0.5231552389596631,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5231552389596631
+    },
+    {
+      "train_step": 7700,
+      "loss": 0.624618591749324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.624618591749324
+    },
+    {
+      "train_step": 7701,
+      "loss": 0.8637893835653498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8637893835653498
+    },
+    {
+      "train_step": 7702,
+      "loss": 0.7188476893818067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7188476893818067
+    },
+    {
+      "train_step": 7703,
+      "loss": 0.48297110999292614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48297110999292614
+    },
+    {
+      "train_step": 7704,
+      "loss": 0.5809732550208464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5809732550208464
+    },
+    {
+      "train_step": 7705,
+      "loss": 0.598970508745689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.598970508745689
+    },
+    {
+      "train_step": 7706,
+      "loss": 0.548408559934866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.548408559934866
+    },
+    {
+      "train_step": 7707,
+      "loss": 0.5488047017132196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5488047017132196
+    },
+    {
+      "train_step": 7708,
+      "loss": 0.5513355408551663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5513355408551663
+    },
+    {
+      "train_step": 7709,
+      "loss": 0.7250268693244266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7250268693244266
+    },
+    {
+      "train_step": 7710,
+      "loss": 0.5464004429062148,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5464004429062148
+    },
+    {
+      "train_step": 7711,
+      "loss": 0.661965620053026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.661965620053026
+    },
+    {
+      "train_step": 7712,
+      "loss": 0.4671872610972204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4671872610972204
+    },
+    {
+      "train_step": 7713,
+      "loss": 0.725815358778159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.725815358778159
+    },
+    {
+      "train_step": 7714,
+      "loss": 0.5364908227422106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5364908227422106
+    },
+    {
+      "train_step": 7715,
+      "loss": 0.575543995471481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.575543995471481
+    },
+    {
+      "train_step": 7716,
+      "loss": 0.5883657997794052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5883657997794052
+    },
+    {
+      "train_step": 7717,
+      "loss": 0.4084757081193818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4084757081193818
+    },
+    {
+      "train_step": 7718,
+      "loss": 0.7134452857505468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7134452857505468
+    },
+    {
+      "train_step": 7719,
+      "loss": 0.426458549309519,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.426458549309519
+    },
+    {
+      "train_step": 7720,
+      "loss": 0.7085303323656639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7085303323656639
+    },
+    {
+      "train_step": 7721,
+      "loss": 0.5230053889644078,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5230053889644078
+    },
+    {
+      "train_step": 7722,
+      "loss": 0.597692515644913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.597692515644913
+    },
+    {
+      "train_step": 7723,
+      "loss": 0.5887637818712772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5887637818712772
+    },
+    {
+      "train_step": 7724,
+      "loss": 0.48864926061640884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48864926061640884
+    },
+    {
+      "train_step": 7725,
+      "loss": 0.4953340528833724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4953340528833724
+    },
+    {
+      "train_step": 7726,
+      "loss": 0.6524196380620397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6524196380620397
+    },
+    {
+      "train_step": 7727,
+      "loss": 0.7291605590894549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7291605590894549
+    },
+    {
+      "train_step": 7728,
+      "loss": 0.5491379081251409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5491379081251409
+    },
+    {
+      "train_step": 7729,
+      "loss": 0.6761472034696416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6761472034696416
+    },
+    {
+      "train_step": 7730,
+      "loss": 0.6559649342259353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6559649342259353
+    },
+    {
+      "train_step": 7731,
+      "loss": 0.47427436782503857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47427436782503857
+    },
+    {
+      "train_step": 7732,
+      "loss": 0.5583302442882967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5583302442882967
+    },
+    {
+      "train_step": 7733,
+      "loss": 0.5833530969682814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5833530969682814
+    },
+    {
+      "train_step": 7734,
+      "loss": 0.5047750193943474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5047750193943474
+    },
+    {
+      "train_step": 7735,
+      "loss": 0.526179194533014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.526179194533014
+    },
+    {
+      "train_step": 7736,
+      "loss": 0.40472868437359255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40472868437359255
+    },
+    {
+      "train_step": 7737,
+      "loss": 0.45953914905970167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45953914905970167
+    },
+    {
+      "train_step": 7738,
+      "loss": 0.6072376245411797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6072376245411797
+    },
+    {
+      "train_step": 7739,
+      "loss": 0.7478187872644009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7478187872644009
+    },
+    {
+      "train_step": 7740,
+      "loss": 0.5960876983448895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5960876983448895
+    },
+    {
+      "train_step": 7741,
+      "loss": 0.4538653420731208,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4538653420731208
+    },
+    {
+      "train_step": 7742,
+      "loss": 0.7014535446919273,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7014535446919273
+    },
+    {
+      "train_step": 7743,
+      "loss": 0.43759112053386057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43759112053386057
+    },
+    {
+      "train_step": 7744,
+      "loss": 0.6570454157897574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6570454157897574
+    },
+    {
+      "train_step": 7745,
+      "loss": 0.6499906060029782,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6499906060029782
+    },
+    {
+      "train_step": 7746,
+      "loss": 0.4572318107587102,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4572318107587102
+    },
+    {
+      "train_step": 7747,
+      "loss": 0.5756157689276029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5756157689276029
+    },
+    {
+      "train_step": 7748,
+      "loss": 0.4628770989367921,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4628770989367921
+    },
+    {
+      "train_step": 7749,
+      "loss": 0.5127963605242666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5127963605242666
+    },
+    {
+      "train_step": 7750,
+      "loss": 0.478889275359808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.478889275359808
+    },
+    {
+      "train_step": 7751,
+      "loss": 0.4643130409443249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4643130409443249
+    },
+    {
+      "train_step": 7752,
+      "loss": 0.5130364410573145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5130364410573145
+    },
+    {
+      "train_step": 7753,
+      "loss": 0.44087226754999803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44087226754999803
+    },
+    {
+      "train_step": 7754,
+      "loss": 0.6900680835706132,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6900680835706132
+    },
+    {
+      "train_step": 7755,
+      "loss": 0.3904889722476454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3904889722476454
+    },
+    {
+      "train_step": 7756,
+      "loss": 0.5243602552810558,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5243602552810558
+    },
+    {
+      "train_step": 7757,
+      "loss": 0.43407336889239134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43407336889239134
+    },
+    {
+      "train_step": 7758,
+      "loss": 0.7425396419990572,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7425396419990572
+    },
+    {
+      "train_step": 7759,
+      "loss": 0.7604762208895693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7604762208895693
+    },
+    {
+      "train_step": 7760,
+      "loss": 0.5400683015561378,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5400683015561378
+    },
+    {
+      "train_step": 7761,
+      "loss": 0.6374268182805988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6374268182805988
+    },
+    {
+      "train_step": 7762,
+      "loss": 0.5483578339102252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5483578339102252
+    },
+    {
+      "train_step": 7763,
+      "loss": 0.5452613697676127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5452613697676127
+    },
+    {
+      "train_step": 7764,
+      "loss": 0.49166209270174277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49166209270174277
+    },
+    {
+      "train_step": 7765,
+      "loss": 0.5825344856076351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5825344856076351
+    },
+    {
+      "train_step": 7766,
+      "loss": 0.4129817414366461,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4129817414366461
+    },
+    {
+      "train_step": 7767,
+      "loss": 0.5862202023023422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5862202023023422
+    },
+    {
+      "train_step": 7768,
+      "loss": 0.4081321493827636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4081321493827636
+    },
+    {
+      "train_step": 7769,
+      "loss": 0.5669841781203464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5669841781203464
+    },
+    {
+      "train_step": 7770,
+      "loss": 0.6810383914602863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6810383914602863
+    },
+    {
+      "train_step": 7771,
+      "loss": 0.5035058822272089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035058822272089
+    },
+    {
+      "train_step": 7772,
+      "loss": 0.5291252620036415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5291252620036415
+    },
+    {
+      "train_step": 7773,
+      "loss": 0.7016495723752784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7016495723752784
+    },
+    {
+      "train_step": 7774,
+      "loss": 0.8275070829715901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8275070829715901
+    },
+    {
+      "train_step": 7775,
+      "loss": 0.4954416228781944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4954416228781944
+    },
+    {
+      "train_step": 7776,
+      "loss": 0.8039352012086723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8039352012086723
+    },
+    {
+      "train_step": 7777,
+      "loss": 0.6176926705200653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6176926705200653
+    },
+    {
+      "train_step": 7778,
+      "loss": 0.7320621587143827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7320621587143827
+    },
+    {
+      "train_step": 7779,
+      "loss": 0.4554291722573818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4554291722573818
+    },
+    {
+      "train_step": 7780,
+      "loss": 0.3685224514632753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3685224514632753
+    },
+    {
+      "train_step": 7781,
+      "loss": 0.542762298085253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.542762298085253
+    },
+    {
+      "train_step": 7782,
+      "loss": 0.5208891021905858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5208891021905858
+    },
+    {
+      "train_step": 7783,
+      "loss": 0.48334868302411155,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48334868302411155
+    },
+    {
+      "train_step": 7784,
+      "loss": 0.5311646665753433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5311646665753433
+    },
+    {
+      "train_step": 7785,
+      "loss": 0.4663718249943022,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4663718249943022
+    },
+    {
+      "train_step": 7786,
+      "loss": 0.44825163647626387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44825163647626387
+    },
+    {
+      "train_step": 7787,
+      "loss": 0.5663144110285374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5663144110285374
+    },
+    {
+      "train_step": 7788,
+      "loss": 0.6199414959506223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6199414959506223
+    },
+    {
+      "train_step": 7789,
+      "loss": 0.34253100304212275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34253100304212275
+    },
+    {
+      "train_step": 7790,
+      "loss": 0.8669031646637853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8669031646637853
+    },
+    {
+      "train_step": 7791,
+      "loss": 0.5020631364791154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5020631364791154
+    },
+    {
+      "train_step": 7792,
+      "loss": 0.49796107486876817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49796107486876817
+    },
+    {
+      "train_step": 7793,
+      "loss": 0.6848330639900436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6848330639900436
+    },
+    {
+      "train_step": 7794,
+      "loss": 0.4572989687889782,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4572989687889782
+    },
+    {
+      "train_step": 7795,
+      "loss": 0.5582272889092148,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5582272889092148
+    },
+    {
+      "train_step": 7796,
+      "loss": 0.6173761900011109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6173761900011109
+    },
+    {
+      "train_step": 7797,
+      "loss": 0.5093917868022377,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5093917868022377
+    },
+    {
+      "train_step": 7798,
+      "loss": 0.39730374897194337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39730374897194337
+    },
+    {
+      "train_step": 7799,
+      "loss": 0.5326448042023235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5326448042023235
+    },
+    {
+      "train_step": 7800,
+      "loss": 0.5030484708494033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5030484708494033
+    },
+    {
+      "train_step": 7801,
+      "loss": 0.5774145812454426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5774145812454426
+    },
+    {
+      "train_step": 7802,
+      "loss": 0.5410424527148504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5410424527148504
+    },
+    {
+      "train_step": 7803,
+      "loss": 0.38718727669373365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38718727669373365
+    },
+    {
+      "train_step": 7804,
+      "loss": 0.3743223607307357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3743223607307357
+    },
+    {
+      "train_step": 7805,
+      "loss": 0.4964938260647747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4964938260647747
+    },
+    {
+      "train_step": 7806,
+      "loss": 0.4831638934113105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4831638934113105
+    },
+    {
+      "train_step": 7807,
+      "loss": 0.5936768525058358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5936768525058358
+    },
+    {
+      "train_step": 7808,
+      "loss": 0.4000972656045333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4000972656045333
+    },
+    {
+      "train_step": 7809,
+      "loss": 0.8164874402197566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8164874402197566
+    },
+    {
+      "train_step": 7810,
+      "loss": 0.4679949256343357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4679949256343357
+    },
+    {
+      "train_step": 7811,
+      "loss": 0.5838227469377374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5838227469377374
+    },
+    {
+      "train_step": 7812,
+      "loss": 0.48431890768556934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48431890768556934
+    },
+    {
+      "train_step": 7813,
+      "loss": 0.5150091059947483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5150091059947483
+    },
+    {
+      "train_step": 7814,
+      "loss": 0.5317487668021748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5317487668021748
+    },
+    {
+      "train_step": 7815,
+      "loss": 0.32748623924847614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32748623924847614
+    },
+    {
+      "train_step": 7816,
+      "loss": 0.4487574776881475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4487574776881475
+    },
+    {
+      "train_step": 7817,
+      "loss": 0.5650016650589739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5650016650589739
+    },
+    {
+      "train_step": 7818,
+      "loss": 0.718228915589058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.718228915589058
+    },
+    {
+      "train_step": 7819,
+      "loss": 0.56645400936706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.56645400936706
+    },
+    {
+      "train_step": 7820,
+      "loss": 0.7764629843087486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7764629843087486
+    },
+    {
+      "train_step": 7821,
+      "loss": 0.5451019649438096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5451019649438096
+    },
+    {
+      "train_step": 7822,
+      "loss": 0.5779353953055656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5779353953055656
+    },
+    {
+      "train_step": 7823,
+      "loss": 0.5804610734247153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5804610734247153
+    },
+    {
+      "train_step": 7824,
+      "loss": 0.29891273180611494,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.29891273180611494
+    },
+    {
+      "train_step": 7825,
+      "loss": 0.5513254857876972,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5513254857876972
+    },
+    {
+      "train_step": 7826,
+      "loss": 0.48387024033529125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48387024033529125
+    },
+    {
+      "train_step": 7827,
+      "loss": 0.7247058969909084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7247058969909084
+    },
+    {
+      "train_step": 7828,
+      "loss": 0.6845982980785437,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6845982980785437
+    },
+    {
+      "train_step": 7829,
+      "loss": 0.5351572592425285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5351572592425285
+    },
+    {
+      "train_step": 7830,
+      "loss": 0.35900818605471113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35900818605471113
+    },
+    {
+      "train_step": 7831,
+      "loss": 0.30709067491844144,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.30709067491844144
+    },
+    {
+      "train_step": 7832,
+      "loss": 0.5409039618150348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5409039618150348
+    },
+    {
+      "train_step": 7833,
+      "loss": 0.550820303333887,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.550820303333887
+    },
+    {
+      "train_step": 7834,
+      "loss": 0.5452859976577081,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5452859976577081
+    },
+    {
+      "train_step": 7835,
+      "loss": 0.5701056608061187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5701056608061187
+    },
+    {
+      "train_step": 7836,
+      "loss": 0.5879900633055974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5879900633055974
+    },
+    {
+      "train_step": 7837,
+      "loss": 0.6178019914661906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6178019914661906
+    },
+    {
+      "train_step": 7838,
+      "loss": 0.41372292283556555,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41372292283556555
+    },
+    {
+      "train_step": 7839,
+      "loss": 0.418111421422375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.418111421422375
+    },
+    {
+      "train_step": 7840,
+      "loss": 0.6643739314624396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6643739314624396
+    },
+    {
+      "train_step": 7841,
+      "loss": 0.4670791822431312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4670791822431312
+    },
+    {
+      "train_step": 7842,
+      "loss": 0.5986066172638562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5986066172638562
+    },
+    {
+      "train_step": 7843,
+      "loss": 0.41451134423199115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41451134423199115
+    },
+    {
+      "train_step": 7844,
+      "loss": 0.7113326421241578,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7113326421241578
+    },
+    {
+      "train_step": 7845,
+      "loss": 0.526142392408238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.526142392408238
+    },
+    {
+      "train_step": 7846,
+      "loss": 0.5259105906783684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5259105906783684
+    },
+    {
+      "train_step": 7847,
+      "loss": 0.6031230975959871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6031230975959871
+    },
+    {
+      "train_step": 7848,
+      "loss": 0.25648520720101525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.25648520720101525
+    },
+    {
+      "train_step": 7849,
+      "loss": 0.7183682953320468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7183682953320468
+    },
+    {
+      "train_step": 7850,
+      "loss": 0.5666310971562433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5666310971562433
+    },
+    {
+      "train_step": 7851,
+      "loss": 0.3896352879919666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3896352879919666
+    },
+    {
+      "train_step": 7852,
+      "loss": 0.49154095668773307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49154095668773307
+    },
+    {
+      "train_step": 7853,
+      "loss": 0.4326119958964177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4326119958964177
+    },
+    {
+      "train_step": 7854,
+      "loss": 0.6270003451738535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6270003451738535
+    },
+    {
+      "train_step": 7855,
+      "loss": 0.5464274229856817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5464274229856817
+    },
+    {
+      "train_step": 7856,
+      "loss": 0.5666240717201048,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5666240717201048
+    },
+    {
+      "train_step": 7857,
+      "loss": 0.36831932547491225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36831932547491225
+    },
+    {
+      "train_step": 7858,
+      "loss": 0.6752070936137653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6752070936137653
+    },
+    {
+      "train_step": 7859,
+      "loss": 0.4658045207287098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4658045207287098
+    },
+    {
+      "train_step": 7860,
+      "loss": 0.5553302216623544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5553302216623544
+    },
+    {
+      "train_step": 7861,
+      "loss": 0.5959438227876974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5959438227876974
+    },
+    {
+      "train_step": 7862,
+      "loss": 0.5134138889064486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5134138889064486
+    },
+    {
+      "train_step": 7863,
+      "loss": 0.5929802287756636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5929802287756636
+    },
+    {
+      "train_step": 7864,
+      "loss": 0.5703377594654933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5703377594654933
+    },
+    {
+      "train_step": 7865,
+      "loss": 0.3550803212337108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3550803212337108
+    },
+    {
+      "train_step": 7866,
+      "loss": 0.494142916219494,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.494142916219494
+    },
+    {
+      "train_step": 7867,
+      "loss": 0.5029197559704416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5029197559704416
+    },
+    {
+      "train_step": 7868,
+      "loss": 0.5197311897704527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5197311897704527
+    },
+    {
+      "train_step": 7869,
+      "loss": 0.5944576719786858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5944576719786858
+    },
+    {
+      "train_step": 7870,
+      "loss": 0.40210973248787446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40210973248787446
+    },
+    {
+      "train_step": 7871,
+      "loss": 0.6376049358279275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6376049358279275
+    },
+    {
+      "train_step": 7872,
+      "loss": 0.5322114948231584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5322114948231584
+    },
+    {
+      "train_step": 7873,
+      "loss": 0.5371022726394109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5371022726394109
+    },
+    {
+      "train_step": 7874,
+      "loss": 0.49140048902598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49140048902598
+    },
+    {
+      "train_step": 7875,
+      "loss": 0.6746095161619935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6746095161619935
+    },
+    {
+      "train_step": 7876,
+      "loss": 0.4414645301508316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4414645301508316
+    },
+    {
+      "train_step": 7877,
+      "loss": 0.6924336226785732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6924336226785732
+    },
+    {
+      "train_step": 7878,
+      "loss": 0.5980794803466674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5980794803466674
+    },
+    {
+      "train_step": 7879,
+      "loss": 0.43313052039704714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43313052039704714
+    },
+    {
+      "train_step": 7880,
+      "loss": 0.7644076145219031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7644076145219031
+    },
+    {
+      "train_step": 7881,
+      "loss": 0.5475992711534801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5475992711534801
+    },
+    {
+      "train_step": 7882,
+      "loss": 0.6181265088663045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6181265088663045
+    },
+    {
+      "train_step": 7883,
+      "loss": 0.6440225431181323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6440225431181323
+    },
+    {
+      "train_step": 7884,
+      "loss": 0.41562751443157836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41562751443157836
+    },
+    {
+      "train_step": 7885,
+      "loss": 0.5785895242121574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5785895242121574
+    },
+    {
+      "train_step": 7886,
+      "loss": 0.3557765501361613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3557765501361613
+    },
+    {
+      "train_step": 7887,
+      "loss": 0.39404169570261155,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39404169570261155
+    },
+    {
+      "train_step": 7888,
+      "loss": 0.7529154404235618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7529154404235618
+    },
+    {
+      "train_step": 7889,
+      "loss": 0.5333074132965242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5333074132965242
+    },
+    {
+      "train_step": 7890,
+      "loss": 0.7287325835888951,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7287325835888951
+    },
+    {
+      "train_step": 7891,
+      "loss": 0.6672485369798786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6672485369798786
+    },
+    {
+      "train_step": 7892,
+      "loss": 0.49676085880916016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49676085880916016
+    },
+    {
+      "train_step": 7893,
+      "loss": 0.5803734993727213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5803734993727213
+    },
+    {
+      "train_step": 7894,
+      "loss": 0.5103651802432427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5103651802432427
+    },
+    {
+      "train_step": 7895,
+      "loss": 0.6735258081508336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6735258081508336
+    },
+    {
+      "train_step": 7896,
+      "loss": 0.5328217210083602,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5328217210083602
+    },
+    {
+      "train_step": 7897,
+      "loss": 0.4558601079933218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4558601079933218
+    },
+    {
+      "train_step": 7898,
+      "loss": 0.61275930061084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.61275930061084
+    },
+    {
+      "train_step": 7899,
+      "loss": 0.3104115598884979,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3104115598884979
+    },
+    {
+      "train_step": 7900,
+      "loss": 0.582021341385098,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.582021341385098
+    },
+    {
+      "train_step": 7901,
+      "loss": 0.45519593352903753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45519593352903753
+    },
+    {
+      "train_step": 7902,
+      "loss": 0.32596989582445896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32596989582445896
+    },
+    {
+      "train_step": 7903,
+      "loss": 0.75097592358764,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.75097592358764
+    },
+    {
+      "train_step": 7904,
+      "loss": 0.5516965621841262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5516965621841262
+    },
+    {
+      "train_step": 7905,
+      "loss": 0.5737517871332727,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5737517871332727
+    },
+    {
+      "train_step": 7906,
+      "loss": 0.39297312765926784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39297312765926784
+    },
+    {
+      "train_step": 7907,
+      "loss": 0.3931279653581869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3931279653581869
+    },
+    {
+      "train_step": 7908,
+      "loss": 0.5546790790374967,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5546790790374967
+    },
+    {
+      "train_step": 7909,
+      "loss": 0.4715140683842543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4715140683842543
+    },
+    {
+      "train_step": 7910,
+      "loss": 0.654670780723332,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.654670780723332
+    },
+    {
+      "train_step": 7911,
+      "loss": 0.49593720978082856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49593720978082856
+    },
+    {
+      "train_step": 7912,
+      "loss": 0.6371427652191215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6371427652191215
+    },
+    {
+      "train_step": 7913,
+      "loss": 0.6420026498547268,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6420026498547268
+    },
+    {
+      "train_step": 7914,
+      "loss": 0.5885177817131623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5885177817131623
+    },
+    {
+      "train_step": 7915,
+      "loss": 0.5341517958987396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5341517958987396
+    },
+    {
+      "train_step": 7916,
+      "loss": 0.5268481369652245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5268481369652245
+    },
+    {
+      "train_step": 7917,
+      "loss": 0.41512749141672434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41512749141672434
+    },
+    {
+      "train_step": 7918,
+      "loss": 0.5623105088022765,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5623105088022765
+    },
+    {
+      "train_step": 7919,
+      "loss": 0.49973823693939956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49973823693939956
+    },
+    {
+      "train_step": 7920,
+      "loss": 0.4045327560340318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4045327560340318
+    },
+    {
+      "train_step": 7921,
+      "loss": 0.4882420332120149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4882420332120149
+    },
+    {
+      "train_step": 7922,
+      "loss": 0.4804616991038194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4804616991038194
+    },
+    {
+      "train_step": 7923,
+      "loss": 0.40616646955710795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40616646955710795
+    },
+    {
+      "train_step": 7924,
+      "loss": 0.6987285104600374,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6987285104600374
+    },
+    {
+      "train_step": 7925,
+      "loss": 0.6043748623797096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6043748623797096
+    },
+    {
+      "train_step": 7926,
+      "loss": 0.47488000185236623,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47488000185236623
+    },
+    {
+      "train_step": 7927,
+      "loss": 0.6494435799708236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6494435799708236
+    },
+    {
+      "train_step": 7928,
+      "loss": 0.691350583543913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.691350583543913
+    },
+    {
+      "train_step": 7929,
+      "loss": 0.45640778851998587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45640778851998587
+    },
+    {
+      "train_step": 7930,
+      "loss": 0.4960077009438857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4960077009438857
+    },
+    {
+      "train_step": 7931,
+      "loss": 0.4386004678917079,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4386004678917079
+    },
+    {
+      "train_step": 7932,
+      "loss": 0.4736243123200912,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4736243123200912
+    },
+    {
+      "train_step": 7933,
+      "loss": 0.602999050585533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.602999050585533
+    },
+    {
+      "train_step": 7934,
+      "loss": 0.4510277818352461,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4510277818352461
+    },
+    {
+      "train_step": 7935,
+      "loss": 0.46508073553893337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46508073553893337
+    },
+    {
+      "train_step": 7936,
+      "loss": 0.5813092219231575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5813092219231575
+    },
+    {
+      "train_step": 7937,
+      "loss": 0.5195312585145353,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5195312585145353
+    },
+    {
+      "train_step": 7938,
+      "loss": 0.6861572226279291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6861572226279291
+    },
+    {
+      "train_step": 7939,
+      "loss": 0.5710212482492693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5710212482492693
+    },
+    {
+      "train_step": 7940,
+      "loss": 0.5702417374937438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5702417374937438
+    },
+    {
+      "train_step": 7941,
+      "loss": 0.6079924249601574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6079924249601574
+    },
+    {
+      "train_step": 7942,
+      "loss": 0.529029690017629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.529029690017629
+    },
+    {
+      "train_step": 7943,
+      "loss": 0.5012781017879429,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5012781017879429
+    },
+    {
+      "train_step": 7944,
+      "loss": 0.4533258746947531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4533258746947531
+    },
+    {
+      "train_step": 7945,
+      "loss": 0.4835964282405003,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4835964282405003
+    },
+    {
+      "train_step": 7946,
+      "loss": 0.6549394436197816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6549394436197816
+    },
+    {
+      "train_step": 7947,
+      "loss": 0.6356179353001115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6356179353001115
+    },
+    {
+      "train_step": 7948,
+      "loss": 0.5509149529290361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5509149529290361
+    },
+    {
+      "train_step": 7949,
+      "loss": 0.6250846951240634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6250846951240634
+    },
+    {
+      "train_step": 7950,
+      "loss": 0.6022574599352251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6022574599352251
+    },
+    {
+      "train_step": 7951,
+      "loss": 0.8482648364088643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8482648364088643
+    },
+    {
+      "train_step": 7952,
+      "loss": 0.5787979902211754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5787979902211754
+    },
+    {
+      "train_step": 7953,
+      "loss": 0.6295486626462531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6295486626462531
+    },
+    {
+      "train_step": 7954,
+      "loss": 0.5598754733371695,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5598754733371695
+    },
+    {
+      "train_step": 7955,
+      "loss": 0.7042220829456158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7042220829456158
+    },
+    {
+      "train_step": 7956,
+      "loss": 0.4423571813537547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4423571813537547
+    },
+    {
+      "train_step": 7957,
+      "loss": 0.44418785022681506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44418785022681506
+    },
+    {
+      "train_step": 7958,
+      "loss": 0.5570873740115918,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5570873740115918
+    },
+    {
+      "train_step": 7959,
+      "loss": 0.5814138631985397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5814138631985397
+    },
+    {
+      "train_step": 7960,
+      "loss": 0.36555377254825644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36555377254825644
+    },
+    {
+      "train_step": 7961,
+      "loss": 0.5306532940752738,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5306532940752738
+    },
+    {
+      "train_step": 7962,
+      "loss": 0.7421031385396978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7421031385396978
+    },
+    {
+      "train_step": 7963,
+      "loss": 0.672059814931164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.672059814931164
+    },
+    {
+      "train_step": 7964,
+      "loss": 0.6261385323954922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6261385323954922
+    },
+    {
+      "train_step": 7965,
+      "loss": 0.6928487550961803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6928487550961803
+    },
+    {
+      "train_step": 7966,
+      "loss": 0.5286084226445382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5286084226445382
+    },
+    {
+      "train_step": 7967,
+      "loss": 0.6073930426915769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6073930426915769
+    },
+    {
+      "train_step": 7968,
+      "loss": 0.7357862364172625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7357862364172625
+    },
+    {
+      "train_step": 7969,
+      "loss": 0.5010579269005175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5010579269005175
+    },
+    {
+      "train_step": 7970,
+      "loss": 0.5716917205323474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5716917205323474
+    },
+    {
+      "train_step": 7971,
+      "loss": 0.5290444315121562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5290444315121562
+    },
+    {
+      "train_step": 7972,
+      "loss": 0.42767265157588213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42767265157588213
+    },
+    {
+      "train_step": 7973,
+      "loss": 0.5102766547007034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5102766547007034
+    },
+    {
+      "train_step": 7974,
+      "loss": 0.5657688876366045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5657688876366045
+    },
+    {
+      "train_step": 7975,
+      "loss": 0.602818310016056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.602818310016056
+    },
+    {
+      "train_step": 7976,
+      "loss": 0.43687564000182205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43687564000182205
+    },
+    {
+      "train_step": 7977,
+      "loss": 0.4422885582399034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4422885582399034
+    },
+    {
+      "train_step": 7978,
+      "loss": 0.7373853734822966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7373853734822966
+    },
+    {
+      "train_step": 7979,
+      "loss": 0.5760832479210659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5760832479210659
+    },
+    {
+      "train_step": 7980,
+      "loss": 0.534230509658276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.534230509658276
+    },
+    {
+      "train_step": 7981,
+      "loss": 0.5519484718235399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5519484718235399
+    },
+    {
+      "train_step": 7982,
+      "loss": 0.32944394258147985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32944394258147985
+    },
+    {
+      "train_step": 7983,
+      "loss": 0.5050828782942564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5050828782942564
+    },
+    {
+      "train_step": 7984,
+      "loss": 0.5627249121492985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5627249121492985
+    },
+    {
+      "train_step": 7985,
+      "loss": 0.5334960464125179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5334960464125179
+    },
+    {
+      "train_step": 7986,
+      "loss": 0.5317230509813283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5317230509813283
+    },
+    {
+      "train_step": 7987,
+      "loss": 0.44558620938233495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44558620938233495
+    },
+    {
+      "train_step": 7988,
+      "loss": 0.686254254592449,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.686254254592449
+    },
+    {
+      "train_step": 7989,
+      "loss": 0.6070230104653433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6070230104653433
+    },
+    {
+      "train_step": 7990,
+      "loss": 0.424095325598402,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.424095325598402
+    },
+    {
+      "train_step": 7991,
+      "loss": 0.5243303561821667,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5243303561821667
+    },
+    {
+      "train_step": 7992,
+      "loss": 0.5314491774464953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5314491774464953
+    },
+    {
+      "train_step": 7993,
+      "loss": 0.45472851765175143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45472851765175143
+    },
+    {
+      "train_step": 7994,
+      "loss": 0.6909798067813279,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6909798067813279
+    },
+    {
+      "train_step": 7995,
+      "loss": 0.4698040869538816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4698040869538816
+    },
+    {
+      "train_step": 7996,
+      "loss": 0.5719922630538441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5719922630538441
+    },
+    {
+      "train_step": 7997,
+      "loss": 0.6682163883733783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6682163883733783
+    },
+    {
+      "train_step": 7998,
+      "loss": 0.48067471581755367,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48067471581755367
+    },
+    {
+      "train_step": 7999,
+      "loss": 0.6305796362927331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6305796362927331
+    },
+    {
+      "train_step": 8000,
+      "loss": 0.6550760796647264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6550760796647264
+    },
+    {
+      "train_step": 8001,
+      "loss": 0.6881135791017442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6881135791017442
+    },
+    {
+      "train_step": 8002,
+      "loss": 0.5436709965864703,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5436709965864703
+    },
+    {
+      "train_step": 8003,
+      "loss": 0.625841220940262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.625841220940262
+    },
+    {
+      "train_step": 8004,
+      "loss": 0.47655334750232253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47655334750232253
+    },
+    {
+      "train_step": 8005,
+      "loss": 0.4154910365027305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4154910365027305
+    },
+    {
+      "train_step": 8006,
+      "loss": 0.46056669008090856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46056669008090856
+    },
+    {
+      "train_step": 8007,
+      "loss": 0.7243086945882462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7243086945882462
+    },
+    {
+      "train_step": 8008,
+      "loss": 1.0737681850950198,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 1.0737681850950198
+    },
+    {
+      "train_step": 8009,
+      "loss": 0.46735379666863297,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46735379666863297
+    },
+    {
+      "train_step": 8010,
+      "loss": 0.46747831236067505,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46747831236067505
+    },
+    {
+      "train_step": 8011,
+      "loss": 0.5550835469522523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5550835469522523
+    },
+    {
+      "train_step": 8012,
+      "loss": 0.48580124421489634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48580124421489634
+    },
+    {
+      "train_step": 8013,
+      "loss": 0.6109782516254648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6109782516254648
+    },
+    {
+      "train_step": 8014,
+      "loss": 0.6605930931965613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6605930931965613
+    },
+    {
+      "train_step": 8015,
+      "loss": 0.41415568489120486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41415568489120486
+    },
+    {
+      "train_step": 8016,
+      "loss": 0.5411963196035993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5411963196035993
+    },
+    {
+      "train_step": 8017,
+      "loss": 0.8767053269935029,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8767053269935029
+    },
+    {
+      "train_step": 8018,
+      "loss": 0.6584278178243916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6584278178243916
+    },
+    {
+      "train_step": 8019,
+      "loss": 0.4421850085258867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4421850085258867
+    },
+    {
+      "train_step": 8020,
+      "loss": 0.5991963175339812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5991963175339812
+    },
+    {
+      "train_step": 8021,
+      "loss": 0.5429700031304232,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5429700031304232
+    },
+    {
+      "train_step": 8022,
+      "loss": 0.4019563862863743,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4019563862863743
+    },
+    {
+      "train_step": 8023,
+      "loss": 0.5095681508379956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5095681508379956
+    },
+    {
+      "train_step": 8024,
+      "loss": 0.6159834407305192,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6159834407305192
+    },
+    {
+      "train_step": 8025,
+      "loss": 0.4547616806779566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4547616806779566
+    },
+    {
+      "train_step": 8026,
+      "loss": 0.35495270661394723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35495270661394723
+    },
+    {
+      "train_step": 8027,
+      "loss": 0.6689477964727728,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6689477964727728
+    },
+    {
+      "train_step": 8028,
+      "loss": 0.6016828030697097,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6016828030697097
+    },
+    {
+      "train_step": 8029,
+      "loss": 0.4434096162455503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4434096162455503
+    },
+    {
+      "train_step": 8030,
+      "loss": 0.5097995063242042,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5097995063242042
+    },
+    {
+      "train_step": 8031,
+      "loss": 0.4575782184902569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4575782184902569
+    },
+    {
+      "train_step": 8032,
+      "loss": 0.6028265491664149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6028265491664149
+    },
+    {
+      "train_step": 8033,
+      "loss": 0.7505080834114743,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7505080834114743
+    },
+    {
+      "train_step": 8034,
+      "loss": 0.6791861232197842,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6791861232197842
+    },
+    {
+      "train_step": 8035,
+      "loss": 0.5329657425831238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5329657425831238
+    },
+    {
+      "train_step": 8036,
+      "loss": 0.5210385712872272,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5210385712872272
+    },
+    {
+      "train_step": 8037,
+      "loss": 0.4605555015241917,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4605555015241917
+    },
+    {
+      "train_step": 8038,
+      "loss": 0.7229329227250456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7229329227250456
+    },
+    {
+      "train_step": 8039,
+      "loss": 0.5314514305065009,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5314514305065009
+    },
+    {
+      "train_step": 8040,
+      "loss": 0.5775284530399082,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5775284530399082
+    },
+    {
+      "train_step": 8041,
+      "loss": 0.7061022263085767,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7061022263085767
+    },
+    {
+      "train_step": 8042,
+      "loss": 0.5949106916696848,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5949106916696848
+    },
+    {
+      "train_step": 8043,
+      "loss": 0.5322895386323575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5322895386323575
+    },
+    {
+      "train_step": 8044,
+      "loss": 0.6124501008172644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6124501008172644
+    },
+    {
+      "train_step": 8045,
+      "loss": 0.5112029466065016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5112029466065016
+    },
+    {
+      "train_step": 8046,
+      "loss": 0.5857173094109654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5857173094109654
+    },
+    {
+      "train_step": 8047,
+      "loss": 0.6305327769882909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6305327769882909
+    },
+    {
+      "train_step": 8048,
+      "loss": 0.63251439668762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.63251439668762
+    },
+    {
+      "train_step": 8049,
+      "loss": 0.5644658826257346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5644658826257346
+    },
+    {
+      "train_step": 8050,
+      "loss": 0.689587271609131,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.689587271609131
+    },
+    {
+      "train_step": 8051,
+      "loss": 0.49999171738048687,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49999171738048687
+    },
+    {
+      "train_step": 8052,
+      "loss": 0.6365198877854392,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6365198877854392
+    },
+    {
+      "train_step": 8053,
+      "loss": 0.3999153864583849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3999153864583849
+    },
+    {
+      "train_step": 8054,
+      "loss": 0.6230015605895368,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6230015605895368
+    },
+    {
+      "train_step": 8055,
+      "loss": 0.548703690525736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.548703690525736
+    },
+    {
+      "train_step": 8056,
+      "loss": 0.6576387059057269,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6576387059057269
+    },
+    {
+      "train_step": 8057,
+      "loss": 0.7032999326505696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7032999326505696
+    },
+    {
+      "train_step": 8058,
+      "loss": 0.4575329408916057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4575329408916057
+    },
+    {
+      "train_step": 8059,
+      "loss": 0.35212332333181384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35212332333181384
+    },
+    {
+      "train_step": 8060,
+      "loss": 0.513868800660718,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.513868800660718
+    },
+    {
+      "train_step": 8061,
+      "loss": 0.6642722630827863,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6642722630827863
+    },
+    {
+      "train_step": 8062,
+      "loss": 0.7896203722764776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7896203722764776
+    },
+    {
+      "train_step": 8063,
+      "loss": 0.5402316546034508,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5402316546034508
+    },
+    {
+      "train_step": 8064,
+      "loss": 0.5858247391728927,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5858247391728927
+    },
+    {
+      "train_step": 8065,
+      "loss": 0.47041548233617425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47041548233617425
+    },
+    {
+      "train_step": 8066,
+      "loss": 0.466932382214697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.466932382214697
+    },
+    {
+      "train_step": 8067,
+      "loss": 0.5362963200968226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5362963200968226
+    },
+    {
+      "train_step": 8068,
+      "loss": 0.5329459979812908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5329459979812908
+    },
+    {
+      "train_step": 8069,
+      "loss": 0.5065557931104242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5065557931104242
+    },
+    {
+      "train_step": 8070,
+      "loss": 0.5887359552596159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5887359552596159
+    },
+    {
+      "train_step": 8071,
+      "loss": 0.686003756080954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.686003756080954
+    },
+    {
+      "train_step": 8072,
+      "loss": 0.5975090066599017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5975090066599017
+    },
+    {
+      "train_step": 8073,
+      "loss": 0.5403615789285989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5403615789285989
+    },
+    {
+      "train_step": 8074,
+      "loss": 0.6301750360162303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6301750360162303
+    },
+    {
+      "train_step": 8075,
+      "loss": 0.4382150269655949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4382150269655949
+    },
+    {
+      "train_step": 8076,
+      "loss": 0.5572901373553909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5572901373553909
+    },
+    {
+      "train_step": 8077,
+      "loss": 0.5803377930361349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5803377930361349
+    },
+    {
+      "train_step": 8078,
+      "loss": 0.6863245072784753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6863245072784753
+    },
+    {
+      "train_step": 8079,
+      "loss": 0.6688135181067763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6688135181067763
+    },
+    {
+      "train_step": 8080,
+      "loss": 0.5144310840762346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5144310840762346
+    },
+    {
+      "train_step": 8081,
+      "loss": 0.6260500663415456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6260500663415456
+    },
+    {
+      "train_step": 8082,
+      "loss": 0.5981412171045455,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5981412171045455
+    },
+    {
+      "train_step": 8083,
+      "loss": 0.5758352570381533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5758352570381533
+    },
+    {
+      "train_step": 8084,
+      "loss": 0.8190761630710943,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8190761630710943
+    },
+    {
+      "train_step": 8085,
+      "loss": 0.5591413078665862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5591413078665862
+    },
+    {
+      "train_step": 8086,
+      "loss": 0.5023372434639329,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5023372434639329
+    },
+    {
+      "train_step": 8087,
+      "loss": 0.6835579923953992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6835579923953992
+    },
+    {
+      "train_step": 8088,
+      "loss": 0.497532388190853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.497532388190853
+    },
+    {
+      "train_step": 8089,
+      "loss": 0.5247354031030657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5247354031030657
+    },
+    {
+      "train_step": 8090,
+      "loss": 0.4593736435264286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4593736435264286
+    },
+    {
+      "train_step": 8091,
+      "loss": 0.5638574888799904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5638574888799904
+    },
+    {
+      "train_step": 8092,
+      "loss": 0.6011588517254104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6011588517254104
+    },
+    {
+      "train_step": 8093,
+      "loss": 0.5089804118325266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5089804118325266
+    },
+    {
+      "train_step": 8094,
+      "loss": 0.6060635983980804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6060635983980804
+    },
+    {
+      "train_step": 8095,
+      "loss": 0.349667387324954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.349667387324954
+    },
+    {
+      "train_step": 8096,
+      "loss": 0.6230380662609689,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6230380662609689
+    },
+    {
+      "train_step": 8097,
+      "loss": 0.6759989008341112,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6759989008341112
+    },
+    {
+      "train_step": 8098,
+      "loss": 0.44827636701752055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44827636701752055
+    },
+    {
+      "train_step": 8099,
+      "loss": 0.46275191759247025,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46275191759247025
+    },
+    {
+      "train_step": 8100,
+      "loss": 0.7635351427397621,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7635351427397621
+    },
+    {
+      "train_step": 8101,
+      "loss": 0.3926942948246036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3926942948246036
+    },
+    {
+      "train_step": 8102,
+      "loss": 0.524988120922427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.524988120922427
+    },
+    {
+      "train_step": 8103,
+      "loss": 0.6008495695878758,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6008495695878758
+    },
+    {
+      "train_step": 8104,
+      "loss": 0.5982718472547471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5982718472547471
+    },
+    {
+      "train_step": 8105,
+      "loss": 0.46077499210497896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46077499210497896
+    },
+    {
+      "train_step": 8106,
+      "loss": 0.37764241768452256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37764241768452256
+    },
+    {
+      "train_step": 8107,
+      "loss": 0.45481659230703375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45481659230703375
+    },
+    {
+      "train_step": 8108,
+      "loss": 0.5965153922854776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5965153922854776
+    },
+    {
+      "train_step": 8109,
+      "loss": 0.5245505599578056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5245505599578056
+    },
+    {
+      "train_step": 8110,
+      "loss": 0.6483288755190186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6483288755190186
+    },
+    {
+      "train_step": 8111,
+      "loss": 0.411904402153329,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.411904402153329
+    },
+    {
+      "train_step": 8112,
+      "loss": 0.48310871402527733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48310871402527733
+    },
+    {
+      "train_step": 8113,
+      "loss": 0.6083027385122552,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6083027385122552
+    },
+    {
+      "train_step": 8114,
+      "loss": 0.4274806081854046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4274806081854046
+    },
+    {
+      "train_step": 8115,
+      "loss": 0.9885840128663563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9885840128663563
+    },
+    {
+      "train_step": 8116,
+      "loss": 0.7392453407217062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7392453407217062
+    },
+    {
+      "train_step": 8117,
+      "loss": 0.44434117238981063,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44434117238981063
+    },
+    {
+      "train_step": 8118,
+      "loss": 0.44880011998998365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44880011998998365
+    },
+    {
+      "train_step": 8119,
+      "loss": 0.4094721923277634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4094721923277634
+    },
+    {
+      "train_step": 8120,
+      "loss": 0.7949147169573002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7949147169573002
+    },
+    {
+      "train_step": 8121,
+      "loss": 0.5139477292131136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5139477292131136
+    },
+    {
+      "train_step": 8122,
+      "loss": 0.6397730034417899,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6397730034417899
+    },
+    {
+      "train_step": 8123,
+      "loss": 0.5164414175380169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5164414175380169
+    },
+    {
+      "train_step": 8124,
+      "loss": 0.5546280056896226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5546280056896226
+    },
+    {
+      "train_step": 8125,
+      "loss": 0.6138755408118648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6138755408118648
+    },
+    {
+      "train_step": 8126,
+      "loss": 0.32108769852872754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32108769852872754
+    },
+    {
+      "train_step": 8127,
+      "loss": 0.6782460418978041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6782460418978041
+    },
+    {
+      "train_step": 8128,
+      "loss": 0.5652856012887363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5652856012887363
+    },
+    {
+      "train_step": 8129,
+      "loss": 0.5497403634718426,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5497403634718426
+    },
+    {
+      "train_step": 8130,
+      "loss": 0.6177875538714659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6177875538714659
+    },
+    {
+      "train_step": 8131,
+      "loss": 0.6192041962939349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6192041962939349
+    },
+    {
+      "train_step": 8132,
+      "loss": 0.4550356839229135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4550356839229135
+    },
+    {
+      "train_step": 8133,
+      "loss": 0.6103804842520949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6103804842520949
+    },
+    {
+      "train_step": 8134,
+      "loss": 0.5057393284640932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5057393284640932
+    },
+    {
+      "train_step": 8135,
+      "loss": 0.5158338269049413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5158338269049413
+    },
+    {
+      "train_step": 8136,
+      "loss": 0.46747763895467787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46747763895467787
+    },
+    {
+      "train_step": 8137,
+      "loss": 0.3775258164720812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3775258164720812
+    },
+    {
+      "train_step": 8138,
+      "loss": 0.6807622977270841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6807622977270841
+    },
+    {
+      "train_step": 8139,
+      "loss": 0.6369083215096847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6369083215096847
+    },
+    {
+      "train_step": 8140,
+      "loss": 0.7433087993013436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7433087993013436
+    },
+    {
+      "train_step": 8141,
+      "loss": 0.5518315318222582,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5518315318222582
+    },
+    {
+      "train_step": 8142,
+      "loss": 0.6194240969701803,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6194240969701803
+    },
+    {
+      "train_step": 8143,
+      "loss": 0.48603029470525877,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48603029470525877
+    },
+    {
+      "train_step": 8144,
+      "loss": 0.5303945647685258,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5303945647685258
+    },
+    {
+      "train_step": 8145,
+      "loss": 0.5441207121742327,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441207121742327
+    },
+    {
+      "train_step": 8146,
+      "loss": 0.5966364797674151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5966364797674151
+    },
+    {
+      "train_step": 8147,
+      "loss": 0.47258110157468264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47258110157468264
+    },
+    {
+      "train_step": 8148,
+      "loss": 0.388471017585897,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.388471017585897
+    },
+    {
+      "train_step": 8149,
+      "loss": 0.5388201257887059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5388201257887059
+    },
+    {
+      "train_step": 8150,
+      "loss": 0.6559997753815394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6559997753815394
+    },
+    {
+      "train_step": 8151,
+      "loss": 0.4068537749800478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4068537749800478
+    },
+    {
+      "train_step": 8152,
+      "loss": 0.6573582949461055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6573582949461055
+    },
+    {
+      "train_step": 8153,
+      "loss": 0.6194641804728196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6194641804728196
+    },
+    {
+      "train_step": 8154,
+      "loss": 0.6368395413522472,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6368395413522472
+    },
+    {
+      "train_step": 8155,
+      "loss": 0.515852624979796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.515852624979796
+    },
+    {
+      "train_step": 8156,
+      "loss": 0.5052101731088553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5052101731088553
+    },
+    {
+      "train_step": 8157,
+      "loss": 0.5771723662706234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5771723662706234
+    },
+    {
+      "train_step": 8158,
+      "loss": 0.6104993902068745,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6104993902068745
+    },
+    {
+      "train_step": 8159,
+      "loss": 0.5396188107960316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5396188107960316
+    },
+    {
+      "train_step": 8160,
+      "loss": 0.752312112044862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.752312112044862
+    },
+    {
+      "train_step": 8161,
+      "loss": 0.5941685609543687,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5941685609543687
+    },
+    {
+      "train_step": 8162,
+      "loss": 0.6488883469585113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6488883469585113
+    },
+    {
+      "train_step": 8163,
+      "loss": 0.5700127232227552,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5700127232227552
+    },
+    {
+      "train_step": 8164,
+      "loss": 0.4479452196357737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4479452196357737
+    },
+    {
+      "train_step": 8165,
+      "loss": 0.4364741624879017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4364741624879017
+    },
+    {
+      "train_step": 8166,
+      "loss": 0.5937729600663665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5937729600663665
+    },
+    {
+      "train_step": 8167,
+      "loss": 0.4398439827269764,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4398439827269764
+    },
+    {
+      "train_step": 8168,
+      "loss": 0.6354941049845769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6354941049845769
+    },
+    {
+      "train_step": 8169,
+      "loss": 0.6907413606573666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6907413606573666
+    },
+    {
+      "train_step": 8170,
+      "loss": 0.5467984349050542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5467984349050542
+    },
+    {
+      "train_step": 8171,
+      "loss": 0.5762981427277698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5762981427277698
+    },
+    {
+      "train_step": 8172,
+      "loss": 0.8578027844115036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8578027844115036
+    },
+    {
+      "train_step": 8173,
+      "loss": 0.48892670312119607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48892670312119607
+    },
+    {
+      "train_step": 8174,
+      "loss": 0.686054173936844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.686054173936844
+    },
+    {
+      "train_step": 8175,
+      "loss": 0.646837138450971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.646837138450971
+    },
+    {
+      "train_step": 8176,
+      "loss": 0.5204959743453994,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5204959743453994
+    },
+    {
+      "train_step": 8177,
+      "loss": 0.7249675819938403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7249675819938403
+    },
+    {
+      "train_step": 8178,
+      "loss": 0.6018968105269618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6018968105269618
+    },
+    {
+      "train_step": 8179,
+      "loss": 0.6656981410258318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6656981410258318
+    },
+    {
+      "train_step": 8180,
+      "loss": 0.6205766039087516,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6205766039087516
+    },
+    {
+      "train_step": 8181,
+      "loss": 0.48973442459981614,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48973442459981614
+    },
+    {
+      "train_step": 8182,
+      "loss": 0.6059087758912929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6059087758912929
+    },
+    {
+      "train_step": 8183,
+      "loss": 0.7036147935835023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7036147935835023
+    },
+    {
+      "train_step": 8184,
+      "loss": 0.6113314974179375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6113314974179375
+    },
+    {
+      "train_step": 8185,
+      "loss": 0.610425122120491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.610425122120491
+    },
+    {
+      "train_step": 8186,
+      "loss": 0.4516378002923937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4516378002923937
+    },
+    {
+      "train_step": 8187,
+      "loss": 0.5931872276625164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5931872276625164
+    },
+    {
+      "train_step": 8188,
+      "loss": 0.5268324169763283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5268324169763283
+    },
+    {
+      "train_step": 8189,
+      "loss": 0.4605803411494199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4605803411494199
+    },
+    {
+      "train_step": 8190,
+      "loss": 0.46110067834555385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46110067834555385
+    },
+    {
+      "train_step": 8191,
+      "loss": 0.6780421301479381,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6780421301479381
+    },
+    {
+      "train_step": 8192,
+      "loss": 0.41519138876072503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41519138876072503
+    },
+    {
+      "train_step": 8193,
+      "loss": 0.8109826336213506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8109826336213506
+    },
+    {
+      "train_step": 8194,
+      "loss": 0.4472688178023125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4472688178023125
+    },
+    {
+      "train_step": 8195,
+      "loss": 0.40132507465528,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40132507465528
+    },
+    {
+      "train_step": 8196,
+      "loss": 0.6284691233421399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6284691233421399
+    },
+    {
+      "train_step": 8197,
+      "loss": 0.6538134486433588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6538134486433588
+    },
+    {
+      "train_step": 8198,
+      "loss": 0.37749337049369225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37749337049369225
+    },
+    {
+      "train_step": 8199,
+      "loss": 0.6633617505631159,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6633617505631159
+    },
+    {
+      "train_step": 8200,
+      "loss": 0.5486553295501158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5486553295501158
+    },
+    {
+      "train_step": 8201,
+      "loss": 0.5350703877324359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5350703877324359
+    },
+    {
+      "train_step": 8202,
+      "loss": 0.7377960460639641,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7377960460639641
+    },
+    {
+      "train_step": 8203,
+      "loss": 0.5355241074513115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5355241074513115
+    },
+    {
+      "train_step": 8204,
+      "loss": 0.649326812908161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.649326812908161
+    },
+    {
+      "train_step": 8205,
+      "loss": 0.5802863009443109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5802863009443109
+    },
+    {
+      "train_step": 8206,
+      "loss": 0.5430907857620082,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5430907857620082
+    },
+    {
+      "train_step": 8207,
+      "loss": 0.5420609453088406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5420609453088406
+    },
+    {
+      "train_step": 8208,
+      "loss": 0.41181961979027226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41181961979027226
+    },
+    {
+      "train_step": 8209,
+      "loss": 0.5770486387800736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5770486387800736
+    },
+    {
+      "train_step": 8210,
+      "loss": 0.5083703519126906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5083703519126906
+    },
+    {
+      "train_step": 8211,
+      "loss": 0.47087956084035676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47087956084035676
+    },
+    {
+      "train_step": 8212,
+      "loss": 0.6849959901557632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6849959901557632
+    },
+    {
+      "train_step": 8213,
+      "loss": 0.4634884267109383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4634884267109383
+    },
+    {
+      "train_step": 8214,
+      "loss": 0.7227049043144338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7227049043144338
+    },
+    {
+      "train_step": 8215,
+      "loss": 0.40247823752628403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40247823752628403
+    },
+    {
+      "train_step": 8216,
+      "loss": 0.5369050812711058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5369050812711058
+    },
+    {
+      "train_step": 8217,
+      "loss": 0.634627221962031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.634627221962031
+    },
+    {
+      "train_step": 8218,
+      "loss": 0.6314236780761774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6314236780761774
+    },
+    {
+      "train_step": 8219,
+      "loss": 0.5543523292240842,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5543523292240842
+    },
+    {
+      "train_step": 8220,
+      "loss": 0.4762095484239723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4762095484239723
+    },
+    {
+      "train_step": 8221,
+      "loss": 0.5250185608260527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5250185608260527
+    },
+    {
+      "train_step": 8222,
+      "loss": 0.48222102108334136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48222102108334136
+    },
+    {
+      "train_step": 8223,
+      "loss": 0.7125790598632136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7125790598632136
+    },
+    {
+      "train_step": 8224,
+      "loss": 0.522403484004735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.522403484004735
+    },
+    {
+      "train_step": 8225,
+      "loss": 0.5492733620273996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5492733620273996
+    },
+    {
+      "train_step": 8226,
+      "loss": 0.7053912822544529,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7053912822544529
+    },
+    {
+      "train_step": 8227,
+      "loss": 0.5026676852907427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5026676852907427
+    },
+    {
+      "train_step": 8228,
+      "loss": 0.557298495663933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.557298495663933
+    },
+    {
+      "train_step": 8229,
+      "loss": 0.6351648447713286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6351648447713286
+    },
+    {
+      "train_step": 8230,
+      "loss": 0.4591966710145969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4591966710145969
+    },
+    {
+      "train_step": 8231,
+      "loss": 0.4997048501683858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4997048501683858
+    },
+    {
+      "train_step": 8232,
+      "loss": 0.42695809123781103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42695809123781103
+    },
+    {
+      "train_step": 8233,
+      "loss": 0.40558186224550724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40558186224550724
+    },
+    {
+      "train_step": 8234,
+      "loss": 0.5078656104908285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5078656104908285
+    },
+    {
+      "train_step": 8235,
+      "loss": 0.49769162975463677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49769162975463677
+    },
+    {
+      "train_step": 8236,
+      "loss": 0.7668846923869458,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7668846923869458
+    },
+    {
+      "train_step": 8237,
+      "loss": 0.5906419663658776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5906419663658776
+    },
+    {
+      "train_step": 8238,
+      "loss": 0.6201333314931781,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6201333314931781
+    },
+    {
+      "train_step": 8239,
+      "loss": 0.6295035981341719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6295035981341719
+    },
+    {
+      "train_step": 8240,
+      "loss": 0.5912168465965296,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5912168465965296
+    },
+    {
+      "train_step": 8241,
+      "loss": 0.6284482342525822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6284482342525822
+    },
+    {
+      "train_step": 8242,
+      "loss": 0.7908888842265852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7908888842265852
+    },
+    {
+      "train_step": 8243,
+      "loss": 0.7285119979389785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7285119979389785
+    },
+    {
+      "train_step": 8244,
+      "loss": 0.5729020680093982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5729020680093982
+    },
+    {
+      "train_step": 8245,
+      "loss": 0.4061551061443036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4061551061443036
+    },
+    {
+      "train_step": 8246,
+      "loss": 0.5867878728564877,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5867878728564877
+    },
+    {
+      "train_step": 8247,
+      "loss": 0.44375492058209637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44375492058209637
+    },
+    {
+      "train_step": 8248,
+      "loss": 0.5604408381347427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5604408381347427
+    },
+    {
+      "train_step": 8249,
+      "loss": 0.5825856612019555,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5825856612019555
+    },
+    {
+      "train_step": 8250,
+      "loss": 0.6277238253329089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6277238253329089
+    },
+    {
+      "train_step": 8251,
+      "loss": 0.6832495047488498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6832495047488498
+    },
+    {
+      "train_step": 8252,
+      "loss": 0.5911564482533599,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5911564482533599
+    },
+    {
+      "train_step": 8253,
+      "loss": 0.46286268291656896,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46286268291656896
+    },
+    {
+      "train_step": 8254,
+      "loss": 0.5518266092586742,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5518266092586742
+    },
+    {
+      "train_step": 8255,
+      "loss": 0.5863182507465878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5863182507465878
+    },
+    {
+      "train_step": 8256,
+      "loss": 0.5798154664895939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5798154664895939
+    },
+    {
+      "train_step": 8257,
+      "loss": 0.6769428949630799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6769428949630799
+    },
+    {
+      "train_step": 8258,
+      "loss": 0.7205988218871621,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7205988218871621
+    },
+    {
+      "train_step": 8259,
+      "loss": 0.5681745116418762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5681745116418762
+    },
+    {
+      "train_step": 8260,
+      "loss": 0.6728851525417977,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6728851525417977
+    },
+    {
+      "train_step": 8261,
+      "loss": 0.5781720603398613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5781720603398613
+    },
+    {
+      "train_step": 8262,
+      "loss": 0.5970714117612966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5970714117612966
+    },
+    {
+      "train_step": 8263,
+      "loss": 0.45249028792599166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45249028792599166
+    },
+    {
+      "train_step": 8264,
+      "loss": 0.6597428443383172,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6597428443383172
+    },
+    {
+      "train_step": 8265,
+      "loss": 0.6981439033411652,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6981439033411652
+    },
+    {
+      "train_step": 8266,
+      "loss": 0.5111776655583476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5111776655583476
+    },
+    {
+      "train_step": 8267,
+      "loss": 0.2998560449943622,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2998560449943622
+    },
+    {
+      "train_step": 8268,
+      "loss": 0.7406461536188111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7406461536188111
+    },
+    {
+      "train_step": 8269,
+      "loss": 0.3432196417292385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3432196417292385
+    },
+    {
+      "train_step": 8270,
+      "loss": 0.7501228918893541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7501228918893541
+    },
+    {
+      "train_step": 8271,
+      "loss": 0.5702181494469347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5702181494469347
+    },
+    {
+      "train_step": 8272,
+      "loss": 0.501583369338075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.501583369338075
+    },
+    {
+      "train_step": 8273,
+      "loss": 0.47988649844259074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47988649844259074
+    },
+    {
+      "train_step": 8274,
+      "loss": 0.6732589013810821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6732589013810821
+    },
+    {
+      "train_step": 8275,
+      "loss": 0.48450582686906685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48450582686906685
+    },
+    {
+      "train_step": 8276,
+      "loss": 0.5634113177404797,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5634113177404797
+    },
+    {
+      "train_step": 8277,
+      "loss": 0.5507344444564475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5507344444564475
+    },
+    {
+      "train_step": 8278,
+      "loss": 0.5341329004997416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5341329004997416
+    },
+    {
+      "train_step": 8279,
+      "loss": 0.2208612650785965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2208612650785965
+    },
+    {
+      "train_step": 8280,
+      "loss": 0.738105912670525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.738105912670525
+    },
+    {
+      "train_step": 8281,
+      "loss": 0.3846782192966734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3846782192966734
+    },
+    {
+      "train_step": 8282,
+      "loss": 0.692892790926313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.692892790926313
+    },
+    {
+      "train_step": 8283,
+      "loss": 0.4725610913805459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4725610913805459
+    },
+    {
+      "train_step": 8284,
+      "loss": 0.7558970327465372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7558970327465372
+    },
+    {
+      "train_step": 8285,
+      "loss": 0.6531327673377298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6531327673377298
+    },
+    {
+      "train_step": 8286,
+      "loss": 0.448771026319438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.448771026319438
+    },
+    {
+      "train_step": 8287,
+      "loss": 0.5666075307169264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5666075307169264
+    },
+    {
+      "train_step": 8288,
+      "loss": 0.5591169726165425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5591169726165425
+    },
+    {
+      "train_step": 8289,
+      "loss": 0.5506421293717305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5506421293717305
+    },
+    {
+      "train_step": 8290,
+      "loss": 0.6416002142229732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6416002142229732
+    },
+    {
+      "train_step": 8291,
+      "loss": 0.45041258292991126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45041258292991126
+    },
+    {
+      "train_step": 8292,
+      "loss": 0.5938136171456802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5938136171456802
+    },
+    {
+      "train_step": 8293,
+      "loss": 0.6283776924211077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6283776924211077
+    },
+    {
+      "train_step": 8294,
+      "loss": 0.5818453107685456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5818453107685456
+    },
+    {
+      "train_step": 8295,
+      "loss": 0.8380957184425183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8380957184425183
+    },
+    {
+      "train_step": 8296,
+      "loss": 0.4405623210523507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4405623210523507
+    },
+    {
+      "train_step": 8297,
+      "loss": 0.4192644022095248,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4192644022095248
+    },
+    {
+      "train_step": 8298,
+      "loss": 0.46959141816412,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46959141816412
+    },
+    {
+      "train_step": 8299,
+      "loss": 0.6531074196730916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6531074196730916
+    },
+    {
+      "train_step": 8300,
+      "loss": 0.5906080533226783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5906080533226783
+    },
+    {
+      "train_step": 8301,
+      "loss": 0.6221435283439076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6221435283439076
+    },
+    {
+      "train_step": 8302,
+      "loss": 0.5773560037229568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5773560037229568
+    },
+    {
+      "train_step": 8303,
+      "loss": 0.7745904165517468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7745904165517468
+    },
+    {
+      "train_step": 8304,
+      "loss": 0.5235529525410189,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5235529525410189
+    },
+    {
+      "train_step": 8305,
+      "loss": 0.8251620080135071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8251620080135071
+    },
+    {
+      "train_step": 8306,
+      "loss": 0.7002958806430106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7002958806430106
+    },
+    {
+      "train_step": 8307,
+      "loss": 0.4326520862034074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4326520862034074
+    },
+    {
+      "train_step": 8308,
+      "loss": 0.45155651676938907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45155651676938907
+    },
+    {
+      "train_step": 8309,
+      "loss": 0.5186197867398895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5186197867398895
+    },
+    {
+      "train_step": 8310,
+      "loss": 0.4634178198449534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4634178198449534
+    },
+    {
+      "train_step": 8311,
+      "loss": 0.6103343045171421,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6103343045171421
+    },
+    {
+      "train_step": 8312,
+      "loss": 0.5364422014935925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5364422014935925
+    },
+    {
+      "train_step": 8313,
+      "loss": 0.3668263494185514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3668263494185514
+    },
+    {
+      "train_step": 8314,
+      "loss": 0.6212241635249998,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6212241635249998
+    },
+    {
+      "train_step": 8315,
+      "loss": 0.8825940708038653,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8825940708038653
+    },
+    {
+      "train_step": 8316,
+      "loss": 0.5146970076951338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5146970076951338
+    },
+    {
+      "train_step": 8317,
+      "loss": 0.4934188721532828,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4934188721532828
+    },
+    {
+      "train_step": 8318,
+      "loss": 0.5017713940765878,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5017713940765878
+    },
+    {
+      "train_step": 8319,
+      "loss": 0.6273954986078037,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6273954986078037
+    },
+    {
+      "train_step": 8320,
+      "loss": 0.39782752634631613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39782752634631613
+    },
+    {
+      "train_step": 8321,
+      "loss": 0.33326453417560076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33326453417560076
+    },
+    {
+      "train_step": 8322,
+      "loss": 0.7798690624105191,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7798690624105191
+    },
+    {
+      "train_step": 8323,
+      "loss": 0.4228719555860409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4228719555860409
+    },
+    {
+      "train_step": 8324,
+      "loss": 0.4622601900527969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4622601900527969
+    },
+    {
+      "train_step": 8325,
+      "loss": 0.6131726347918545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6131726347918545
+    },
+    {
+      "train_step": 8326,
+      "loss": 0.5742928543804625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5742928543804625
+    },
+    {
+      "train_step": 8327,
+      "loss": 0.6906799383429243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6906799383429243
+    },
+    {
+      "train_step": 8328,
+      "loss": 0.5284324842322808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5284324842322808
+    },
+    {
+      "train_step": 8329,
+      "loss": 0.5113676126466096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5113676126466096
+    },
+    {
+      "train_step": 8330,
+      "loss": 0.7433344701425697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7433344701425697
+    },
+    {
+      "train_step": 8331,
+      "loss": 0.4643918173512755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4643918173512755
+    },
+    {
+      "train_step": 8332,
+      "loss": 0.3713501281855323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3713501281855323
+    },
+    {
+      "train_step": 8333,
+      "loss": 0.35435424464881254,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35435424464881254
+    },
+    {
+      "train_step": 8334,
+      "loss": 0.5030069354127652,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5030069354127652
+    },
+    {
+      "train_step": 8335,
+      "loss": 0.5853719183554729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5853719183554729
+    },
+    {
+      "train_step": 8336,
+      "loss": 0.4473875889294611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4473875889294611
+    },
+    {
+      "train_step": 8337,
+      "loss": 0.6168475048413591,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6168475048413591
+    },
+    {
+      "train_step": 8338,
+      "loss": 0.6212432440365615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6212432440365615
+    },
+    {
+      "train_step": 8339,
+      "loss": 0.40011284225518595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40011284225518595
+    },
+    {
+      "train_step": 8340,
+      "loss": 0.6171308394153427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6171308394153427
+    },
+    {
+      "train_step": 8341,
+      "loss": 0.4377220317059108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4377220317059108
+    },
+    {
+      "train_step": 8342,
+      "loss": 0.5448270000080108,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5448270000080108
+    },
+    {
+      "train_step": 8343,
+      "loss": 0.5990729824093041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5990729824093041
+    },
+    {
+      "train_step": 8344,
+      "loss": 0.289060486441012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.289060486441012
+    },
+    {
+      "train_step": 8345,
+      "loss": 0.7064499178083576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7064499178083576
+    },
+    {
+      "train_step": 8346,
+      "loss": 0.5441353304302691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441353304302691
+    },
+    {
+      "train_step": 8347,
+      "loss": 0.6526591704060666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6526591704060666
+    },
+    {
+      "train_step": 8348,
+      "loss": 0.6051813969652136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6051813969652136
+    },
+    {
+      "train_step": 8349,
+      "loss": 0.5836126237073752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5836126237073752
+    },
+    {
+      "train_step": 8350,
+      "loss": 0.46853172644619157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46853172644619157
+    },
+    {
+      "train_step": 8351,
+      "loss": 0.6688731689053882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6688731689053882
+    },
+    {
+      "train_step": 8352,
+      "loss": 0.4540210624069349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4540210624069349
+    },
+    {
+      "train_step": 8353,
+      "loss": 0.4296881678916352,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4296881678916352
+    },
+    {
+      "train_step": 8354,
+      "loss": 0.5283593023409857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5283593023409857
+    },
+    {
+      "train_step": 8355,
+      "loss": 0.3767597437257379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3767597437257379
+    },
+    {
+      "train_step": 8356,
+      "loss": 0.44444136280497076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44444136280497076
+    },
+    {
+      "train_step": 8357,
+      "loss": 0.49873935630100413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49873935630100413
+    },
+    {
+      "train_step": 8358,
+      "loss": 0.5427408204384772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5427408204384772
+    },
+    {
+      "train_step": 8359,
+      "loss": 0.6084390924525833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6084390924525833
+    },
+    {
+      "train_step": 8360,
+      "loss": 0.7088273685522024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7088273685522024
+    },
+    {
+      "train_step": 8361,
+      "loss": 0.45201476977452054,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45201476977452054
+    },
+    {
+      "train_step": 8362,
+      "loss": 0.483216555811355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.483216555811355
+    },
+    {
+      "train_step": 8363,
+      "loss": 0.6006420533581119,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6006420533581119
+    },
+    {
+      "train_step": 8364,
+      "loss": 0.47735324562735193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47735324562735193
+    },
+    {
+      "train_step": 8365,
+      "loss": 0.7275618162376162,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7275618162376162
+    },
+    {
+      "train_step": 8366,
+      "loss": 0.5804681187842033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5804681187842033
+    },
+    {
+      "train_step": 8367,
+      "loss": 0.5569104342420562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5569104342420562
+    },
+    {
+      "train_step": 8368,
+      "loss": 0.4925777574756368,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4925777574756368
+    },
+    {
+      "train_step": 8369,
+      "loss": 0.6042031594221459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6042031594221459
+    },
+    {
+      "train_step": 8370,
+      "loss": 0.3265182816999965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3265182816999965
+    },
+    {
+      "train_step": 8371,
+      "loss": 0.6692227385503835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6692227385503835
+    },
+    {
+      "train_step": 8372,
+      "loss": 0.5284540115767787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5284540115767787
+    },
+    {
+      "train_step": 8373,
+      "loss": 0.5665627326071333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5665627326071333
+    },
+    {
+      "train_step": 8374,
+      "loss": 0.7233901097874825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7233901097874825
+    },
+    {
+      "train_step": 8375,
+      "loss": 0.47710390803044533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47710390803044533
+    },
+    {
+      "train_step": 8376,
+      "loss": 0.5960080371113605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5960080371113605
+    },
+    {
+      "train_step": 8377,
+      "loss": 0.42787124260990417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42787124260990417
+    },
+    {
+      "train_step": 8378,
+      "loss": 0.512493194220932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.512493194220932
+    },
+    {
+      "train_step": 8379,
+      "loss": 0.4360735387500463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4360735387500463
+    },
+    {
+      "train_step": 8380,
+      "loss": 0.5885046287389649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5885046287389649
+    },
+    {
+      "train_step": 8381,
+      "loss": 0.5827602766463685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5827602766463685
+    },
+    {
+      "train_step": 8382,
+      "loss": 0.47089692008200323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47089692008200323
+    },
+    {
+      "train_step": 8383,
+      "loss": 0.6666572674666543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6666572674666543
+    },
+    {
+      "train_step": 8384,
+      "loss": 0.637134576124259,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.637134576124259
+    },
+    {
+      "train_step": 8385,
+      "loss": 0.7473242196978624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7473242196978624
+    },
+    {
+      "train_step": 8386,
+      "loss": 0.8396101924802383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8396101924802383
+    },
+    {
+      "train_step": 8387,
+      "loss": 0.5277331231944257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5277331231944257
+    },
+    {
+      "train_step": 8388,
+      "loss": 0.28868886084266737,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.28868886084266737
+    },
+    {
+      "train_step": 8389,
+      "loss": 0.6193995369583432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6193995369583432
+    },
+    {
+      "train_step": 8390,
+      "loss": 0.6130993045978972,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6130993045978972
+    },
+    {
+      "train_step": 8391,
+      "loss": 0.4728698024465416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4728698024465416
+    },
+    {
+      "train_step": 8392,
+      "loss": 0.5686385197570498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5686385197570498
+    },
+    {
+      "train_step": 8393,
+      "loss": 0.5530370400464941,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5530370400464941
+    },
+    {
+      "train_step": 8394,
+      "loss": 0.4925380417694285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4925380417694285
+    },
+    {
+      "train_step": 8395,
+      "loss": 0.563849010962642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.563849010962642
+    },
+    {
+      "train_step": 8396,
+      "loss": 0.3840056145076004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3840056145076004
+    },
+    {
+      "train_step": 8397,
+      "loss": 0.5931449779397165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5931449779397165
+    },
+    {
+      "train_step": 8398,
+      "loss": 0.5605135052733005,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5605135052733005
+    },
+    {
+      "train_step": 8399,
+      "loss": 0.5176158935120605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5176158935120605
+    },
+    {
+      "train_step": 8400,
+      "loss": 0.6088061741969846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6088061741969846
+    },
+    {
+      "train_step": 8401,
+      "loss": 0.6133733023852157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6133733023852157
+    },
+    {
+      "train_step": 8402,
+      "loss": 0.6160377402194579,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6160377402194579
+    },
+    {
+      "train_step": 8403,
+      "loss": 0.8362591598786546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8362591598786546
+    },
+    {
+      "train_step": 8404,
+      "loss": 0.34962946085628654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34962946085628654
+    },
+    {
+      "train_step": 8405,
+      "loss": 0.46187723271528225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46187723271528225
+    },
+    {
+      "train_step": 8406,
+      "loss": 0.579451125195241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.579451125195241
+    },
+    {
+      "train_step": 8407,
+      "loss": 0.5956404251373487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5956404251373487
+    },
+    {
+      "train_step": 8408,
+      "loss": 0.3839647975043003,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3839647975043003
+    },
+    {
+      "train_step": 8409,
+      "loss": 0.5474777369615985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5474777369615985
+    },
+    {
+      "train_step": 8410,
+      "loss": 0.5658876354692995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5658876354692995
+    },
+    {
+      "train_step": 8411,
+      "loss": 0.39099047526883085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39099047526883085
+    },
+    {
+      "train_step": 8412,
+      "loss": 0.5997503384327971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5997503384327971
+    },
+    {
+      "train_step": 8413,
+      "loss": 0.5349399106426594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5349399106426594
+    },
+    {
+      "train_step": 8414,
+      "loss": 0.444431574256183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.444431574256183
+    },
+    {
+      "train_step": 8415,
+      "loss": 0.6938547048466946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6938547048466946
+    },
+    {
+      "train_step": 8416,
+      "loss": 0.5561063212340526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5561063212340526
+    },
+    {
+      "train_step": 8417,
+      "loss": 0.5208142331485043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5208142331485043
+    },
+    {
+      "train_step": 8418,
+      "loss": 0.4883411563033501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4883411563033501
+    },
+    {
+      "train_step": 8419,
+      "loss": 0.40512913572209497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40512913572209497
+    },
+    {
+      "train_step": 8420,
+      "loss": 0.6700151083571412,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6700151083571412
+    },
+    {
+      "train_step": 8421,
+      "loss": 0.2877859317582952,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2877859317582952
+    },
+    {
+      "train_step": 8422,
+      "loss": 0.46597155161578696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46597155161578696
+    },
+    {
+      "train_step": 8423,
+      "loss": 0.30177248520472977,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.30177248520472977
+    },
+    {
+      "train_step": 8424,
+      "loss": 0.44357992529918955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44357992529918955
+    },
+    {
+      "train_step": 8425,
+      "loss": 0.5038203110385822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5038203110385822
+    },
+    {
+      "train_step": 8426,
+      "loss": 0.5975408304035659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5975408304035659
+    },
+    {
+      "train_step": 8427,
+      "loss": 0.5076103274284157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5076103274284157
+    },
+    {
+      "train_step": 8428,
+      "loss": 0.45022115557997794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45022115557997794
+    },
+    {
+      "train_step": 8429,
+      "loss": 0.6437889675906702,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6437889675906702
+    },
+    {
+      "train_step": 8430,
+      "loss": 0.49077960627752815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49077960627752815
+    },
+    {
+      "train_step": 8431,
+      "loss": 0.6009604539155026,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6009604539155026
+    },
+    {
+      "train_step": 8432,
+      "loss": 0.5419297935233016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5419297935233016
+    },
+    {
+      "train_step": 8433,
+      "loss": 0.6201382476457233,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6201382476457233
+    },
+    {
+      "train_step": 8434,
+      "loss": 0.3822578322892238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3822578322892238
+    },
+    {
+      "train_step": 8435,
+      "loss": 0.6042176797874523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6042176797874523
+    },
+    {
+      "train_step": 8436,
+      "loss": 0.5725472515651373,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5725472515651373
+    },
+    {
+      "train_step": 8437,
+      "loss": 0.7474384796592264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7474384796592264
+    },
+    {
+      "train_step": 8438,
+      "loss": 0.6803872706721386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6803872706721386
+    },
+    {
+      "train_step": 8439,
+      "loss": 0.4038226127853483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4038226127853483
+    },
+    {
+      "train_step": 8440,
+      "loss": 0.5246345823700593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5246345823700593
+    },
+    {
+      "train_step": 8441,
+      "loss": 0.5300517750807469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5300517750807469
+    },
+    {
+      "train_step": 8442,
+      "loss": 0.7683008758811827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7683008758811827
+    },
+    {
+      "train_step": 8443,
+      "loss": 0.5470483864608399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5470483864608399
+    },
+    {
+      "train_step": 8444,
+      "loss": 0.5366697331983773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5366697331983773
+    },
+    {
+      "train_step": 8445,
+      "loss": 0.5892208256526051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5892208256526051
+    },
+    {
+      "train_step": 8446,
+      "loss": 0.48744606422651865,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48744606422651865
+    },
+    {
+      "train_step": 8447,
+      "loss": 0.43191710053598004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43191710053598004
+    },
+    {
+      "train_step": 8448,
+      "loss": 0.5540482910910668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5540482910910668
+    },
+    {
+      "train_step": 8449,
+      "loss": 0.7273861860371852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7273861860371852
+    },
+    {
+      "train_step": 8450,
+      "loss": 0.37607004289832513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37607004289832513
+    },
+    {
+      "train_step": 8451,
+      "loss": 0.5000812845200291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5000812845200291
+    },
+    {
+      "train_step": 8452,
+      "loss": 0.6311981558757769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6311981558757769
+    },
+    {
+      "train_step": 8453,
+      "loss": 0.5991898549134771,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5991898549134771
+    },
+    {
+      "train_step": 8454,
+      "loss": 0.6760021768461256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6760021768461256
+    },
+    {
+      "train_step": 8455,
+      "loss": 0.5886354714376161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5886354714376161
+    },
+    {
+      "train_step": 8456,
+      "loss": 0.5316026670043534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5316026670043534
+    },
+    {
+      "train_step": 8457,
+      "loss": 0.5111781193826711,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5111781193826711
+    },
+    {
+      "train_step": 8458,
+      "loss": 0.4237604522730679,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4237604522730679
+    },
+    {
+      "train_step": 8459,
+      "loss": 0.7110992959137424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7110992959137424
+    },
+    {
+      "train_step": 8460,
+      "loss": 0.39952441733483224,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39952441733483224
+    },
+    {
+      "train_step": 8461,
+      "loss": 0.7673927252621693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7673927252621693
+    },
+    {
+      "train_step": 8462,
+      "loss": 0.5449494633209417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5449494633209417
+    },
+    {
+      "train_step": 8463,
+      "loss": 0.3341258022589209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3341258022589209
+    },
+    {
+      "train_step": 8464,
+      "loss": 0.47963424945675237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47963424945675237
+    },
+    {
+      "train_step": 8465,
+      "loss": 0.3920906165969527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3920906165969527
+    },
+    {
+      "train_step": 8466,
+      "loss": 0.7768775622196882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7768775622196882
+    },
+    {
+      "train_step": 8467,
+      "loss": 0.5285468089320905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5285468089320905
+    },
+    {
+      "train_step": 8468,
+      "loss": 0.403324778248219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.403324778248219
+    },
+    {
+      "train_step": 8469,
+      "loss": 0.5186914792494817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5186914792494817
+    },
+    {
+      "train_step": 8470,
+      "loss": 0.4461174913004482,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4461174913004482
+    },
+    {
+      "train_step": 8471,
+      "loss": 0.30754585676559215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.30754585676559215
+    },
+    {
+      "train_step": 8472,
+      "loss": 0.556536355394823,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.556536355394823
+    },
+    {
+      "train_step": 8473,
+      "loss": 0.5210761548856117,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5210761548856117
+    },
+    {
+      "train_step": 8474,
+      "loss": 0.5334073963493415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5334073963493415
+    },
+    {
+      "train_step": 8475,
+      "loss": 0.5417522635581303,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5417522635581303
+    },
+    {
+      "train_step": 8476,
+      "loss": 0.5519471151076927,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5519471151076927
+    },
+    {
+      "train_step": 8477,
+      "loss": 0.6487073631709974,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6487073631709974
+    },
+    {
+      "train_step": 8478,
+      "loss": 0.5413105443837497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5413105443837497
+    },
+    {
+      "train_step": 8479,
+      "loss": 0.5706050279549012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706050279549012
+    },
+    {
+      "train_step": 8480,
+      "loss": 0.7089076669326004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7089076669326004
+    },
+    {
+      "train_step": 8481,
+      "loss": 0.6197468928465205,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6197468928465205
+    },
+    {
+      "train_step": 8482,
+      "loss": 0.3822502340384535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3822502340384535
+    },
+    {
+      "train_step": 8483,
+      "loss": 0.59466717483773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.59466717483773
+    },
+    {
+      "train_step": 8484,
+      "loss": 0.54948882606474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.54948882606474
+    },
+    {
+      "train_step": 8485,
+      "loss": 0.6115114231120479,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6115114231120479
+    },
+    {
+      "train_step": 8486,
+      "loss": 0.4695215491032245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4695215491032245
+    },
+    {
+      "train_step": 8487,
+      "loss": 0.5105535892640887,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5105535892640887
+    },
+    {
+      "train_step": 8488,
+      "loss": 0.6710218254965897,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6710218254965897
+    },
+    {
+      "train_step": 8489,
+      "loss": 0.5028128211158618,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5028128211158618
+    },
+    {
+      "train_step": 8490,
+      "loss": 0.5569573585130988,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5569573585130988
+    },
+    {
+      "train_step": 8491,
+      "loss": 0.4333619729226089,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4333619729226089
+    },
+    {
+      "train_step": 8492,
+      "loss": 0.6206397062475678,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6206397062475678
+    },
+    {
+      "train_step": 8493,
+      "loss": 0.559148774478581,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.559148774478581
+    },
+    {
+      "train_step": 8494,
+      "loss": 0.6838000747926888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6838000747926888
+    },
+    {
+      "train_step": 8495,
+      "loss": 0.6788285222767979,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6788285222767979
+    },
+    {
+      "train_step": 8496,
+      "loss": 0.5617752410143317,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5617752410143317
+    },
+    {
+      "train_step": 8497,
+      "loss": 0.3398732773459869,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3398732773459869
+    },
+    {
+      "train_step": 8498,
+      "loss": 0.47861578983256065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47861578983256065
+    },
+    {
+      "train_step": 8499,
+      "loss": 0.5129435754445503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5129435754445503
+    },
+    {
+      "train_step": 8500,
+      "loss": 0.5823178015919532,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5823178015919532
+    },
+    {
+      "train_step": 8501,
+      "loss": 0.4968279470811656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4968279470811656
+    },
+    {
+      "train_step": 8502,
+      "loss": 0.4564834170878396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4564834170878396
+    },
+    {
+      "train_step": 8503,
+      "loss": 0.7759668855167774,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7759668855167774
+    },
+    {
+      "train_step": 8504,
+      "loss": 0.5386730751558354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5386730751558354
+    },
+    {
+      "train_step": 8505,
+      "loss": 0.48786221674320396,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48786221674320396
+    },
+    {
+      "train_step": 8506,
+      "loss": 0.5440026010859583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5440026010859583
+    },
+    {
+      "train_step": 8507,
+      "loss": 0.6031614722659301,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6031614722659301
+    },
+    {
+      "train_step": 8508,
+      "loss": 0.5982348961433871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5982348961433871
+    },
+    {
+      "train_step": 8509,
+      "loss": 0.4275192266714709,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4275192266714709
+    },
+    {
+      "train_step": 8510,
+      "loss": 0.6319250015236588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6319250015236588
+    },
+    {
+      "train_step": 8511,
+      "loss": 0.776566985379214,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.776566985379214
+    },
+    {
+      "train_step": 8512,
+      "loss": 0.6857790551040346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6857790551040346
+    },
+    {
+      "train_step": 8513,
+      "loss": 0.6074163345582355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6074163345582355
+    },
+    {
+      "train_step": 8514,
+      "loss": 0.5813868902732507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5813868902732507
+    },
+    {
+      "train_step": 8515,
+      "loss": 0.5455837537295634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5455837537295634
+    },
+    {
+      "train_step": 8516,
+      "loss": 0.715661210866038,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.715661210866038
+    },
+    {
+      "train_step": 8517,
+      "loss": 0.5193277553382489,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5193277553382489
+    },
+    {
+      "train_step": 8518,
+      "loss": 0.8240432078583241,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8240432078583241
+    },
+    {
+      "train_step": 8519,
+      "loss": 0.4540992383825265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4540992383825265
+    },
+    {
+      "train_step": 8520,
+      "loss": 0.6817750645189438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6817750645189438
+    },
+    {
+      "train_step": 8521,
+      "loss": 0.50369301801569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.50369301801569
+    },
+    {
+      "train_step": 8522,
+      "loss": 0.5463515224292507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5463515224292507
+    },
+    {
+      "train_step": 8523,
+      "loss": 0.601443607169167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.601443607169167
+    },
+    {
+      "train_step": 8524,
+      "loss": 0.7068923589874048,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7068923589874048
+    },
+    {
+      "train_step": 8525,
+      "loss": 0.6057149513942331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6057149513942331
+    },
+    {
+      "train_step": 8526,
+      "loss": 0.613483051596752,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.613483051596752
+    },
+    {
+      "train_step": 8527,
+      "loss": 0.5434862193148874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5434862193148874
+    },
+    {
+      "train_step": 8528,
+      "loss": 0.5104144268832609,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5104144268832609
+    },
+    {
+      "train_step": 8529,
+      "loss": 0.5649815346166643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5649815346166643
+    },
+    {
+      "train_step": 8530,
+      "loss": 0.622950744697389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.622950744697389
+    },
+    {
+      "train_step": 8531,
+      "loss": 0.6915197951517362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6915197951517362
+    },
+    {
+      "train_step": 8532,
+      "loss": 0.4721431918896884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4721431918896884
+    },
+    {
+      "train_step": 8533,
+      "loss": 0.5562865671707715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5562865671707715
+    },
+    {
+      "train_step": 8534,
+      "loss": 0.6255820384289792,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6255820384289792
+    },
+    {
+      "train_step": 8535,
+      "loss": 0.4175344644123418,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4175344644123418
+    },
+    {
+      "train_step": 8536,
+      "loss": 0.7365225781616502,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7365225781616502
+    },
+    {
+      "train_step": 8537,
+      "loss": 0.4415479703391861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4415479703391861
+    },
+    {
+      "train_step": 8538,
+      "loss": 0.7214147794047185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7214147794047185
+    },
+    {
+      "train_step": 8539,
+      "loss": 0.5443772049442204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5443772049442204
+    },
+    {
+      "train_step": 8540,
+      "loss": 0.3953751563640371,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3953751563640371
+    },
+    {
+      "train_step": 8541,
+      "loss": 0.37233355562451176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37233355562451176
+    },
+    {
+      "train_step": 8542,
+      "loss": 0.4281219257490325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4281219257490325
+    },
+    {
+      "train_step": 8543,
+      "loss": 0.41394776530333255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41394776530333255
+    },
+    {
+      "train_step": 8544,
+      "loss": 0.5671210617723901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5671210617723901
+    },
+    {
+      "train_step": 8545,
+      "loss": 0.482033702723688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.482033702723688
+    },
+    {
+      "train_step": 8546,
+      "loss": 0.3617466182114659,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3617466182114659
+    },
+    {
+      "train_step": 8547,
+      "loss": 0.5262582833346185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5262582833346185
+    },
+    {
+      "train_step": 8548,
+      "loss": 0.506548465422531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.506548465422531
+    },
+    {
+      "train_step": 8549,
+      "loss": 0.8440649125658534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8440649125658534
+    },
+    {
+      "train_step": 8550,
+      "loss": 0.48763745707577333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48763745707577333
+    },
+    {
+      "train_step": 8551,
+      "loss": 0.4913456413558002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4913456413558002
+    },
+    {
+      "train_step": 8552,
+      "loss": 0.5329295146856876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5329295146856876
+    },
+    {
+      "train_step": 8553,
+      "loss": 0.40220300679220006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40220300679220006
+    },
+    {
+      "train_step": 8554,
+      "loss": 0.43170837902787107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43170837902787107
+    },
+    {
+      "train_step": 8555,
+      "loss": 0.7394576005608309,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7394576005608309
+    },
+    {
+      "train_step": 8556,
+      "loss": 0.5855424028983331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5855424028983331
+    },
+    {
+      "train_step": 8557,
+      "loss": 0.37875290768417275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37875290768417275
+    },
+    {
+      "train_step": 8558,
+      "loss": 0.6621393946945919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6621393946945919
+    },
+    {
+      "train_step": 8559,
+      "loss": 0.5443828952465726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5443828952465726
+    },
+    {
+      "train_step": 8560,
+      "loss": 0.517059101376388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.517059101376388
+    },
+    {
+      "train_step": 8561,
+      "loss": 0.4560894833862675,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4560894833862675
+    },
+    {
+      "train_step": 8562,
+      "loss": 0.5146405225308162,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5146405225308162
+    },
+    {
+      "train_step": 8563,
+      "loss": 0.4329281899543586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4329281899543586
+    },
+    {
+      "train_step": 8564,
+      "loss": 0.44680322602644473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44680322602644473
+    },
+    {
+      "train_step": 8565,
+      "loss": 0.46688441473776787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46688441473776787
+    },
+    {
+      "train_step": 8566,
+      "loss": 0.4670266853498732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4670266853498732
+    },
+    {
+      "train_step": 8567,
+      "loss": 0.589383061473238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.589383061473238
+    },
+    {
+      "train_step": 8568,
+      "loss": 0.4626973483787178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4626973483787178
+    },
+    {
+      "train_step": 8569,
+      "loss": 0.5973775106566002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5973775106566002
+    },
+    {
+      "train_step": 8570,
+      "loss": 0.38001915780198114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38001915780198114
+    },
+    {
+      "train_step": 8571,
+      "loss": 0.7285132971705333,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7285132971705333
+    },
+    {
+      "train_step": 8572,
+      "loss": 0.6033072887347442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6033072887347442
+    },
+    {
+      "train_step": 8573,
+      "loss": 0.5910590755917398,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5910590755917398
+    },
+    {
+      "train_step": 8574,
+      "loss": 0.47320750688560936,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47320750688560936
+    },
+    {
+      "train_step": 8575,
+      "loss": 0.36585463591350953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36585463591350953
+    },
+    {
+      "train_step": 8576,
+      "loss": 0.6412885536354509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6412885536354509
+    },
+    {
+      "train_step": 8577,
+      "loss": 0.5548731245638282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5548731245638282
+    },
+    {
+      "train_step": 8578,
+      "loss": 0.6548299158971915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6548299158971915
+    },
+    {
+      "train_step": 8579,
+      "loss": 0.5745028411773434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5745028411773434
+    },
+    {
+      "train_step": 8580,
+      "loss": 0.6179222818818323,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6179222818818323
+    },
+    {
+      "train_step": 8581,
+      "loss": 0.44860960837000197,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44860960837000197
+    },
+    {
+      "train_step": 8582,
+      "loss": 0.5568597093446022,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5568597093446022
+    },
+    {
+      "train_step": 8583,
+      "loss": 0.6419444013204895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6419444013204895
+    },
+    {
+      "train_step": 8584,
+      "loss": 0.5290721068388167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5290721068388167
+    },
+    {
+      "train_step": 8585,
+      "loss": 0.675395669539935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.675395669539935
+    },
+    {
+      "train_step": 8586,
+      "loss": 0.32734038135876115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32734038135876115
+    },
+    {
+      "train_step": 8587,
+      "loss": 0.66277998328785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.66277998328785
+    },
+    {
+      "train_step": 8588,
+      "loss": 0.46293115858408246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46293115858408246
+    },
+    {
+      "train_step": 8589,
+      "loss": 0.8324231175038534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8324231175038534
+    },
+    {
+      "train_step": 8590,
+      "loss": 0.43119749102390986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43119749102390986
+    },
+    {
+      "train_step": 8591,
+      "loss": 0.4659054142835176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4659054142835176
+    },
+    {
+      "train_step": 8592,
+      "loss": 0.5321417209859384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5321417209859384
+    },
+    {
+      "train_step": 8593,
+      "loss": 0.5155123745332747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5155123745332747
+    },
+    {
+      "train_step": 8594,
+      "loss": 0.5475835267225821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5475835267225821
+    },
+    {
+      "train_step": 8595,
+      "loss": 0.4351314911941309,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4351314911941309
+    },
+    {
+      "train_step": 8596,
+      "loss": 0.5169452460343794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5169452460343794
+    },
+    {
+      "train_step": 8597,
+      "loss": 0.47168858291482507,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47168858291482507
+    },
+    {
+      "train_step": 8598,
+      "loss": 0.5649304470746372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5649304470746372
+    },
+    {
+      "train_step": 8599,
+      "loss": 0.5934908427886049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5934908427886049
+    },
+    {
+      "train_step": 8600,
+      "loss": 0.45661336907610023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45661336907610023
+    },
+    {
+      "train_step": 8601,
+      "loss": 0.5621730814865911,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5621730814865911
+    },
+    {
+      "train_step": 8602,
+      "loss": 0.4371735547297925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4371735547297925
+    },
+    {
+      "train_step": 8603,
+      "loss": 0.6364356512409065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6364356512409065
+    },
+    {
+      "train_step": 8604,
+      "loss": 0.44929308963942804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44929308963942804
+    },
+    {
+      "train_step": 8605,
+      "loss": 0.5425118570473244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5425118570473244
+    },
+    {
+      "train_step": 8606,
+      "loss": 0.7940688881871134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7940688881871134
+    },
+    {
+      "train_step": 8607,
+      "loss": 0.47906075894564953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47906075894564953
+    },
+    {
+      "train_step": 8608,
+      "loss": 0.782380823197848,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.782380823197848
+    },
+    {
+      "train_step": 8609,
+      "loss": 0.3994042359031748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3994042359031748
+    },
+    {
+      "train_step": 8610,
+      "loss": 0.7601357056172104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7601357056172104
+    },
+    {
+      "train_step": 8611,
+      "loss": 0.7995850445995003,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7995850445995003
+    },
+    {
+      "train_step": 8612,
+      "loss": 0.4329421539831947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4329421539831947
+    },
+    {
+      "train_step": 8613,
+      "loss": 0.4963542825488893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4963542825488893
+    },
+    {
+      "train_step": 8614,
+      "loss": 0.5995474155098196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5995474155098196
+    },
+    {
+      "train_step": 8615,
+      "loss": 0.549213990096395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.549213990096395
+    },
+    {
+      "train_step": 8616,
+      "loss": 0.5386712148769417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5386712148769417
+    },
+    {
+      "train_step": 8617,
+      "loss": 0.5538744792587382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5538744792587382
+    },
+    {
+      "train_step": 8618,
+      "loss": 0.5719069210144068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5719069210144068
+    },
+    {
+      "train_step": 8619,
+      "loss": 0.6663843928394872,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6663843928394872
+    },
+    {
+      "train_step": 8620,
+      "loss": 0.5202130320332419,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5202130320332419
+    },
+    {
+      "train_step": 8621,
+      "loss": 0.4447406013821444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4447406013821444
+    },
+    {
+      "train_step": 8622,
+      "loss": 0.5471386864926034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5471386864926034
+    },
+    {
+      "train_step": 8623,
+      "loss": 0.6151677011806004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6151677011806004
+    },
+    {
+      "train_step": 8624,
+      "loss": 0.5594887912064808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5594887912064808
+    },
+    {
+      "train_step": 8625,
+      "loss": 0.6750512741604795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6750512741604795
+    },
+    {
+      "train_step": 8626,
+      "loss": 0.6764480561597853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6764480561597853
+    },
+    {
+      "train_step": 8627,
+      "loss": 0.4635059219313587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4635059219313587
+    },
+    {
+      "train_step": 8628,
+      "loss": 0.5089352528491566,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5089352528491566
+    },
+    {
+      "train_step": 8629,
+      "loss": 0.5973675172122366,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5973675172122366
+    },
+    {
+      "train_step": 8630,
+      "loss": 0.473642472591451,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.473642472591451
+    },
+    {
+      "train_step": 8631,
+      "loss": 0.698469894506007,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.698469894506007
+    },
+    {
+      "train_step": 8632,
+      "loss": 0.6701083285856134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6701083285856134
+    },
+    {
+      "train_step": 8633,
+      "loss": 0.6460773056703691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6460773056703691
+    },
+    {
+      "train_step": 8634,
+      "loss": 0.5558928721087014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5558928721087014
+    },
+    {
+      "train_step": 8635,
+      "loss": 0.5100408252834454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5100408252834454
+    },
+    {
+      "train_step": 8636,
+      "loss": 0.5213965378732393,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5213965378732393
+    },
+    {
+      "train_step": 8637,
+      "loss": 0.5903683246984812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5903683246984812
+    },
+    {
+      "train_step": 8638,
+      "loss": 0.6942710246502808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6942710246502808
+    },
+    {
+      "train_step": 8639,
+      "loss": 0.7515278063836862,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7515278063836862
+    },
+    {
+      "train_step": 8640,
+      "loss": 0.3674949647735694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3674949647735694
+    },
+    {
+      "train_step": 8641,
+      "loss": 0.5271090015023978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5271090015023978
+    },
+    {
+      "train_step": 8642,
+      "loss": 0.4425226783071786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4425226783071786
+    },
+    {
+      "train_step": 8643,
+      "loss": 0.5815390682341086,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5815390682341086
+    },
+    {
+      "train_step": 8644,
+      "loss": 0.528455653385093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.528455653385093
+    },
+    {
+      "train_step": 8645,
+      "loss": 0.625160735740492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.625160735740492
+    },
+    {
+      "train_step": 8646,
+      "loss": 0.5384282760810913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5384282760810913
+    },
+    {
+      "train_step": 8647,
+      "loss": 0.6196864609058352,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6196864609058352
+    },
+    {
+      "train_step": 8648,
+      "loss": 0.5364393813649867,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5364393813649867
+    },
+    {
+      "train_step": 8649,
+      "loss": 0.7064025170246799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7064025170246799
+    },
+    {
+      "train_step": 8650,
+      "loss": 0.6866154656041881,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6866154656041881
+    },
+    {
+      "train_step": 8651,
+      "loss": 0.5479971029602109,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5479971029602109
+    },
+    {
+      "train_step": 8652,
+      "loss": 0.3918963854809125,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3918963854809125
+    },
+    {
+      "train_step": 8653,
+      "loss": 0.7557372057024171,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7557372057024171
+    },
+    {
+      "train_step": 8654,
+      "loss": 0.6256635040780013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6256635040780013
+    },
+    {
+      "train_step": 8655,
+      "loss": 0.5297518125693549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5297518125693549
+    },
+    {
+      "train_step": 8656,
+      "loss": 0.5471033082626436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5471033082626436
+    },
+    {
+      "train_step": 8657,
+      "loss": 0.5513780652594802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5513780652594802
+    },
+    {
+      "train_step": 8658,
+      "loss": 0.4797789655102038,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4797789655102038
+    },
+    {
+      "train_step": 8659,
+      "loss": 0.6094529261345729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6094529261345729
+    },
+    {
+      "train_step": 8660,
+      "loss": 0.4676434205467587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4676434205467587
+    },
+    {
+      "train_step": 8661,
+      "loss": 0.6360139088531757,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6360139088531757
+    },
+    {
+      "train_step": 8662,
+      "loss": 0.6323748711701965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6323748711701965
+    },
+    {
+      "train_step": 8663,
+      "loss": 0.6388830508497454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6388830508497454
+    },
+    {
+      "train_step": 8664,
+      "loss": 0.5001722431035679,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5001722431035679
+    },
+    {
+      "train_step": 8665,
+      "loss": 0.42727448664827916,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42727448664827916
+    },
+    {
+      "train_step": 8666,
+      "loss": 0.49860113420901153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49860113420901153
+    },
+    {
+      "train_step": 8667,
+      "loss": 0.7063820425062132,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7063820425062132
+    },
+    {
+      "train_step": 8668,
+      "loss": 0.5478422140103324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5478422140103324
+    },
+    {
+      "train_step": 8669,
+      "loss": 0.5877175909755068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5877175909755068
+    },
+    {
+      "train_step": 8670,
+      "loss": 0.6343031785964137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6343031785964137
+    },
+    {
+      "train_step": 8671,
+      "loss": 0.60438910306575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.60438910306575
+    },
+    {
+      "train_step": 8672,
+      "loss": 0.5713011201454629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5713011201454629
+    },
+    {
+      "train_step": 8673,
+      "loss": 0.4122432765882155,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4122432765882155
+    },
+    {
+      "train_step": 8674,
+      "loss": 0.4890059197681762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4890059197681762
+    },
+    {
+      "train_step": 8675,
+      "loss": 0.6471354593988291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6471354593988291
+    },
+    {
+      "train_step": 8676,
+      "loss": 0.8093556538514497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8093556538514497
+    },
+    {
+      "train_step": 8677,
+      "loss": 0.6913897394513906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6913897394513906
+    },
+    {
+      "train_step": 8678,
+      "loss": 0.7525976718054448,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7525976718054448
+    },
+    {
+      "train_step": 8679,
+      "loss": 0.6446862777571657,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6446862777571657
+    },
+    {
+      "train_step": 8680,
+      "loss": 0.5632942130340793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5632942130340793
+    },
+    {
+      "train_step": 8681,
+      "loss": 0.48670526485205245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48670526485205245
+    },
+    {
+      "train_step": 8682,
+      "loss": 0.4966453995547613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4966453995547613
+    },
+    {
+      "train_step": 8683,
+      "loss": 0.6153066054829307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6153066054829307
+    },
+    {
+      "train_step": 8684,
+      "loss": 0.7623695363093439,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7623695363093439
+    },
+    {
+      "train_step": 8685,
+      "loss": 0.6864760322660063,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6864760322660063
+    },
+    {
+      "train_step": 8686,
+      "loss": 0.4295750113682821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4295750113682821
+    },
+    {
+      "train_step": 8687,
+      "loss": 0.6142105196592563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6142105196592563
+    },
+    {
+      "train_step": 8688,
+      "loss": 0.8865643756907892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8865643756907892
+    },
+    {
+      "train_step": 8689,
+      "loss": 0.4536726425785971,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4536726425785971
+    },
+    {
+      "train_step": 8690,
+      "loss": 0.7418904213189859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7418904213189859
+    },
+    {
+      "train_step": 8691,
+      "loss": 0.6611100825524975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6611100825524975
+    },
+    {
+      "train_step": 8692,
+      "loss": 0.5157997046688011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5157997046688011
+    },
+    {
+      "train_step": 8693,
+      "loss": 0.639870837459485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.639870837459485
+    },
+    {
+      "train_step": 8694,
+      "loss": 0.50859241421338,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.50859241421338
+    },
+    {
+      "train_step": 8695,
+      "loss": 0.615741668305466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.615741668305466
+    },
+    {
+      "train_step": 8696,
+      "loss": 0.42254737255592256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42254737255592256
+    },
+    {
+      "train_step": 8697,
+      "loss": 0.5152602364780002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5152602364780002
+    },
+    {
+      "train_step": 8698,
+      "loss": 0.933747758446844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.933747758446844
+    },
+    {
+      "train_step": 8699,
+      "loss": 0.7338683298425954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7338683298425954
+    },
+    {
+      "train_step": 8700,
+      "loss": 0.5268152319951487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5268152319951487
+    },
+    {
+      "train_step": 8701,
+      "loss": 0.4975556929165763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4975556929165763
+    },
+    {
+      "train_step": 8702,
+      "loss": 0.5348402177747305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5348402177747305
+    },
+    {
+      "train_step": 8703,
+      "loss": 0.6367363977354277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6367363977354277
+    },
+    {
+      "train_step": 8704,
+      "loss": 0.7343593921353754,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7343593921353754
+    },
+    {
+      "train_step": 8705,
+      "loss": 0.5629480714596209,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5629480714596209
+    },
+    {
+      "train_step": 8706,
+      "loss": 0.46040512675544837,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46040512675544837
+    },
+    {
+      "train_step": 8707,
+      "loss": 0.6474725522332097,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6474725522332097
+    },
+    {
+      "train_step": 8708,
+      "loss": 0.4671310832624677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4671310832624677
+    },
+    {
+      "train_step": 8709,
+      "loss": 0.5560148153275853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5560148153275853
+    },
+    {
+      "train_step": 8710,
+      "loss": 0.5610783533177316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5610783533177316
+    },
+    {
+      "train_step": 8711,
+      "loss": 0.6451463905990708,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6451463905990708
+    },
+    {
+      "train_step": 8712,
+      "loss": 0.5030770167012085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5030770167012085
+    },
+    {
+      "train_step": 8713,
+      "loss": 0.5517725590703785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5517725590703785
+    },
+    {
+      "train_step": 8714,
+      "loss": 0.7456375182212541,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7456375182212541
+    },
+    {
+      "train_step": 8715,
+      "loss": 0.39882324041259465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39882324041259465
+    },
+    {
+      "train_step": 8716,
+      "loss": 0.5581766526692798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5581766526692798
+    },
+    {
+      "train_step": 8717,
+      "loss": 0.4666380023087168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4666380023087168
+    },
+    {
+      "train_step": 8718,
+      "loss": 0.3235801816532017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3235801816532017
+    },
+    {
+      "train_step": 8719,
+      "loss": 0.475763741735351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.475763741735351
+    },
+    {
+      "train_step": 8720,
+      "loss": 0.559617058500476,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.559617058500476
+    },
+    {
+      "train_step": 8721,
+      "loss": 0.5682728737809202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5682728737809202
+    },
+    {
+      "train_step": 8722,
+      "loss": 0.5962272003140412,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962272003140412
+    },
+    {
+      "train_step": 8723,
+      "loss": 0.5923096558832014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5923096558832014
+    },
+    {
+      "train_step": 8724,
+      "loss": 0.590803566152576,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.590803566152576
+    },
+    {
+      "train_step": 8725,
+      "loss": 0.4519525103282103,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4519525103282103
+    },
+    {
+      "train_step": 8726,
+      "loss": 0.526640248082852,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.526640248082852
+    },
+    {
+      "train_step": 8727,
+      "loss": 0.8065032011758065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8065032011758065
+    },
+    {
+      "train_step": 8728,
+      "loss": 0.43318454251886673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43318454251886673
+    },
+    {
+      "train_step": 8729,
+      "loss": 0.5252268407461027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5252268407461027
+    },
+    {
+      "train_step": 8730,
+      "loss": 0.6023496458794305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6023496458794305
+    },
+    {
+      "train_step": 8731,
+      "loss": 0.5706707991257625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706707991257625
+    },
+    {
+      "train_step": 8732,
+      "loss": 0.7129205226424693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7129205226424693
+    },
+    {
+      "train_step": 8733,
+      "loss": 0.459962895947406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.459962895947406
+    },
+    {
+      "train_step": 8734,
+      "loss": 0.5503910166402051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5503910166402051
+    },
+    {
+      "train_step": 8735,
+      "loss": 0.4479926100564213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4479926100564213
+    },
+    {
+      "train_step": 8736,
+      "loss": 0.6859584918379638,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6859584918379638
+    },
+    {
+      "train_step": 8737,
+      "loss": 0.48751877014850065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48751877014850065
+    },
+    {
+      "train_step": 8738,
+      "loss": 0.5068080525549785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5068080525549785
+    },
+    {
+      "train_step": 8739,
+      "loss": 0.49686527567802014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49686527567802014
+    },
+    {
+      "train_step": 8740,
+      "loss": 0.4757002032842331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4757002032842331
+    },
+    {
+      "train_step": 8741,
+      "loss": 0.6847984883464798,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6847984883464798
+    },
+    {
+      "train_step": 8742,
+      "loss": 0.5314353970376635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5314353970376635
+    },
+    {
+      "train_step": 8743,
+      "loss": 0.5081620039569463,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5081620039569463
+    },
+    {
+      "train_step": 8744,
+      "loss": 0.5881266497320947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5881266497320947
+    },
+    {
+      "train_step": 8745,
+      "loss": 0.5791177947555172,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5791177947555172
+    },
+    {
+      "train_step": 8746,
+      "loss": 0.5347899549427085,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5347899549427085
+    },
+    {
+      "train_step": 8747,
+      "loss": 0.4818035096189602,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4818035096189602
+    },
+    {
+      "train_step": 8748,
+      "loss": 0.6336882527054213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6336882527054213
+    },
+    {
+      "train_step": 8749,
+      "loss": 0.6405814356283478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6405814356283478
+    },
+    {
+      "train_step": 8750,
+      "loss": 0.35445805684400666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35445805684400666
+    },
+    {
+      "train_step": 8751,
+      "loss": 0.398807089269335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.398807089269335
+    },
+    {
+      "train_step": 8752,
+      "loss": 0.529319300963747,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.529319300963747
+    },
+    {
+      "train_step": 8753,
+      "loss": 0.3819106201051595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3819106201051595
+    },
+    {
+      "train_step": 8754,
+      "loss": 0.6937512698860493,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6937512698860493
+    },
+    {
+      "train_step": 8755,
+      "loss": 0.36957128499460024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36957128499460024
+    },
+    {
+      "train_step": 8756,
+      "loss": 0.5835347811167167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5835347811167167
+    },
+    {
+      "train_step": 8757,
+      "loss": 0.6068171658708983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6068171658708983
+    },
+    {
+      "train_step": 8758,
+      "loss": 0.43239546109712906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43239546109712906
+    },
+    {
+      "train_step": 8759,
+      "loss": 0.614687681427325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.614687681427325
+    },
+    {
+      "train_step": 8760,
+      "loss": 0.4343169633123127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4343169633123127
+    },
+    {
+      "train_step": 8761,
+      "loss": 0.3452154735749679,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3452154735749679
+    },
+    {
+      "train_step": 8762,
+      "loss": 0.41465983352194213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41465983352194213
+    },
+    {
+      "train_step": 8763,
+      "loss": 0.6308612318539534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6308612318539534
+    },
+    {
+      "train_step": 8764,
+      "loss": 0.5506071270859715,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5506071270859715
+    },
+    {
+      "train_step": 8765,
+      "loss": 0.4551978181362202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4551978181362202
+    },
+    {
+      "train_step": 8766,
+      "loss": 0.5007541117669422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5007541117669422
+    },
+    {
+      "train_step": 8767,
+      "loss": 0.5761554523297351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5761554523297351
+    },
+    {
+      "train_step": 8768,
+      "loss": 0.6382361095851194,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6382361095851194
+    },
+    {
+      "train_step": 8769,
+      "loss": 0.5226981535329595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5226981535329595
+    },
+    {
+      "train_step": 8770,
+      "loss": 0.5798782504918989,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5798782504918989
+    },
+    {
+      "train_step": 8771,
+      "loss": 0.666974745980407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.666974745980407
+    },
+    {
+      "train_step": 8772,
+      "loss": 0.48309051456630814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48309051456630814
+    },
+    {
+      "train_step": 8773,
+      "loss": 0.35477438717936755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35477438717936755
+    },
+    {
+      "train_step": 8774,
+      "loss": 0.4316885378300257,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4316885378300257
+    },
+    {
+      "train_step": 8775,
+      "loss": 0.6174157157792781,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6174157157792781
+    },
+    {
+      "train_step": 8776,
+      "loss": 0.494482789129666,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.494482789129666
+    },
+    {
+      "train_step": 8777,
+      "loss": 0.6609972479735263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6609972479735263
+    },
+    {
+      "train_step": 8778,
+      "loss": 0.6810702735932772,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6810702735932772
+    },
+    {
+      "train_step": 8779,
+      "loss": 0.6021832689057692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6021832689057692
+    },
+    {
+      "train_step": 8780,
+      "loss": 0.7213376154726898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7213376154726898
+    },
+    {
+      "train_step": 8781,
+      "loss": 0.5034637194435467,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5034637194435467
+    },
+    {
+      "train_step": 8782,
+      "loss": 0.5575051413053317,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5575051413053317
+    },
+    {
+      "train_step": 8783,
+      "loss": 0.5995075079657546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5995075079657546
+    },
+    {
+      "train_step": 8784,
+      "loss": 0.7039387059615071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7039387059615071
+    },
+    {
+      "train_step": 8785,
+      "loss": 0.5350198371230073,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5350198371230073
+    },
+    {
+      "train_step": 8786,
+      "loss": 0.544519826379819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.544519826379819
+    },
+    {
+      "train_step": 8787,
+      "loss": 0.5706904588607626,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706904588607626
+    },
+    {
+      "train_step": 8788,
+      "loss": 0.5970761095516722,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5970761095516722
+    },
+    {
+      "train_step": 8789,
+      "loss": 0.7050089750323361,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7050089750323361
+    },
+    {
+      "train_step": 8790,
+      "loss": 0.4749900604622735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4749900604622735
+    },
+    {
+      "train_step": 8791,
+      "loss": 0.5213457025927545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5213457025927545
+    },
+    {
+      "train_step": 8792,
+      "loss": 0.6163671020218726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6163671020218726
+    },
+    {
+      "train_step": 8793,
+      "loss": 0.6673663878545885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6673663878545885
+    },
+    {
+      "train_step": 8794,
+      "loss": 0.6617365456378956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6617365456378956
+    },
+    {
+      "train_step": 8795,
+      "loss": 0.5033969957070525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5033969957070525
+    },
+    {
+      "train_step": 8796,
+      "loss": 0.6418811031412058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6418811031412058
+    },
+    {
+      "train_step": 8797,
+      "loss": 0.7311061122049773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7311061122049773
+    },
+    {
+      "train_step": 8798,
+      "loss": 0.5594793316335216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5594793316335216
+    },
+    {
+      "train_step": 8799,
+      "loss": 0.5604218706381855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5604218706381855
+    },
+    {
+      "train_step": 8800,
+      "loss": 0.6810690900302999,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6810690900302999
+    },
+    {
+      "train_step": 8801,
+      "loss": 0.455014615966331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.455014615966331
+    },
+    {
+      "train_step": 8802,
+      "loss": 0.5912412346829886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5912412346829886
+    },
+    {
+      "train_step": 8803,
+      "loss": 0.6649341687236358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6649341687236358
+    },
+    {
+      "train_step": 8804,
+      "loss": 0.6154079844012447,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6154079844012447
+    },
+    {
+      "train_step": 8805,
+      "loss": 0.37939044609429107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37939044609429107
+    },
+    {
+      "train_step": 8806,
+      "loss": 0.5781171454090003,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5781171454090003
+    },
+    {
+      "train_step": 8807,
+      "loss": 0.4584514071550425,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4584514071550425
+    },
+    {
+      "train_step": 8808,
+      "loss": 0.4934345979395515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4934345979395515
+    },
+    {
+      "train_step": 8809,
+      "loss": 0.38964462433462854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38964462433462854
+    },
+    {
+      "train_step": 8810,
+      "loss": 0.4016477177696509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4016477177696509
+    },
+    {
+      "train_step": 8811,
+      "loss": 0.3212385467271709,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3212385467271709
+    },
+    {
+      "train_step": 8812,
+      "loss": 0.42497183594194565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42497183594194565
+    },
+    {
+      "train_step": 8813,
+      "loss": 0.29856707805290794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.29856707805290794
+    },
+    {
+      "train_step": 8814,
+      "loss": 0.6316879755987801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6316879755987801
+    },
+    {
+      "train_step": 8815,
+      "loss": 0.5705719368922048,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5705719368922048
+    },
+    {
+      "train_step": 8816,
+      "loss": 0.49375974640869164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49375974640869164
+    },
+    {
+      "train_step": 8817,
+      "loss": 0.635406921292093,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.635406921292093
+    },
+    {
+      "train_step": 8818,
+      "loss": 0.4850779870258035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4850779870258035
+    },
+    {
+      "train_step": 8819,
+      "loss": 0.40376192265094607,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40376192265094607
+    },
+    {
+      "train_step": 8820,
+      "loss": 0.4811552695400071,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4811552695400071
+    },
+    {
+      "train_step": 8821,
+      "loss": 0.5475715034713017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5475715034713017
+    },
+    {
+      "train_step": 8822,
+      "loss": 0.5287857237837816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5287857237837816
+    },
+    {
+      "train_step": 8823,
+      "loss": 0.5630494630979681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5630494630979681
+    },
+    {
+      "train_step": 8824,
+      "loss": 0.5976639282096172,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5976639282096172
+    },
+    {
+      "train_step": 8825,
+      "loss": 0.7464052924237827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7464052924237827
+    },
+    {
+      "train_step": 8826,
+      "loss": 0.5403787667677649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5403787667677649
+    },
+    {
+      "train_step": 8827,
+      "loss": 0.5844167304111322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5844167304111322
+    },
+    {
+      "train_step": 8828,
+      "loss": 0.6537099297242586,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6537099297242586
+    },
+    {
+      "train_step": 8829,
+      "loss": 0.5566898819679286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5566898819679286
+    },
+    {
+      "train_step": 8830,
+      "loss": 0.5181969444713154,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5181969444713154
+    },
+    {
+      "train_step": 8831,
+      "loss": 0.5272108562495665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5272108562495665
+    },
+    {
+      "train_step": 8832,
+      "loss": 0.42818232292540354,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42818232292540354
+    },
+    {
+      "train_step": 8833,
+      "loss": 0.4061447788382663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4061447788382663
+    },
+    {
+      "train_step": 8834,
+      "loss": 0.4829492039030688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4829492039030688
+    },
+    {
+      "train_step": 8835,
+      "loss": 0.6283363640756492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6283363640756492
+    },
+    {
+      "train_step": 8836,
+      "loss": 0.5496503507972014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5496503507972014
+    },
+    {
+      "train_step": 8837,
+      "loss": 0.7277251746816672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7277251746816672
+    },
+    {
+      "train_step": 8838,
+      "loss": 0.5935701878228161,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5935701878228161
+    },
+    {
+      "train_step": 8839,
+      "loss": 0.4798251966727969,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4798251966727969
+    },
+    {
+      "train_step": 8840,
+      "loss": 0.4589762443082498,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4589762443082498
+    },
+    {
+      "train_step": 8841,
+      "loss": 0.6467189036353069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6467189036353069
+    },
+    {
+      "train_step": 8842,
+      "loss": 0.41821882517011055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41821882517011055
+    },
+    {
+      "train_step": 8843,
+      "loss": 0.5322137158700481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5322137158700481
+    },
+    {
+      "train_step": 8844,
+      "loss": 0.628158217992662,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.628158217992662
+    },
+    {
+      "train_step": 8845,
+      "loss": 0.5899145909562898,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5899145909562898
+    },
+    {
+      "train_step": 8846,
+      "loss": 0.6936590991007129,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6936590991007129
+    },
+    {
+      "train_step": 8847,
+      "loss": 0.48213818128648545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48213818128648545
+    },
+    {
+      "train_step": 8848,
+      "loss": 0.3867129124076773,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3867129124076773
+    },
+    {
+      "train_step": 8849,
+      "loss": 0.5385476858672716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5385476858672716
+    },
+    {
+      "train_step": 8850,
+      "loss": 0.595876868347492,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.595876868347492
+    },
+    {
+      "train_step": 8851,
+      "loss": 0.5486642850669913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5486642850669913
+    },
+    {
+      "train_step": 8852,
+      "loss": 0.5706366688056138,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706366688056138
+    },
+    {
+      "train_step": 8853,
+      "loss": 0.5719683377458729,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5719683377458729
+    },
+    {
+      "train_step": 8854,
+      "loss": 0.6860602158601319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6860602158601319
+    },
+    {
+      "train_step": 8855,
+      "loss": 0.5839258322547236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5839258322547236
+    },
+    {
+      "train_step": 8856,
+      "loss": 0.29707225002669624,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.29707225002669624
+    },
+    {
+      "train_step": 8857,
+      "loss": 0.6114771085066953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6114771085066953
+    },
+    {
+      "train_step": 8858,
+      "loss": 0.5218990179856564,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5218990179856564
+    },
+    {
+      "train_step": 8859,
+      "loss": 0.4424587043868331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4424587043868331
+    },
+    {
+      "train_step": 8860,
+      "loss": 0.5051040377408077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5051040377408077
+    },
+    {
+      "train_step": 8861,
+      "loss": 0.5806735358072808,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5806735358072808
+    },
+    {
+      "train_step": 8862,
+      "loss": 0.6332947492831106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6332947492831106
+    },
+    {
+      "train_step": 8863,
+      "loss": 0.6124264018694603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6124264018694603
+    },
+    {
+      "train_step": 8864,
+      "loss": 0.5212345445960596,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5212345445960596
+    },
+    {
+      "train_step": 8865,
+      "loss": 0.4945991194759506,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4945991194759506
+    },
+    {
+      "train_step": 8866,
+      "loss": 0.5636616207742723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5636616207742723
+    },
+    {
+      "train_step": 8867,
+      "loss": 0.7881871888718277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7881871888718277
+    },
+    {
+      "train_step": 8868,
+      "loss": 0.47732932101397413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47732932101397413
+    },
+    {
+      "train_step": 8869,
+      "loss": 0.7184782536615132,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7184782536615132
+    },
+    {
+      "train_step": 8870,
+      "loss": 0.566096100140897,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.566096100140897
+    },
+    {
+      "train_step": 8871,
+      "loss": 0.4524949700709873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4524949700709873
+    },
+    {
+      "train_step": 8872,
+      "loss": 0.7074229031508704,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7074229031508704
+    },
+    {
+      "train_step": 8873,
+      "loss": 0.5428884673215949,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5428884673215949
+    },
+    {
+      "train_step": 8874,
+      "loss": 0.5170108463331818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5170108463331818
+    },
+    {
+      "train_step": 8875,
+      "loss": 0.5297340555064489,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5297340555064489
+    },
+    {
+      "train_step": 8876,
+      "loss": 0.533797090743548,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.533797090743548
+    },
+    {
+      "train_step": 8877,
+      "loss": 0.49329878890278833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49329878890278833
+    },
+    {
+      "train_step": 8878,
+      "loss": 0.6768687657639874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6768687657639874
+    },
+    {
+      "train_step": 8879,
+      "loss": 0.6842182933777055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6842182933777055
+    },
+    {
+      "train_step": 8880,
+      "loss": 0.5938601240150336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5938601240150336
+    },
+    {
+      "train_step": 8881,
+      "loss": 0.49693440906926495,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49693440906926495
+    },
+    {
+      "train_step": 8882,
+      "loss": 0.3397989376674539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3397989376674539
+    },
+    {
+      "train_step": 8883,
+      "loss": 0.24176034755787068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.24176034755787068
+    },
+    {
+      "train_step": 8884,
+      "loss": 0.5846325740541753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5846325740541753
+    },
+    {
+      "train_step": 8885,
+      "loss": 0.43042611099611355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43042611099611355
+    },
+    {
+      "train_step": 8886,
+      "loss": 0.5642865305776166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5642865305776166
+    },
+    {
+      "train_step": 8887,
+      "loss": 0.5149685086836188,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5149685086836188
+    },
+    {
+      "train_step": 8888,
+      "loss": 0.7018478896274927,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7018478896274927
+    },
+    {
+      "train_step": 8889,
+      "loss": 0.5007163104926905,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5007163104926905
+    },
+    {
+      "train_step": 8890,
+      "loss": 0.40190963008219416,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40190963008219416
+    },
+    {
+      "train_step": 8891,
+      "loss": 0.6825003417829574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6825003417829574
+    },
+    {
+      "train_step": 8892,
+      "loss": 0.3593974535495694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3593974535495694
+    },
+    {
+      "train_step": 8893,
+      "loss": 0.5212017700797651,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5212017700797651
+    },
+    {
+      "train_step": 8894,
+      "loss": 0.6572558828476481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6572558828476481
+    },
+    {
+      "train_step": 8895,
+      "loss": 0.6451161625869345,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6451161625869345
+    },
+    {
+      "train_step": 8896,
+      "loss": 0.4779434499656263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4779434499656263
+    },
+    {
+      "train_step": 8897,
+      "loss": 0.593449552933849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.593449552933849
+    },
+    {
+      "train_step": 8898,
+      "loss": 0.5633919037777178,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5633919037777178
+    },
+    {
+      "train_step": 8899,
+      "loss": 0.4180038370851696,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4180038370851696
+    },
+    {
+      "train_step": 8900,
+      "loss": 0.5221820974515318,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5221820974515318
+    },
+    {
+      "train_step": 8901,
+      "loss": 0.5138433269600902,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5138433269600902
+    },
+    {
+      "train_step": 8902,
+      "loss": 0.7530035613988573,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7530035613988573
+    },
+    {
+      "train_step": 8903,
+      "loss": 0.49998250803582417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49998250803582417
+    },
+    {
+      "train_step": 8904,
+      "loss": 0.4983801660376167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4983801660376167
+    },
+    {
+      "train_step": 8905,
+      "loss": 0.5323379902870977,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5323379902870977
+    },
+    {
+      "train_step": 8906,
+      "loss": 0.4328279174774218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4328279174774218
+    },
+    {
+      "train_step": 8907,
+      "loss": 0.5538503246468776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5538503246468776
+    },
+    {
+      "train_step": 8908,
+      "loss": 0.6039069706944462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6039069706944462
+    },
+    {
+      "train_step": 8909,
+      "loss": 0.4765371233643864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4765371233643864
+    },
+    {
+      "train_step": 8910,
+      "loss": 0.652813596853555,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.652813596853555
+    },
+    {
+      "train_step": 8911,
+      "loss": 0.5328685978159454,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5328685978159454
+    },
+    {
+      "train_step": 8912,
+      "loss": 0.4678038285863937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4678038285863937
+    },
+    {
+      "train_step": 8913,
+      "loss": 0.5313331141268459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5313331141268459
+    },
+    {
+      "train_step": 8914,
+      "loss": 0.673513934793208,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.673513934793208
+    },
+    {
+      "train_step": 8915,
+      "loss": 0.6139798547293166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6139798547293166
+    },
+    {
+      "train_step": 8916,
+      "loss": 0.6513296936431691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6513296936431691
+    },
+    {
+      "train_step": 8917,
+      "loss": 0.5980949224380795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5980949224380795
+    },
+    {
+      "train_step": 8918,
+      "loss": 0.446676895364065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.446676895364065
+    },
+    {
+      "train_step": 8919,
+      "loss": 0.6286364445945702,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6286364445945702
+    },
+    {
+      "train_step": 8920,
+      "loss": 0.580517837660005,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.580517837660005
+    },
+    {
+      "train_step": 8921,
+      "loss": 0.5829484982717655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5829484982717655
+    },
+    {
+      "train_step": 8922,
+      "loss": 0.6242585651609527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6242585651609527
+    },
+    {
+      "train_step": 8923,
+      "loss": 0.444546970650052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.444546970650052
+    },
+    {
+      "train_step": 8924,
+      "loss": 0.5521260666891176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5521260666891176
+    },
+    {
+      "train_step": 8925,
+      "loss": 0.2774552802940538,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2774552802940538
+    },
+    {
+      "train_step": 8926,
+      "loss": 0.5172617174993621,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5172617174993621
+    },
+    {
+      "train_step": 8927,
+      "loss": 0.48303847656509996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48303847656509996
+    },
+    {
+      "train_step": 8928,
+      "loss": 0.5850096752945662,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5850096752945662
+    },
+    {
+      "train_step": 8929,
+      "loss": 0.6247546404057531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6247546404057531
+    },
+    {
+      "train_step": 8930,
+      "loss": 0.34322369661237057,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34322369661237057
+    },
+    {
+      "train_step": 8931,
+      "loss": 0.6853271464708162,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6853271464708162
+    },
+    {
+      "train_step": 8932,
+      "loss": 0.5928566392517158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5928566392517158
+    },
+    {
+      "train_step": 8933,
+      "loss": 0.640825589648587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.640825589648587
+    },
+    {
+      "train_step": 8934,
+      "loss": 0.604938663130226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.604938663130226
+    },
+    {
+      "train_step": 8935,
+      "loss": 0.5631589445591104,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5631589445591104
+    },
+    {
+      "train_step": 8936,
+      "loss": 0.5043261019972791,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5043261019972791
+    },
+    {
+      "train_step": 8937,
+      "loss": 0.565333811450876,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.565333811450876
+    },
+    {
+      "train_step": 8938,
+      "loss": 0.5297708216998908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5297708216998908
+    },
+    {
+      "train_step": 8939,
+      "loss": 0.5801661875626709,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5801661875626709
+    },
+    {
+      "train_step": 8940,
+      "loss": 0.5491485631414047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5491485631414047
+    },
+    {
+      "train_step": 8941,
+      "loss": 0.45178288363474184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45178288363474184
+    },
+    {
+      "train_step": 8942,
+      "loss": 0.7243733813760572,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7243733813760572
+    },
+    {
+      "train_step": 8943,
+      "loss": 0.6019100657326597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6019100657326597
+    },
+    {
+      "train_step": 8944,
+      "loss": 0.5658153322149373,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5658153322149373
+    },
+    {
+      "train_step": 8945,
+      "loss": 0.7162015917341322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7162015917341322
+    },
+    {
+      "train_step": 8946,
+      "loss": 0.5749254741705513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5749254741705513
+    },
+    {
+      "train_step": 8947,
+      "loss": 0.5935295298626548,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5935295298626548
+    },
+    {
+      "train_step": 8948,
+      "loss": 0.3587837646981076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3587837646981076
+    },
+    {
+      "train_step": 8949,
+      "loss": 0.5199636984734242,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5199636984734242
+    },
+    {
+      "train_step": 8950,
+      "loss": 0.5527727570427158,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5527727570427158
+    },
+    {
+      "train_step": 8951,
+      "loss": 0.49020510431658504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49020510431658504
+    },
+    {
+      "train_step": 8952,
+      "loss": 0.32116067168531165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.32116067168531165
+    },
+    {
+      "train_step": 8953,
+      "loss": 0.6705773559774082,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6705773559774082
+    },
+    {
+      "train_step": 8954,
+      "loss": 0.522724904337846,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.522724904337846
+    },
+    {
+      "train_step": 8955,
+      "loss": 0.46841741099215845,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46841741099215845
+    },
+    {
+      "train_step": 8956,
+      "loss": 0.5697974560316382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5697974560316382
+    },
+    {
+      "train_step": 8957,
+      "loss": 0.351649494812699,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.351649494812699
+    },
+    {
+      "train_step": 8958,
+      "loss": 0.29269024907053187,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.29269024907053187
+    },
+    {
+      "train_step": 8959,
+      "loss": 0.6859046093582025,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6859046093582025
+    },
+    {
+      "train_step": 8960,
+      "loss": 0.4807963598274594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4807963598274594
+    },
+    {
+      "train_step": 8961,
+      "loss": 0.6245203696017191,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6245203696017191
+    },
+    {
+      "train_step": 8962,
+      "loss": 0.6181142205454908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6181142205454908
+    },
+    {
+      "train_step": 8963,
+      "loss": 0.6228717557388568,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6228717557388568
+    },
+    {
+      "train_step": 8964,
+      "loss": 0.46181437978836204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46181437978836204
+    },
+    {
+      "train_step": 8965,
+      "loss": 0.7159393080009088,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7159393080009088
+    },
+    {
+      "train_step": 8966,
+      "loss": 0.5213772062820048,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5213772062820048
+    },
+    {
+      "train_step": 8967,
+      "loss": 0.5197261709619947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5197261709619947
+    },
+    {
+      "train_step": 8968,
+      "loss": 0.45430276821999976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45430276821999976
+    },
+    {
+      "train_step": 8969,
+      "loss": 0.5172237261927175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5172237261927175
+    },
+    {
+      "train_step": 8970,
+      "loss": 0.6215079546201274,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6215079546201274
+    },
+    {
+      "train_step": 8971,
+      "loss": 0.4559991083847563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4559991083847563
+    },
+    {
+      "train_step": 8972,
+      "loss": 0.3713403316554105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3713403316554105
+    },
+    {
+      "train_step": 8973,
+      "loss": 0.5414064933323531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5414064933323531
+    },
+    {
+      "train_step": 8974,
+      "loss": 0.4356809483226337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4356809483226337
+    },
+    {
+      "train_step": 8975,
+      "loss": 0.5234169899665986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5234169899665986
+    },
+    {
+      "train_step": 8976,
+      "loss": 0.46251613655215934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46251613655215934
+    },
+    {
+      "train_step": 8977,
+      "loss": 0.47004842007513226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47004842007513226
+    },
+    {
+      "train_step": 8978,
+      "loss": 0.5157408995516978,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5157408995516978
+    },
+    {
+      "train_step": 8979,
+      "loss": 0.41893764923394733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41893764923394733
+    },
+    {
+      "train_step": 8980,
+      "loss": 0.5497522145202995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5497522145202995
+    },
+    {
+      "train_step": 8981,
+      "loss": 0.41376245553915525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41376245553915525
+    },
+    {
+      "train_step": 8982,
+      "loss": 0.3191852678283432,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3191852678283432
+    },
+    {
+      "train_step": 8983,
+      "loss": 0.4163106539432221,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4163106539432221
+    },
+    {
+      "train_step": 8984,
+      "loss": 0.7588144695168595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7588144695168595
+    },
+    {
+      "train_step": 8985,
+      "loss": 0.37567054326260885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37567054326260885
+    },
+    {
+      "train_step": 8986,
+      "loss": 0.44194482818491226,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44194482818491226
+    },
+    {
+      "train_step": 8987,
+      "loss": 0.483764064684539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.483764064684539
+    },
+    {
+      "train_step": 8988,
+      "loss": 0.5234673120134319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5234673120134319
+    },
+    {
+      "train_step": 8989,
+      "loss": 0.7529426983774273,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7529426983774273
+    },
+    {
+      "train_step": 8990,
+      "loss": 0.6282951417005829,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6282951417005829
+    },
+    {
+      "train_step": 8991,
+      "loss": 0.440310152233595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.440310152233595
+    },
+    {
+      "train_step": 8992,
+      "loss": 0.7663250739016509,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7663250739016509
+    },
+    {
+      "train_step": 8993,
+      "loss": 0.48434107803807047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48434107803807047
+    },
+    {
+      "train_step": 8994,
+      "loss": 0.6455940334010217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6455940334010217
+    },
+    {
+      "train_step": 8995,
+      "loss": 0.7654643404634307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7654643404634307
+    },
+    {
+      "train_step": 8996,
+      "loss": 0.49647301619452183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49647301619452183
+    },
+    {
+      "train_step": 8997,
+      "loss": 0.5829553157270674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5829553157270674
+    },
+    {
+      "train_step": 8998,
+      "loss": 0.5309019029302016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5309019029302016
+    },
+    {
+      "train_step": 8999,
+      "loss": 0.5552686257284238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5552686257284238
+    },
+    {
+      "train_step": 9000,
+      "loss": 0.42175613280044105,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42175613280044105
+    },
+    {
+      "train_step": 9001,
+      "loss": 0.697585891676926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.697585891676926
+    },
+    {
+      "train_step": 9002,
+      "loss": 0.44447822349728583,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44447822349728583
+    },
+    {
+      "train_step": 9003,
+      "loss": 0.49571799052705134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49571799052705134
+    },
+    {
+      "train_step": 9004,
+      "loss": 0.4126660112457403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4126660112457403
+    },
+    {
+      "train_step": 9005,
+      "loss": 0.49457033355509067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49457033355509067
+    },
+    {
+      "train_step": 9006,
+      "loss": 0.33718691965891706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33718691965891706
+    },
+    {
+      "train_step": 9007,
+      "loss": 0.6341922294871849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6341922294871849
+    },
+    {
+      "train_step": 9008,
+      "loss": 0.4757526689073229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4757526689073229
+    },
+    {
+      "train_step": 9009,
+      "loss": 0.4220890828344078,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4220890828344078
+    },
+    {
+      "train_step": 9010,
+      "loss": 0.4552100396381964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4552100396381964
+    },
+    {
+      "train_step": 9011,
+      "loss": 0.5098483736678814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5098483736678814
+    },
+    {
+      "train_step": 9012,
+      "loss": 0.6666668288635866,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6666668288635866
+    },
+    {
+      "train_step": 9013,
+      "loss": 0.6209306198974357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6209306198974357
+    },
+    {
+      "train_step": 9014,
+      "loss": 0.6144542620580805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6144542620580805
+    },
+    {
+      "train_step": 9015,
+      "loss": 0.5805385396855962,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5805385396855962
+    },
+    {
+      "train_step": 9016,
+      "loss": 0.37031133920469894,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37031133920469894
+    },
+    {
+      "train_step": 9017,
+      "loss": 0.5137306021196133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5137306021196133
+    },
+    {
+      "train_step": 9018,
+      "loss": 0.49869222564806504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49869222564806504
+    },
+    {
+      "train_step": 9019,
+      "loss": 0.5498811979287823,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5498811979287823
+    },
+    {
+      "train_step": 9020,
+      "loss": 0.5378290296222051,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5378290296222051
+    },
+    {
+      "train_step": 9021,
+      "loss": 0.562217192026244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.562217192026244
+    },
+    {
+      "train_step": 9022,
+      "loss": 0.4817852802251497,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4817852802251497
+    },
+    {
+      "train_step": 9023,
+      "loss": 0.4682521091556672,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4682521091556672
+    },
+    {
+      "train_step": 9024,
+      "loss": 0.6156745362842939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6156745362842939
+    },
+    {
+      "train_step": 9025,
+      "loss": 0.724930009091199,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.724930009091199
+    },
+    {
+      "train_step": 9026,
+      "loss": 0.8135045375319643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8135045375319643
+    },
+    {
+      "train_step": 9027,
+      "loss": 0.8239464696813542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8239464696813542
+    },
+    {
+      "train_step": 9028,
+      "loss": 0.508143795132948,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.508143795132948
+    },
+    {
+      "train_step": 9029,
+      "loss": 0.6811577379300727,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6811577379300727
+    },
+    {
+      "train_step": 9030,
+      "loss": 0.4449906708500975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4449906708500975
+    },
+    {
+      "train_step": 9031,
+      "loss": 0.4055639633445546,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4055639633445546
+    },
+    {
+      "train_step": 9032,
+      "loss": 0.5379781285096801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5379781285096801
+    },
+    {
+      "train_step": 9033,
+      "loss": 0.390248335180784,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.390248335180784
+    },
+    {
+      "train_step": 9034,
+      "loss": 0.5233897825834753,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5233897825834753
+    },
+    {
+      "train_step": 9035,
+      "loss": 0.5512130947445799,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5512130947445799
+    },
+    {
+      "train_step": 9036,
+      "loss": 0.4646146466517515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4646146466517515
+    },
+    {
+      "train_step": 9037,
+      "loss": 0.5662460401171362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5662460401171362
+    },
+    {
+      "train_step": 9038,
+      "loss": 0.665897427124853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.665897427124853
+    },
+    {
+      "train_step": 9039,
+      "loss": 0.5155434672423724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5155434672423724
+    },
+    {
+      "train_step": 9040,
+      "loss": 0.43613240169590306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43613240169590306
+    },
+    {
+      "train_step": 9041,
+      "loss": 0.5881184909489644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5881184909489644
+    },
+    {
+      "train_step": 9042,
+      "loss": 0.4862820526252298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4862820526252298
+    },
+    {
+      "train_step": 9043,
+      "loss": 0.6695780109132633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6695780109132633
+    },
+    {
+      "train_step": 9044,
+      "loss": 0.5117868439354801,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5117868439354801
+    },
+    {
+      "train_step": 9045,
+      "loss": 0.5906325298093068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5906325298093068
+    },
+    {
+      "train_step": 9046,
+      "loss": 0.4123483200356691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4123483200356691
+    },
+    {
+      "train_step": 9047,
+      "loss": 0.6503735778242266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6503735778242266
+    },
+    {
+      "train_step": 9048,
+      "loss": 0.3770181229057525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3770181229057525
+    },
+    {
+      "train_step": 9049,
+      "loss": 0.5164391177744585,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5164391177744585
+    },
+    {
+      "train_step": 9050,
+      "loss": 0.5585482648075156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5585482648075156
+    },
+    {
+      "train_step": 9051,
+      "loss": 0.5730079137230312,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5730079137230312
+    },
+    {
+      "train_step": 9052,
+      "loss": 0.6043964740348027,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6043964740348027
+    },
+    {
+      "train_step": 9053,
+      "loss": 0.7835506843159422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7835506843159422
+    },
+    {
+      "train_step": 9054,
+      "loss": 0.5000754067341707,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5000754067341707
+    },
+    {
+      "train_step": 9055,
+      "loss": 0.5269503887233642,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5269503887233642
+    },
+    {
+      "train_step": 9056,
+      "loss": 0.41933892291926883,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41933892291926883
+    },
+    {
+      "train_step": 9057,
+      "loss": 0.5424387634113124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5424387634113124
+    },
+    {
+      "train_step": 9058,
+      "loss": 0.4740497807677946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4740497807677946
+    },
+    {
+      "train_step": 9059,
+      "loss": 0.6522635519402342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6522635519402342
+    },
+    {
+      "train_step": 9060,
+      "loss": 0.6215134388052681,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6215134388052681
+    },
+    {
+      "train_step": 9061,
+      "loss": 0.5706019731289964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5706019731289964
+    },
+    {
+      "train_step": 9062,
+      "loss": 0.5844619001766203,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5844619001766203
+    },
+    {
+      "train_step": 9063,
+      "loss": 0.44289630436444133,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44289630436444133
+    },
+    {
+      "train_step": 9064,
+      "loss": 0.5951390733033413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5951390733033413
+    },
+    {
+      "train_step": 9065,
+      "loss": 0.5849437416308091,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5849437416308091
+    },
+    {
+      "train_step": 9066,
+      "loss": 0.5576463142837207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5576463142837207
+    },
+    {
+      "train_step": 9067,
+      "loss": 0.35929240296331705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35929240296331705
+    },
+    {
+      "train_step": 9068,
+      "loss": 0.6430837681235952,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6430837681235952
+    },
+    {
+      "train_step": 9069,
+      "loss": 0.5216172418219578,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5216172418219578
+    },
+    {
+      "train_step": 9070,
+      "loss": 0.4627237813996445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4627237813996445
+    },
+    {
+      "train_step": 9071,
+      "loss": 0.5554358783093716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5554358783093716
+    },
+    {
+      "train_step": 9072,
+      "loss": 0.3711982190690313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3711982190690313
+    },
+    {
+      "train_step": 9073,
+      "loss": 0.6006246190677937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6006246190677937
+    },
+    {
+      "train_step": 9074,
+      "loss": 0.5191477385569717,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5191477385569717
+    },
+    {
+      "train_step": 9075,
+      "loss": 0.48670414663895734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48670414663895734
+    },
+    {
+      "train_step": 9076,
+      "loss": 0.40319247314672163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40319247314672163
+    },
+    {
+      "train_step": 9077,
+      "loss": 0.6027481484522654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6027481484522654
+    },
+    {
+      "train_step": 9078,
+      "loss": 0.5423588427634062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5423588427634062
+    },
+    {
+      "train_step": 9079,
+      "loss": 0.5229243286533259,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5229243286533259
+    },
+    {
+      "train_step": 9080,
+      "loss": 0.6646723133457058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6646723133457058
+    },
+    {
+      "train_step": 9081,
+      "loss": 0.4250865833207462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4250865833207462
+    },
+    {
+      "train_step": 9082,
+      "loss": 0.632314592887272,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.632314592887272
+    },
+    {
+      "train_step": 9083,
+      "loss": 0.6575092974192627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6575092974192627
+    },
+    {
+      "train_step": 9084,
+      "loss": 0.5770151673496972,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5770151673496972
+    },
+    {
+      "train_step": 9085,
+      "loss": 0.6662039660084526,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6662039660084526
+    },
+    {
+      "train_step": 9086,
+      "loss": 0.4802311327477451,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4802311327477451
+    },
+    {
+      "train_step": 9087,
+      "loss": 0.6783319432515387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6783319432515387
+    },
+    {
+      "train_step": 9088,
+      "loss": 0.5852783782894504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5852783782894504
+    },
+    {
+      "train_step": 9089,
+      "loss": 0.7906050570973436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7906050570973436
+    },
+    {
+      "train_step": 9090,
+      "loss": 0.5813005157937655,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5813005157937655
+    },
+    {
+      "train_step": 9091,
+      "loss": 0.5655404518337401,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5655404518337401
+    },
+    {
+      "train_step": 9092,
+      "loss": 0.6270067491258174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6270067491258174
+    },
+    {
+      "train_step": 9093,
+      "loss": 0.6141074530758167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6141074530758167
+    },
+    {
+      "train_step": 9094,
+      "loss": 0.699928922028377,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.699928922028377
+    },
+    {
+      "train_step": 9095,
+      "loss": 0.7890430738272732,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7890430738272732
+    },
+    {
+      "train_step": 9096,
+      "loss": 0.43492646188265527,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43492646188265527
+    },
+    {
+      "train_step": 9097,
+      "loss": 0.5641042938074889,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5641042938074889
+    },
+    {
+      "train_step": 9098,
+      "loss": 0.5107544775777647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5107544775777647
+    },
+    {
+      "train_step": 9099,
+      "loss": 0.4226077235837616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4226077235837616
+    },
+    {
+      "train_step": 9100,
+      "loss": 0.5355159067572166,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5355159067572166
+    },
+    {
+      "train_step": 9101,
+      "loss": 0.43743593468689684,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43743593468689684
+    },
+    {
+      "train_step": 9102,
+      "loss": 0.34553263492618796,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34553263492618796
+    },
+    {
+      "train_step": 9103,
+      "loss": 0.5859606022687185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5859606022687185
+    },
+    {
+      "train_step": 9104,
+      "loss": 0.6373566370295313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6373566370295313
+    },
+    {
+      "train_step": 9105,
+      "loss": 0.7368000823294469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7368000823294469
+    },
+    {
+      "train_step": 9106,
+      "loss": 0.5577422277544926,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5577422277544926
+    },
+    {
+      "train_step": 9107,
+      "loss": 0.503155159847127,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.503155159847127
+    },
+    {
+      "train_step": 9108,
+      "loss": 0.45980244282398247,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45980244282398247
+    },
+    {
+      "train_step": 9109,
+      "loss": 0.6202759111447175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6202759111447175
+    },
+    {
+      "train_step": 9110,
+      "loss": 0.45710981639829945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45710981639829945
+    },
+    {
+      "train_step": 9111,
+      "loss": 0.44546541123845823,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44546541123845823
+    },
+    {
+      "train_step": 9112,
+      "loss": 0.31083756303814986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.31083756303814986
+    },
+    {
+      "train_step": 9113,
+      "loss": 0.5629200217013809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5629200217013809
+    },
+    {
+      "train_step": 9114,
+      "loss": 0.4612228037709149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4612228037709149
+    },
+    {
+      "train_step": 9115,
+      "loss": 0.4811350432741266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4811350432741266
+    },
+    {
+      "train_step": 9116,
+      "loss": 0.6413045144880871,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6413045144880871
+    },
+    {
+      "train_step": 9117,
+      "loss": 0.6672672881823483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6672672881823483
+    },
+    {
+      "train_step": 9118,
+      "loss": 0.5309120865084687,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5309120865084687
+    },
+    {
+      "train_step": 9119,
+      "loss": 0.43242236237699316,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43242236237699316
+    },
+    {
+      "train_step": 9120,
+      "loss": 0.4177744416668028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4177744416668028
+    },
+    {
+      "train_step": 9121,
+      "loss": 0.7433287507394601,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7433287507394601
+    },
+    {
+      "train_step": 9122,
+      "loss": 0.4664432677199175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4664432677199175
+    },
+    {
+      "train_step": 9123,
+      "loss": 0.6723336371789441,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6723336371789441
+    },
+    {
+      "train_step": 9124,
+      "loss": 0.34432201404378143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34432201404378143
+    },
+    {
+      "train_step": 9125,
+      "loss": 0.7656849867792335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7656849867792335
+    },
+    {
+      "train_step": 9126,
+      "loss": 0.4163199506599288,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4163199506599288
+    },
+    {
+      "train_step": 9127,
+      "loss": 0.47407628729511553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47407628729511553
+    },
+    {
+      "train_step": 9128,
+      "loss": 0.540106581024966,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.540106581024966
+    },
+    {
+      "train_step": 9129,
+      "loss": 0.5406970010486646,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5406970010486646
+    },
+    {
+      "train_step": 9130,
+      "loss": 0.7008679285253514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7008679285253514
+    },
+    {
+      "train_step": 9131,
+      "loss": 0.4695259592571854,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4695259592571854
+    },
+    {
+      "train_step": 9132,
+      "loss": 0.49917514907204824,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49917514907204824
+    },
+    {
+      "train_step": 9133,
+      "loss": 0.34873425612480197,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34873425612480197
+    },
+    {
+      "train_step": 9134,
+      "loss": 0.35788652713265984,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35788652713265984
+    },
+    {
+      "train_step": 9135,
+      "loss": 0.4360324738367634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4360324738367634
+    },
+    {
+      "train_step": 9136,
+      "loss": 0.3882595913269397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3882595913269397
+    },
+    {
+      "train_step": 9137,
+      "loss": 0.43641486138701663,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43641486138701663
+    },
+    {
+      "train_step": 9138,
+      "loss": 0.5725513833842252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5725513833842252
+    },
+    {
+      "train_step": 9139,
+      "loss": 0.39871527162164094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39871527162164094
+    },
+    {
+      "train_step": 9140,
+      "loss": 0.4996949902534222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4996949902534222
+    },
+    {
+      "train_step": 9141,
+      "loss": 0.5469703591679721,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5469703591679721
+    },
+    {
+      "train_step": 9142,
+      "loss": 0.6630349133626339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6630349133626339
+    },
+    {
+      "train_step": 9143,
+      "loss": 0.5969641937111939,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5969641937111939
+    },
+    {
+      "train_step": 9144,
+      "loss": 0.645481729093864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.645481729093864
+    },
+    {
+      "train_step": 9145,
+      "loss": 0.5187631367006216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5187631367006216
+    },
+    {
+      "train_step": 9146,
+      "loss": 0.457241609152726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.457241609152726
+    },
+    {
+      "train_step": 9147,
+      "loss": 0.5671833921772433,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5671833921772433
+    },
+    {
+      "train_step": 9148,
+      "loss": 0.6093667635693021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6093667635693021
+    },
+    {
+      "train_step": 9149,
+      "loss": 0.44400842035867355,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44400842035867355
+    },
+    {
+      "train_step": 9150,
+      "loss": 0.5571990370784251,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5571990370784251
+    },
+    {
+      "train_step": 9151,
+      "loss": 0.4844600544442208,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4844600544442208
+    },
+    {
+      "train_step": 9152,
+      "loss": 0.6473855206365859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6473855206365859
+    },
+    {
+      "train_step": 9153,
+      "loss": 0.5875604516916065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5875604516916065
+    },
+    {
+      "train_step": 9154,
+      "loss": 0.6085825952653084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6085825952653084
+    },
+    {
+      "train_step": 9155,
+      "loss": 0.8292671539102664,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8292671539102664
+    },
+    {
+      "train_step": 9156,
+      "loss": 0.6047799228979482,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6047799228979482
+    },
+    {
+      "train_step": 9157,
+      "loss": 0.5269962118039049,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5269962118039049
+    },
+    {
+      "train_step": 9158,
+      "loss": 0.568927278532569,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.568927278532569
+    },
+    {
+      "train_step": 9159,
+      "loss": 0.6159718499797032,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6159718499797032
+    },
+    {
+      "train_step": 9160,
+      "loss": 0.5396277212623781,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5396277212623781
+    },
+    {
+      "train_step": 9161,
+      "loss": 0.5734159098061368,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5734159098061368
+    },
+    {
+      "train_step": 9162,
+      "loss": 0.7142313742226937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7142313742226937
+    },
+    {
+      "train_step": 9163,
+      "loss": 0.47978190091370504,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47978190091370504
+    },
+    {
+      "train_step": 9164,
+      "loss": 0.6064416760611016,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6064416760611016
+    },
+    {
+      "train_step": 9165,
+      "loss": 0.4155054210503886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4155054210503886
+    },
+    {
+      "train_step": 9166,
+      "loss": 0.5727004442431992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5727004442431992
+    },
+    {
+      "train_step": 9167,
+      "loss": 0.4783651822454439,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4783651822454439
+    },
+    {
+      "train_step": 9168,
+      "loss": 0.5783639984928481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5783639984928481
+    },
+    {
+      "train_step": 9169,
+      "loss": 0.4336540632975685,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4336540632975685
+    },
+    {
+      "train_step": 9170,
+      "loss": 0.7179523405476167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7179523405476167
+    },
+    {
+      "train_step": 9171,
+      "loss": 0.34532956582729957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34532956582729957
+    },
+    {
+      "train_step": 9172,
+      "loss": 0.4368017689245811,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4368017689245811
+    },
+    {
+      "train_step": 9173,
+      "loss": 0.4715964000188427,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4715964000188427
+    },
+    {
+      "train_step": 9174,
+      "loss": 0.47555818329509736,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47555818329509736
+    },
+    {
+      "train_step": 9175,
+      "loss": 0.5402436232739211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5402436232739211
+    },
+    {
+      "train_step": 9176,
+      "loss": 0.5354384002385923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5354384002385923
+    },
+    {
+      "train_step": 9177,
+      "loss": 0.6507544720352945,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6507544720352945
+    },
+    {
+      "train_step": 9178,
+      "loss": 0.6345688645499611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6345688645499611
+    },
+    {
+      "train_step": 9179,
+      "loss": 0.4537500970790448,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4537500970790448
+    },
+    {
+      "train_step": 9180,
+      "loss": 0.3640615211348706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3640615211348706
+    },
+    {
+      "train_step": 9181,
+      "loss": 0.6735194287275632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6735194287275632
+    },
+    {
+      "train_step": 9182,
+      "loss": 0.3910631791500013,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3910631791500013
+    },
+    {
+      "train_step": 9183,
+      "loss": 0.5552578591685514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5552578591685514
+    },
+    {
+      "train_step": 9184,
+      "loss": 0.6916970959800733,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6916970959800733
+    },
+    {
+      "train_step": 9185,
+      "loss": 0.7112110351625462,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7112110351625462
+    },
+    {
+      "train_step": 9186,
+      "loss": 0.576789742318337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.576789742318337
+    },
+    {
+      "train_step": 9187,
+      "loss": 0.45525109510074413,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45525109510074413
+    },
+    {
+      "train_step": 9188,
+      "loss": 0.5210181653384486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5210181653384486
+    },
+    {
+      "train_step": 9189,
+      "loss": 0.6254663214004484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6254663214004484
+    },
+    {
+      "train_step": 9190,
+      "loss": 0.5967349288867589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5967349288867589
+    },
+    {
+      "train_step": 9191,
+      "loss": 0.604348067839113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.604348067839113
+    },
+    {
+      "train_step": 9192,
+      "loss": 0.42251466256606185,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42251466256606185
+    },
+    {
+      "train_step": 9193,
+      "loss": 0.5123210697415644,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5123210697415644
+    },
+    {
+      "train_step": 9194,
+      "loss": 0.37386583696708475,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37386583696708475
+    },
+    {
+      "train_step": 9195,
+      "loss": 0.38655422915176973,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38655422915176973
+    },
+    {
+      "train_step": 9196,
+      "loss": 0.9517183161130831,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9517183161130831
+    },
+    {
+      "train_step": 9197,
+      "loss": 0.4813980664284977,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4813980664284977
+    },
+    {
+      "train_step": 9198,
+      "loss": 0.7926154323786635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7926154323786635
+    },
+    {
+      "train_step": 9199,
+      "loss": 0.6113006174851074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6113006174851074
+    },
+    {
+      "train_step": 9200,
+      "loss": 0.2947738941371266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.2947738941371266
+    },
+    {
+      "train_step": 9201,
+      "loss": 0.5524738318778113,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5524738318778113
+    },
+    {
+      "train_step": 9202,
+      "loss": 0.7153632076462934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7153632076462934
+    },
+    {
+      "train_step": 9203,
+      "loss": 0.5745866048107313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5745866048107313
+    },
+    {
+      "train_step": 9204,
+      "loss": 0.36477997723086053,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.36477997723086053
+    },
+    {
+      "train_step": 9205,
+      "loss": 0.6723853917970244,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6723853917970244
+    },
+    {
+      "train_step": 9206,
+      "loss": 0.4661370037094074,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4661370037094074
+    },
+    {
+      "train_step": 9207,
+      "loss": 0.6921321025754249,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6921321025754249
+    },
+    {
+      "train_step": 9208,
+      "loss": 0.38508201878553394,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38508201878553394
+    },
+    {
+      "train_step": 9209,
+      "loss": 0.4360486088778075,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4360486088778075
+    },
+    {
+      "train_step": 9210,
+      "loss": 0.5889482512269035,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5889482512269035
+    },
+    {
+      "train_step": 9211,
+      "loss": 0.5659163612577598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5659163612577598
+    },
+    {
+      "train_step": 9212,
+      "loss": 0.3805052486879111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3805052486879111
+    },
+    {
+      "train_step": 9213,
+      "loss": 0.783719798262728,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.783719798262728
+    },
+    {
+      "train_step": 9214,
+      "loss": 0.44550855660137584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44550855660137584
+    },
+    {
+      "train_step": 9215,
+      "loss": 0.4732948975904975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4732948975904975
+    },
+    {
+      "train_step": 9216,
+      "loss": 0.41097845759123114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41097845759123114
+    },
+    {
+      "train_step": 9217,
+      "loss": 0.5233986631373554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5233986631373554
+    },
+    {
+      "train_step": 9218,
+      "loss": 0.7283954082455649,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7283954082455649
+    },
+    {
+      "train_step": 9219,
+      "loss": 0.60687795493631,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.60687795493631
+    },
+    {
+      "train_step": 9220,
+      "loss": 0.6261100105750277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6261100105750277
+    },
+    {
+      "train_step": 9221,
+      "loss": 0.4840255894831724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4840255894831724
+    },
+    {
+      "train_step": 9222,
+      "loss": 0.5328421485574938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5328421485574938
+    },
+    {
+      "train_step": 9223,
+      "loss": 0.4898452492115611,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4898452492115611
+    },
+    {
+      "train_step": 9224,
+      "loss": 0.5565232755658411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5565232755658411
+    },
+    {
+      "train_step": 9225,
+      "loss": 0.6410896459166211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6410896459166211
+    },
+    {
+      "train_step": 9226,
+      "loss": 0.44191116502299277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44191116502299277
+    },
+    {
+      "train_step": 9227,
+      "loss": 0.7243283204234315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7243283204234315
+    },
+    {
+      "train_step": 9228,
+      "loss": 0.6661161451477307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6661161451477307
+    },
+    {
+      "train_step": 9229,
+      "loss": 0.7209668878179456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7209668878179456
+    },
+    {
+      "train_step": 9230,
+      "loss": 0.6102087010158594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6102087010158594
+    },
+    {
+      "train_step": 9231,
+      "loss": 0.4008802490216305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4008802490216305
+    },
+    {
+      "train_step": 9232,
+      "loss": 0.48291555145708176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48291555145708176
+    },
+    {
+      "train_step": 9233,
+      "loss": 0.4161210455638937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4161210455638937
+    },
+    {
+      "train_step": 9234,
+      "loss": 0.28582875253548357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.28582875253548357
+    },
+    {
+      "train_step": 9235,
+      "loss": 0.38442391936366693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38442391936366693
+    },
+    {
+      "train_step": 9236,
+      "loss": 0.6176581413599225,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6176581413599225
+    },
+    {
+      "train_step": 9237,
+      "loss": 0.6020312779065012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6020312779065012
+    },
+    {
+      "train_step": 9238,
+      "loss": 0.5855080602405965,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5855080602405965
+    },
+    {
+      "train_step": 9239,
+      "loss": 0.42522793272448045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42522793272448045
+    },
+    {
+      "train_step": 9240,
+      "loss": 0.48788649775885023,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48788649775885023
+    },
+    {
+      "train_step": 9241,
+      "loss": 0.5853375678901406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5853375678901406
+    },
+    {
+      "train_step": 9242,
+      "loss": 0.3963186656326112,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3963186656326112
+    },
+    {
+      "train_step": 9243,
+      "loss": 0.6334146341625942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6334146341625942
+    },
+    {
+      "train_step": 9244,
+      "loss": 0.44892564112740097,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44892564112740097
+    },
+    {
+      "train_step": 9245,
+      "loss": 0.6307793660689814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6307793660689814
+    },
+    {
+      "train_step": 9246,
+      "loss": 0.5035642626480261,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5035642626480261
+    },
+    {
+      "train_step": 9247,
+      "loss": 0.6443424237467517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6443424237467517
+    },
+    {
+      "train_step": 9248,
+      "loss": 0.44533154903471306,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44533154903471306
+    },
+    {
+      "train_step": 9249,
+      "loss": 0.6452224827054183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6452224827054183
+    },
+    {
+      "train_step": 9250,
+      "loss": 0.5432459082164923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5432459082164923
+    },
+    {
+      "train_step": 9251,
+      "loss": 0.4985998011345833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4985998011345833
+    },
+    {
+      "train_step": 9252,
+      "loss": 0.5550561968587513,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5550561968587513
+    },
+    {
+      "train_step": 9253,
+      "loss": 0.5220470028295802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5220470028295802
+    },
+    {
+      "train_step": 9254,
+      "loss": 0.48013620438852006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48013620438852006
+    },
+    {
+      "train_step": 9255,
+      "loss": 0.4123643150034402,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4123643150034402
+    },
+    {
+      "train_step": 9256,
+      "loss": 0.5468102978055874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5468102978055874
+    },
+    {
+      "train_step": 9257,
+      "loss": 0.6179491288236975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6179491288236975
+    },
+    {
+      "train_step": 9258,
+      "loss": 0.5682627083548734,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5682627083548734
+    },
+    {
+      "train_step": 9259,
+      "loss": 0.5922487470649892,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5922487470649892
+    },
+    {
+      "train_step": 9260,
+      "loss": 0.5258035292230443,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5258035292230443
+    },
+    {
+      "train_step": 9261,
+      "loss": 0.6612296669233946,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6612296669233946
+    },
+    {
+      "train_step": 9262,
+      "loss": 0.48183792189915464,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48183792189915464
+    },
+    {
+      "train_step": 9263,
+      "loss": 0.57329097092809,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.57329097092809
+    },
+    {
+      "train_step": 9264,
+      "loss": 0.476059003034331,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.476059003034331
+    },
+    {
+      "train_step": 9265,
+      "loss": 0.4349646312091047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4349646312091047
+    },
+    {
+      "train_step": 9266,
+      "loss": 0.8181801731737308,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8181801731737308
+    },
+    {
+      "train_step": 9267,
+      "loss": 0.5510498748211895,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5510498748211895
+    },
+    {
+      "train_step": 9268,
+      "loss": 0.7162050005085339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7162050005085339
+    },
+    {
+      "train_step": 9269,
+      "loss": 0.6246244771811483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6246244771811483
+    },
+    {
+      "train_step": 9270,
+      "loss": 0.4841366613516263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4841366613516263
+    },
+    {
+      "train_step": 9271,
+      "loss": 0.6635810012710962,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6635810012710962
+    },
+    {
+      "train_step": 9272,
+      "loss": 0.5526224300933028,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5526224300933028
+    },
+    {
+      "train_step": 9273,
+      "loss": 0.6174934707810439,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6174934707810439
+    },
+    {
+      "train_step": 9274,
+      "loss": 0.4646728235458671,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4646728235458671
+    },
+    {
+      "train_step": 9275,
+      "loss": 0.6081064119603219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6081064119603219
+    },
+    {
+      "train_step": 9276,
+      "loss": 0.47781536137236746,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47781536137236746
+    },
+    {
+      "train_step": 9277,
+      "loss": 0.41881691698153545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41881691698153545
+    },
+    {
+      "train_step": 9278,
+      "loss": 0.35895097007588594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35895097007588594
+    },
+    {
+      "train_step": 9279,
+      "loss": 0.5341934261448342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5341934261448342
+    },
+    {
+      "train_step": 9280,
+      "loss": 0.6566445077353176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6566445077353176
+    },
+    {
+      "train_step": 9281,
+      "loss": 0.6076084738465444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6076084738465444
+    },
+    {
+      "train_step": 9282,
+      "loss": 0.4555661253630362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4555661253630362
+    },
+    {
+      "train_step": 9283,
+      "loss": 0.4473969230550114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4473969230550114
+    },
+    {
+      "train_step": 9284,
+      "loss": 0.5743761215114322,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5743761215114322
+    },
+    {
+      "train_step": 9285,
+      "loss": 0.5907323631985889,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5907323631985889
+    },
+    {
+      "train_step": 9286,
+      "loss": 0.5602378228861422,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5602378228861422
+    },
+    {
+      "train_step": 9287,
+      "loss": 0.44054122607595975,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44054122607595975
+    },
+    {
+      "train_step": 9288,
+      "loss": 0.6442122964988218,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6442122964988218
+    },
+    {
+      "train_step": 9289,
+      "loss": 0.43877398680172525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43877398680172525
+    },
+    {
+      "train_step": 9290,
+      "loss": 0.48762579350828705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48762579350828705
+    },
+    {
+      "train_step": 9291,
+      "loss": 0.5158675829664731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5158675829664731
+    },
+    {
+      "train_step": 9292,
+      "loss": 0.5548913552530635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5548913552530635
+    },
+    {
+      "train_step": 9293,
+      "loss": 0.5009973309858436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5009973309858436
+    },
+    {
+      "train_step": 9294,
+      "loss": 0.3209012983356822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3209012983356822
+    },
+    {
+      "train_step": 9295,
+      "loss": 0.6708770148459314,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6708770148459314
+    },
+    {
+      "train_step": 9296,
+      "loss": 0.44778346351456044,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44778346351456044
+    },
+    {
+      "train_step": 9297,
+      "loss": 0.6415389866814691,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6415389866814691
+    },
+    {
+      "train_step": 9298,
+      "loss": 0.46909597737146014,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46909597737146014
+    },
+    {
+      "train_step": 9299,
+      "loss": 0.5834453042216778,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5834453042216778
+    },
+    {
+      "train_step": 9300,
+      "loss": 0.5062463117298237,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5062463117298237
+    },
+    {
+      "train_step": 9301,
+      "loss": 0.468140348017135,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.468140348017135
+    },
+    {
+      "train_step": 9302,
+      "loss": 0.7682011052503217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7682011052503217
+    },
+    {
+      "train_step": 9303,
+      "loss": 0.42688653383834235,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42688653383834235
+    },
+    {
+      "train_step": 9304,
+      "loss": 0.667995619902478,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.667995619902478
+    },
+    {
+      "train_step": 9305,
+      "loss": 0.4978613107877448,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4978613107877448
+    },
+    {
+      "train_step": 9306,
+      "loss": 0.545771406023957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.545771406023957
+    },
+    {
+      "train_step": 9307,
+      "loss": 0.6200599043754176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6200599043754176
+    },
+    {
+      "train_step": 9308,
+      "loss": 0.7194132228501542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7194132228501542
+    },
+    {
+      "train_step": 9309,
+      "loss": 0.6096986314253741,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6096986314253741
+    },
+    {
+      "train_step": 9310,
+      "loss": 0.46565021913371446,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46565021913371446
+    },
+    {
+      "train_step": 9311,
+      "loss": 0.3837868111163183,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3837868111163183
+    },
+    {
+      "train_step": 9312,
+      "loss": 0.38718722641813247,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38718722641813247
+    },
+    {
+      "train_step": 9313,
+      "loss": 0.6768675574628731,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6768675574628731
+    },
+    {
+      "train_step": 9314,
+      "loss": 0.5868112845692951,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5868112845692951
+    },
+    {
+      "train_step": 9315,
+      "loss": 0.3952694614192825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3952694614192825
+    },
+    {
+      "train_step": 9316,
+      "loss": 0.7787236992584595,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7787236992584595
+    },
+    {
+      "train_step": 9317,
+      "loss": 0.5145399685410512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5145399685410512
+    },
+    {
+      "train_step": 9318,
+      "loss": 0.9147231669311195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.9147231669311195
+    },
+    {
+      "train_step": 9319,
+      "loss": 0.6545472178963924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6545472178963924
+    },
+    {
+      "train_step": 9320,
+      "loss": 0.7064003535749069,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7064003535749069
+    },
+    {
+      "train_step": 9321,
+      "loss": 0.6063606572134993,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6063606572134993
+    },
+    {
+      "train_step": 9322,
+      "loss": 0.7840643538452658,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7840643538452658
+    },
+    {
+      "train_step": 9323,
+      "loss": 0.597305109801106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.597305109801106
+    },
+    {
+      "train_step": 9324,
+      "loss": 0.5030072443253923,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5030072443253923
+    },
+    {
+      "train_step": 9325,
+      "loss": 0.6773268344205985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6773268344205985
+    },
+    {
+      "train_step": 9326,
+      "loss": 0.47823810306044584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47823810306044584
+    },
+    {
+      "train_step": 9327,
+      "loss": 0.7185445230737543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7185445230737543
+    },
+    {
+      "train_step": 9328,
+      "loss": 0.45049433955788065,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45049433955788065
+    },
+    {
+      "train_step": 9329,
+      "loss": 0.6123864265809982,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6123864265809982
+    },
+    {
+      "train_step": 9330,
+      "loss": 0.35615993293304193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35615993293304193
+    },
+    {
+      "train_step": 9331,
+      "loss": 0.613766045092152,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.613766045092152
+    },
+    {
+      "train_step": 9332,
+      "loss": 0.6322541126405927,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6322541126405927
+    },
+    {
+      "train_step": 9333,
+      "loss": 0.5184036403937217,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5184036403937217
+    },
+    {
+      "train_step": 9334,
+      "loss": 0.751599415988619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.751599415988619
+    },
+    {
+      "train_step": 9335,
+      "loss": 0.6273237331687983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6273237331687983
+    },
+    {
+      "train_step": 9336,
+      "loss": 0.5576314801270262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5576314801270262
+    },
+    {
+      "train_step": 9337,
+      "loss": 0.5331413485705524,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5331413485705524
+    },
+    {
+      "train_step": 9338,
+      "loss": 0.6080832470704385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6080832470704385
+    },
+    {
+      "train_step": 9339,
+      "loss": 0.6299856008320253,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6299856008320253
+    },
+    {
+      "train_step": 9340,
+      "loss": 0.5581996676963785,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5581996676963785
+    },
+    {
+      "train_step": 9341,
+      "loss": 0.43202867695667435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43202867695667435
+    },
+    {
+      "train_step": 9342,
+      "loss": 0.45088954220003163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45088954220003163
+    },
+    {
+      "train_step": 9343,
+      "loss": 0.5504955648327383,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5504955648327383
+    },
+    {
+      "train_step": 9344,
+      "loss": 0.6514288300694976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6514288300694976
+    },
+    {
+      "train_step": 9345,
+      "loss": 0.40574817906060656,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40574817906060656
+    },
+    {
+      "train_step": 9346,
+      "loss": 0.4379733846067929,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4379733846067929
+    },
+    {
+      "train_step": 9347,
+      "loss": 0.6253946784548554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6253946784548554
+    },
+    {
+      "train_step": 9348,
+      "loss": 0.6688410601134768,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6688410601134768
+    },
+    {
+      "train_step": 9349,
+      "loss": 0.4748625899687841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4748625899687841
+    },
+    {
+      "train_step": 9350,
+      "loss": 0.6076686310294471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6076686310294471
+    },
+    {
+      "train_step": 9351,
+      "loss": 0.4549233816987833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4549233816987833
+    },
+    {
+      "train_step": 9352,
+      "loss": 0.5394071059241888,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5394071059241888
+    },
+    {
+      "train_step": 9353,
+      "loss": 0.4511429984079543,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4511429984079543
+    },
+    {
+      "train_step": 9354,
+      "loss": 0.49494188463526906,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49494188463526906
+    },
+    {
+      "train_step": 9355,
+      "loss": 0.5584749091996399,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5584749091996399
+    },
+    {
+      "train_step": 9356,
+      "loss": 0.5605359217275021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5605359217275021
+    },
+    {
+      "train_step": 9357,
+      "loss": 0.5422004654235985,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5422004654235985
+    },
+    {
+      "train_step": 9358,
+      "loss": 0.6235655817703175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6235655817703175
+    },
+    {
+      "train_step": 9359,
+      "loss": 0.7684999686196415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7684999686196415
+    },
+    {
+      "train_step": 9360,
+      "loss": 0.6567043508010313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6567043508010313
+    },
+    {
+      "train_step": 9361,
+      "loss": 0.6037606740111486,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6037606740111486
+    },
+    {
+      "train_step": 9362,
+      "loss": 0.463239645764315,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.463239645764315
+    },
+    {
+      "train_step": 9363,
+      "loss": 0.7205426993868512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7205426993868512
+    },
+    {
+      "train_step": 9364,
+      "loss": 0.4866552098121927,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4866552098121927
+    },
+    {
+      "train_step": 9365,
+      "loss": 0.4412920780661996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4412920780661996
+    },
+    {
+      "train_step": 9366,
+      "loss": 0.6467343110996995,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6467343110996995
+    },
+    {
+      "train_step": 9367,
+      "loss": 0.7286338565166893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7286338565166893
+    },
+    {
+      "train_step": 9368,
+      "loss": 0.372805603443276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.372805603443276
+    },
+    {
+      "train_step": 9369,
+      "loss": 0.5441101650920953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5441101650920953
+    },
+    {
+      "train_step": 9370,
+      "loss": 0.5846104946470115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5846104946470115
+    },
+    {
+      "train_step": 9371,
+      "loss": 0.5437855677832499,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5437855677832499
+    },
+    {
+      "train_step": 9372,
+      "loss": 0.5549129777437676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5549129777437676
+    },
+    {
+      "train_step": 9373,
+      "loss": 0.49622766676053665,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49622766676053665
+    },
+    {
+      "train_step": 9374,
+      "loss": 0.3394236439250414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3394236439250414
+    },
+    {
+      "train_step": 9375,
+      "loss": 0.48613271748053705,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48613271748053705
+    },
+    {
+      "train_step": 9376,
+      "loss": 0.4362804232201265,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4362804232201265
+    },
+    {
+      "train_step": 9377,
+      "loss": 0.45154842152374786,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45154842152374786
+    },
+    {
+      "train_step": 9378,
+      "loss": 0.6615208476241817,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6615208476241817
+    },
+    {
+      "train_step": 9379,
+      "loss": 0.5107831821539001,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5107831821539001
+    },
+    {
+      "train_step": 9380,
+      "loss": 0.5667514289834339,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5667514289834339
+    },
+    {
+      "train_step": 9381,
+      "loss": 0.4706099691989648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4706099691989648
+    },
+    {
+      "train_step": 9382,
+      "loss": 0.4796946396862557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4796946396862557
+    },
+    {
+      "train_step": 9383,
+      "loss": 0.3230422814099245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3230422814099245
+    },
+    {
+      "train_step": 9384,
+      "loss": 0.6041809160429406,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6041809160429406
+    },
+    {
+      "train_step": 9385,
+      "loss": 0.5209517758273957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5209517758273957
+    },
+    {
+      "train_step": 9386,
+      "loss": 0.4646906961937024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4646906961937024
+    },
+    {
+      "train_step": 9387,
+      "loss": 0.6852223210908421,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6852223210908421
+    },
+    {
+      "train_step": 9388,
+      "loss": 0.4302892057868385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4302892057868385
+    },
+    {
+      "train_step": 9389,
+      "loss": 0.6359443829456372,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6359443829456372
+    },
+    {
+      "train_step": 9390,
+      "loss": 0.4210301613091387,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4210301613091387
+    },
+    {
+      "train_step": 9391,
+      "loss": 0.67475310903735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.67475310903735
+    },
+    {
+      "train_step": 9392,
+      "loss": 0.49767716842674153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49767716842674153
+    },
+    {
+      "train_step": 9393,
+      "loss": 0.6050327842936126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6050327842936126
+    },
+    {
+      "train_step": 9394,
+      "loss": 0.6126297533382976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6126297533382976
+    },
+    {
+      "train_step": 9395,
+      "loss": 0.6593795068423548,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6593795068423548
+    },
+    {
+      "train_step": 9396,
+      "loss": 0.8350218224938015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8350218224938015
+    },
+    {
+      "train_step": 9397,
+      "loss": 0.6480086626629442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6480086626629442
+    },
+    {
+      "train_step": 9398,
+      "loss": 0.6979602474447572,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6979602474447572
+    },
+    {
+      "train_step": 9399,
+      "loss": 0.6599807235681827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6599807235681827
+    },
+    {
+      "train_step": 9400,
+      "loss": 0.6627020485687134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6627020485687134
+    },
+    {
+      "train_step": 9401,
+      "loss": 0.6756467951278682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6756467951278682
+    },
+    {
+      "train_step": 9402,
+      "loss": 0.6893196177281243,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6893196177281243
+    },
+    {
+      "train_step": 9403,
+      "loss": 0.6167868261682342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6167868261682342
+    },
+    {
+      "train_step": 9404,
+      "loss": 0.5977252716395769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5977252716395769
+    },
+    {
+      "train_step": 9405,
+      "loss": 0.44534936181239565,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44534936181239565
+    },
+    {
+      "train_step": 9406,
+      "loss": 0.5023755114832986,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5023755114832986
+    },
+    {
+      "train_step": 9407,
+      "loss": 0.6491493360526077,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6491493360526077
+    },
+    {
+      "train_step": 9408,
+      "loss": 0.4706939920277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4706939920277
+    },
+    {
+      "train_step": 9409,
+      "loss": 0.52062402623494,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.52062402623494
+    },
+    {
+      "train_step": 9410,
+      "loss": 0.4061901077991882,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4061901077991882
+    },
+    {
+      "train_step": 9411,
+      "loss": 0.6576363242137059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6576363242137059
+    },
+    {
+      "train_step": 9412,
+      "loss": 0.6242768507877047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6242768507877047
+    },
+    {
+      "train_step": 9413,
+      "loss": 0.39463671865251726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39463671865251726
+    },
+    {
+      "train_step": 9414,
+      "loss": 0.42792177758677885,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42792177758677885
+    },
+    {
+      "train_step": 9415,
+      "loss": 0.6148051961526438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6148051961526438
+    },
+    {
+      "train_step": 9416,
+      "loss": 0.5168938131406968,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5168938131406968
+    },
+    {
+      "train_step": 9417,
+      "loss": 0.3938949396507245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3938949396507245
+    },
+    {
+      "train_step": 9418,
+      "loss": 0.5398387775538384,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5398387775538384
+    },
+    {
+      "train_step": 9419,
+      "loss": 0.6528328710935823,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6528328710935823
+    },
+    {
+      "train_step": 9420,
+      "loss": 0.550788024358092,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.550788024358092
+    },
+    {
+      "train_step": 9421,
+      "loss": 0.7252088366873819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7252088366873819
+    },
+    {
+      "train_step": 9422,
+      "loss": 0.5593329955090256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5593329955090256
+    },
+    {
+      "train_step": 9423,
+      "loss": 0.47870390434031423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47870390434031423
+    },
+    {
+      "train_step": 9424,
+      "loss": 0.33025427952915437,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33025427952915437
+    },
+    {
+      "train_step": 9425,
+      "loss": 0.6676016108650632,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6676016108650632
+    },
+    {
+      "train_step": 9426,
+      "loss": 0.6344187318476548,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6344187318476548
+    },
+    {
+      "train_step": 9427,
+      "loss": 0.48015760808391533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48015760808391533
+    },
+    {
+      "train_step": 9428,
+      "loss": 0.6570022015171407,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6570022015171407
+    },
+    {
+      "train_step": 9429,
+      "loss": 0.669249946605429,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.669249946605429
+    },
+    {
+      "train_step": 9430,
+      "loss": 0.7567014059129359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7567014059129359
+    },
+    {
+      "train_step": 9431,
+      "loss": 0.5514148507401522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5514148507401522
+    },
+    {
+      "train_step": 9432,
+      "loss": 0.41479367182788307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41479367182788307
+    },
+    {
+      "train_step": 9433,
+      "loss": 0.46322629739807286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46322629739807286
+    },
+    {
+      "train_step": 9434,
+      "loss": 0.534669049455899,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.534669049455899
+    },
+    {
+      "train_step": 9435,
+      "loss": 0.501419521937708,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.501419521937708
+    },
+    {
+      "train_step": 9436,
+      "loss": 0.5061231226418763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5061231226418763
+    },
+    {
+      "train_step": 9437,
+      "loss": 0.42811742293130395,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42811742293130395
+    },
+    {
+      "train_step": 9438,
+      "loss": 0.4541933457115097,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4541933457115097
+    },
+    {
+      "train_step": 9439,
+      "loss": 0.5000957110236903,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5000957110236903
+    },
+    {
+      "train_step": 9440,
+      "loss": 0.3925032007131919,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3925032007131919
+    },
+    {
+      "train_step": 9441,
+      "loss": 0.6347011860428907,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6347011860428907
+    },
+    {
+      "train_step": 9442,
+      "loss": 0.5179661947319348,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5179661947319348
+    },
+    {
+      "train_step": 9443,
+      "loss": 0.4922987055561459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4922987055561459
+    },
+    {
+      "train_step": 9444,
+      "loss": 0.5388694121008423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5388694121008423
+    },
+    {
+      "train_step": 9445,
+      "loss": 0.4275202480623879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4275202480623879
+    },
+    {
+      "train_step": 9446,
+      "loss": 0.5060368938644615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5060368938644615
+    },
+    {
+      "train_step": 9447,
+      "loss": 0.4690984166582059,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4690984166582059
+    },
+    {
+      "train_step": 9448,
+      "loss": 0.42478452084797275,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42478452084797275
+    },
+    {
+      "train_step": 9449,
+      "loss": 0.5533191206607556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5533191206607556
+    },
+    {
+      "train_step": 9450,
+      "loss": 0.4490990793831848,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4490990793831848
+    },
+    {
+      "train_step": 9451,
+      "loss": 0.4729562573534794,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4729562573534794
+    },
+    {
+      "train_step": 9452,
+      "loss": 0.6005129241337804,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6005129241337804
+    },
+    {
+      "train_step": 9453,
+      "loss": 0.5803335057290915,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5803335057290915
+    },
+    {
+      "train_step": 9454,
+      "loss": 0.5828056904668415,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5828056904668415
+    },
+    {
+      "train_step": 9455,
+      "loss": 0.45890319139023955,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45890319139023955
+    },
+    {
+      "train_step": 9456,
+      "loss": 0.34617481453714616,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34617481453714616
+    },
+    {
+      "train_step": 9457,
+      "loss": 0.3108802604104358,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3108802604104358
+    },
+    {
+      "train_step": 9458,
+      "loss": 0.34770916084913556,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34770916084913556
+    },
+    {
+      "train_step": 9459,
+      "loss": 0.6412119311837255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6412119311837255
+    },
+    {
+      "train_step": 9460,
+      "loss": 0.5012754878995157,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5012754878995157
+    },
+    {
+      "train_step": 9461,
+      "loss": 0.580385265082359,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.580385265082359
+    },
+    {
+      "train_step": 9462,
+      "loss": 0.7696052040620983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7696052040620983
+    },
+    {
+      "train_step": 9463,
+      "loss": 0.5570739528367742,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5570739528367742
+    },
+    {
+      "train_step": 9464,
+      "loss": 0.5699233998409223,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5699233998409223
+    },
+    {
+      "train_step": 9465,
+      "loss": 0.3689419623276909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3689419623276909
+    },
+    {
+      "train_step": 9466,
+      "loss": 0.7421904591217575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7421904591217575
+    },
+    {
+      "train_step": 9467,
+      "loss": 0.41692261577026574,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41692261577026574
+    },
+    {
+      "train_step": 9468,
+      "loss": 0.4765304753429222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4765304753429222
+    },
+    {
+      "train_step": 9469,
+      "loss": 0.43066891135154634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43066891135154634
+    },
+    {
+      "train_step": 9470,
+      "loss": 0.6418701130000671,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6418701130000671
+    },
+    {
+      "train_step": 9471,
+      "loss": 0.3371467660557842,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3371467660557842
+    },
+    {
+      "train_step": 9472,
+      "loss": 0.6019732235750176,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6019732235750176
+    },
+    {
+      "train_step": 9473,
+      "loss": 0.6585280985883118,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6585280985883118
+    },
+    {
+      "train_step": 9474,
+      "loss": 0.42279376048218004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42279376048218004
+    },
+    {
+      "train_step": 9475,
+      "loss": 0.5373773037433034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5373773037433034
+    },
+    {
+      "train_step": 9476,
+      "loss": 0.6877366178191126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6877366178191126
+    },
+    {
+      "train_step": 9477,
+      "loss": 0.511159472409219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.511159472409219
+    },
+    {
+      "train_step": 9478,
+      "loss": 0.6109120055502186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6109120055502186
+    },
+    {
+      "train_step": 9479,
+      "loss": 0.48665661852715925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48665661852715925
+    },
+    {
+      "train_step": 9480,
+      "loss": 0.46161727903773997,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46161727903773997
+    },
+    {
+      "train_step": 9481,
+      "loss": 0.5873128171554638,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5873128171554638
+    },
+    {
+      "train_step": 9482,
+      "loss": 0.4984836284286692,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4984836284286692
+    },
+    {
+      "train_step": 9483,
+      "loss": 0.3843982656928493,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3843982656928493
+    },
+    {
+      "train_step": 9484,
+      "loss": 0.5078468992780124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5078468992780124
+    },
+    {
+      "train_step": 9485,
+      "loss": 0.7972235981431428,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7972235981431428
+    },
+    {
+      "train_step": 9486,
+      "loss": 0.6060594744854167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6060594744854167
+    },
+    {
+      "train_step": 9487,
+      "loss": 0.4968733704229137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4968733704229137
+    },
+    {
+      "train_step": 9488,
+      "loss": 0.4029735457415806,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4029735457415806
+    },
+    {
+      "train_step": 9489,
+      "loss": 0.5119201746934542,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5119201746934542
+    },
+    {
+      "train_step": 9490,
+      "loss": 0.5550084150106305,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5550084150106305
+    },
+    {
+      "train_step": 9491,
+      "loss": 0.737395831250697,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.737395831250697
+    },
+    {
+      "train_step": 9492,
+      "loss": 0.5658774610458857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5658774610458857
+    },
+    {
+      "train_step": 9493,
+      "loss": 0.5046042927692819,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5046042927692819
+    },
+    {
+      "train_step": 9494,
+      "loss": 0.4188753157819627,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4188753157819627
+    },
+    {
+      "train_step": 9495,
+      "loss": 0.4970594302248523,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4970594302248523
+    },
+    {
+      "train_step": 9496,
+      "loss": 0.37874849991285875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37874849991285875
+    },
+    {
+      "train_step": 9497,
+      "loss": 0.6304090434623719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6304090434623719
+    },
+    {
+      "train_step": 9498,
+      "loss": 0.4425965274396938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4425965274396938
+    },
+    {
+      "train_step": 9499,
+      "loss": 0.43481477487553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43481477487553
+    },
+    {
+      "train_step": 9500,
+      "loss": 0.4695573875901337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4695573875901337
+    },
+    {
+      "train_step": 9501,
+      "loss": 0.4815159124167605,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4815159124167605
+    },
+    {
+      "train_step": 9502,
+      "loss": 0.7139784104083474,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7139784104083474
+    },
+    {
+      "train_step": 9503,
+      "loss": 0.6710184638926977,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6710184638926977
+    },
+    {
+      "train_step": 9504,
+      "loss": 0.5794752199267562,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5794752199267562
+    },
+    {
+      "train_step": 9505,
+      "loss": 0.5681970041748168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5681970041748168
+    },
+    {
+      "train_step": 9506,
+      "loss": 0.45157907661753477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45157907661753477
+    },
+    {
+      "train_step": 9507,
+      "loss": 0.43552152913742964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43552152913742964
+    },
+    {
+      "train_step": 9508,
+      "loss": 0.6007525918427615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6007525918427615
+    },
+    {
+      "train_step": 9509,
+      "loss": 0.5301131322230122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5301131322230122
+    },
+    {
+      "train_step": 9510,
+      "loss": 0.6152128486063938,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6152128486063938
+    },
+    {
+      "train_step": 9511,
+      "loss": 0.625704438409337,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.625704438409337
+    },
+    {
+      "train_step": 9512,
+      "loss": 0.507310762580806,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.507310762580806
+    },
+    {
+      "train_step": 9513,
+      "loss": 0.44203297432964256,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44203297432964256
+    },
+    {
+      "train_step": 9514,
+      "loss": 0.5128628500843465,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5128628500843465
+    },
+    {
+      "train_step": 9515,
+      "loss": 0.6537312740401958,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6537312740401958
+    },
+    {
+      "train_step": 9516,
+      "loss": 0.4110384635894587,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4110384635894587
+    },
+    {
+      "train_step": 9517,
+      "loss": 0.5399465918600849,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5399465918600849
+    },
+    {
+      "train_step": 9518,
+      "loss": 0.45672747599270386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45672747599270386
+    },
+    {
+      "train_step": 9519,
+      "loss": 0.6568021393115469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6568021393115469
+    },
+    {
+      "train_step": 9520,
+      "loss": 0.5186264586900812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5186264586900812
+    },
+    {
+      "train_step": 9521,
+      "loss": 0.43017903627370196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43017903627370196
+    },
+    {
+      "train_step": 9522,
+      "loss": 0.4793236797042106,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4793236797042106
+    },
+    {
+      "train_step": 9523,
+      "loss": 0.6354942676166749,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6354942676166749
+    },
+    {
+      "train_step": 9524,
+      "loss": 0.33116950548686364,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33116950548686364
+    },
+    {
+      "train_step": 9525,
+      "loss": 0.7601236775171036,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7601236775171036
+    },
+    {
+      "train_step": 9526,
+      "loss": 0.318226858323362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.318226858323362
+    },
+    {
+      "train_step": 9527,
+      "loss": 0.6164720077398714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6164720077398714
+    },
+    {
+      "train_step": 9528,
+      "loss": 0.6142479228135534,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6142479228135534
+    },
+    {
+      "train_step": 9529,
+      "loss": 0.3404460605359002,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3404460605359002
+    },
+    {
+      "train_step": 9530,
+      "loss": 0.519276589769456,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.519276589769456
+    },
+    {
+      "train_step": 9531,
+      "loss": 0.4895158870177076,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4895158870177076
+    },
+    {
+      "train_step": 9532,
+      "loss": 0.3828254331272439,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3828254331272439
+    },
+    {
+      "train_step": 9533,
+      "loss": 0.6333477235741024,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6333477235741024
+    },
+    {
+      "train_step": 9534,
+      "loss": 0.5443792379501879,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5443792379501879
+    },
+    {
+      "train_step": 9535,
+      "loss": 0.6299408808372067,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6299408808372067
+    },
+    {
+      "train_step": 9536,
+      "loss": 0.5748312842536795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5748312842536795
+    },
+    {
+      "train_step": 9537,
+      "loss": 0.5639223085500216,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5639223085500216
+    },
+    {
+      "train_step": 9538,
+      "loss": 0.5605755644090676,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5605755644090676
+    },
+    {
+      "train_step": 9539,
+      "loss": 0.5170714376738227,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5170714376738227
+    },
+    {
+      "train_step": 9540,
+      "loss": 0.5321988627594748,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5321988627594748
+    },
+    {
+      "train_step": 9541,
+      "loss": 0.6204939660686976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6204939660686976
+    },
+    {
+      "train_step": 9542,
+      "loss": 0.35370570067773943,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35370570067773943
+    },
+    {
+      "train_step": 9543,
+      "loss": 0.4544272634802944,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4544272634802944
+    },
+    {
+      "train_step": 9544,
+      "loss": 0.6077044419868411,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6077044419868411
+    },
+    {
+      "train_step": 9545,
+      "loss": 0.44256325934340834,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44256325934340834
+    },
+    {
+      "train_step": 9546,
+      "loss": 0.47093093206151604,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47093093206151604
+    },
+    {
+      "train_step": 9547,
+      "loss": 0.6773515017490951,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6773515017490951
+    },
+    {
+      "train_step": 9548,
+      "loss": 0.6630454993180471,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6630454993180471
+    },
+    {
+      "train_step": 9549,
+      "loss": 0.47784539637130496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47784539637130496
+    },
+    {
+      "train_step": 9550,
+      "loss": 0.4422783207026739,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4422783207026739
+    },
+    {
+      "train_step": 9551,
+      "loss": 0.4456864576646723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4456864576646723
+    },
+    {
+      "train_step": 9552,
+      "loss": 0.48879471592027635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48879471592027635
+    },
+    {
+      "train_step": 9553,
+      "loss": 0.4547654612548134,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4547654612548134
+    },
+    {
+      "train_step": 9554,
+      "loss": 0.6146945124543735,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6146945124543735
+    },
+    {
+      "train_step": 9555,
+      "loss": 0.47798173934579724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47798173934579724
+    },
+    {
+      "train_step": 9556,
+      "loss": 0.4802211560319174,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4802211560319174
+    },
+    {
+      "train_step": 9557,
+      "loss": 0.546646642741292,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.546646642741292
+    },
+    {
+      "train_step": 9558,
+      "loss": 0.5511967788658553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5511967788658553
+    },
+    {
+      "train_step": 9559,
+      "loss": 0.547819151356143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.547819151356143
+    },
+    {
+      "train_step": 9560,
+      "loss": 0.737407786851219,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.737407786851219
+    },
+    {
+      "train_step": 9561,
+      "loss": 0.5181631110962341,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5181631110962341
+    },
+    {
+      "train_step": 9562,
+      "loss": 0.8784050197114385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8784050197114385
+    },
+    {
+      "train_step": 9563,
+      "loss": 0.43819404959472047,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43819404959472047
+    },
+    {
+      "train_step": 9564,
+      "loss": 0.6145693655395094,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6145693655395094
+    },
+    {
+      "train_step": 9565,
+      "loss": 0.6012953247473448,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6012953247473448
+    },
+    {
+      "train_step": 9566,
+      "loss": 0.6337711563483535,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6337711563483535
+    },
+    {
+      "train_step": 9567,
+      "loss": 0.6654279465876114,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6654279465876114
+    },
+    {
+      "train_step": 9568,
+      "loss": 0.4731743969548382,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4731743969548382
+    },
+    {
+      "train_step": 9569,
+      "loss": 0.4916851211779718,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4916851211779718
+    },
+    {
+      "train_step": 9570,
+      "loss": 0.5681349670689215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5681349670689215
+    },
+    {
+      "train_step": 9571,
+      "loss": 0.4826155593841291,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4826155593841291
+    },
+    {
+      "train_step": 9572,
+      "loss": 0.5330459411159068,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5330459411159068
+    },
+    {
+      "train_step": 9573,
+      "loss": 0.4537122792723324,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4537122792723324
+    },
+    {
+      "train_step": 9574,
+      "loss": 0.6572579020466434,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6572579020466434
+    },
+    {
+      "train_step": 9575,
+      "loss": 0.46994247077690515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46994247077690515
+    },
+    {
+      "train_step": 9576,
+      "loss": 0.5749103482003347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5749103482003347
+    },
+    {
+      "train_step": 9577,
+      "loss": 0.5845719266517598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5845719266517598
+    },
+    {
+      "train_step": 9578,
+      "loss": 0.7325929709888326,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7325929709888326
+    },
+    {
+      "train_step": 9579,
+      "loss": 0.7075802255103311,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7075802255103311
+    },
+    {
+      "train_step": 9580,
+      "loss": 0.603787308221889,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.603787308221889
+    },
+    {
+      "train_step": 9581,
+      "loss": 0.4973963723036909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4973963723036909
+    },
+    {
+      "train_step": 9582,
+      "loss": 0.44924512272973793,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44924512272973793
+    },
+    {
+      "train_step": 9583,
+      "loss": 0.5091328212253045,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5091328212253045
+    },
+    {
+      "train_step": 9584,
+      "loss": 0.5512553738443501,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5512553738443501
+    },
+    {
+      "train_step": 9585,
+      "loss": 0.5264835644055414,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5264835644055414
+    },
+    {
+      "train_step": 9586,
+      "loss": 0.6134282719895177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6134282719895177
+    },
+    {
+      "train_step": 9587,
+      "loss": 0.520637163490909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.520637163490909
+    },
+    {
+      "train_step": 9588,
+      "loss": 0.3833477744762278,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3833477744762278
+    },
+    {
+      "train_step": 9589,
+      "loss": 0.5811445496261469,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5811445496261469
+    },
+    {
+      "train_step": 9590,
+      "loss": 0.48622660029782855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48622660029782855
+    },
+    {
+      "train_step": 9591,
+      "loss": 0.6070645973897832,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6070645973897832
+    },
+    {
+      "train_step": 9592,
+      "loss": 0.34545650274817613,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34545650274817613
+    },
+    {
+      "train_step": 9593,
+      "loss": 0.600509489519111,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.600509489519111
+    },
+    {
+      "train_step": 9594,
+      "loss": 0.6564383342505444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6564383342505444
+    },
+    {
+      "train_step": 9595,
+      "loss": 0.4866320225231229,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4866320225231229
+    },
+    {
+      "train_step": 9596,
+      "loss": 0.5289583900847987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5289583900847987
+    },
+    {
+      "train_step": 9597,
+      "loss": 0.5264863374111343,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5264863374111343
+    },
+    {
+      "train_step": 9598,
+      "loss": 0.6080125142679006,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6080125142679006
+    },
+    {
+      "train_step": 9599,
+      "loss": 0.4914654990502282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4914654990502282
+    },
+    {
+      "train_step": 9600,
+      "loss": 0.7289150369729281,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7289150369729281
+    },
+    {
+      "train_step": 9601,
+      "loss": 0.4905708688527818,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4905708688527818
+    },
+    {
+      "train_step": 9602,
+      "loss": 0.5464556153719533,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5464556153719533
+    },
+    {
+      "train_step": 9603,
+      "loss": 0.6649817017387388,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6649817017387388
+    },
+    {
+      "train_step": 9604,
+      "loss": 0.5982828777498619,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5982828777498619
+    },
+    {
+      "train_step": 9605,
+      "loss": 0.453100384929629,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.453100384929629
+    },
+    {
+      "train_step": 9606,
+      "loss": 0.42039417066524726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42039417066524726
+    },
+    {
+      "train_step": 9607,
+      "loss": 0.504522249947477,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.504522249947477
+    },
+    {
+      "train_step": 9608,
+      "loss": 0.4071731732475421,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4071731732475421
+    },
+    {
+      "train_step": 9609,
+      "loss": 0.7543337170311126,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7543337170311126
+    },
+    {
+      "train_step": 9610,
+      "loss": 0.42107576913399875,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42107576913399875
+    },
+    {
+      "train_step": 9611,
+      "loss": 0.571502268803213,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.571502268803213
+    },
+    {
+      "train_step": 9612,
+      "loss": 0.5194862163312699,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5194862163312699
+    },
+    {
+      "train_step": 9613,
+      "loss": 0.44211238063620484,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44211238063620484
+    },
+    {
+      "train_step": 9614,
+      "loss": 0.6440491855686873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6440491855686873
+    },
+    {
+      "train_step": 9615,
+      "loss": 0.5830036661770588,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5830036661770588
+    },
+    {
+      "train_step": 9616,
+      "loss": 0.6665422026548267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6665422026548267
+    },
+    {
+      "train_step": 9617,
+      "loss": 0.6116232474792164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6116232474792164
+    },
+    {
+      "train_step": 9618,
+      "loss": 0.3900553994002405,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3900553994002405
+    },
+    {
+      "train_step": 9619,
+      "loss": 0.7726980334929195,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7726980334929195
+    },
+    {
+      "train_step": 9620,
+      "loss": 0.3096032107208285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3096032107208285
+    },
+    {
+      "train_step": 9621,
+      "loss": 0.5794416283267594,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5794416283267594
+    },
+    {
+      "train_step": 9622,
+      "loss": 0.49221329433992544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49221329433992544
+    },
+    {
+      "train_step": 9623,
+      "loss": 0.5827713838009864,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5827713838009864
+    },
+    {
+      "train_step": 9624,
+      "loss": 0.7043753014613984,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7043753014613984
+    },
+    {
+      "train_step": 9625,
+      "loss": 0.3898667318673598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3898667318673598
+    },
+    {
+      "train_step": 9626,
+      "loss": 0.5103747759597708,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5103747759597708
+    },
+    {
+      "train_step": 9627,
+      "loss": 0.361171665747628,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.361171665747628
+    },
+    {
+      "train_step": 9628,
+      "loss": 0.6188302823546145,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6188302823546145
+    },
+    {
+      "train_step": 9629,
+      "loss": 0.6072407174515853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6072407174515853
+    },
+    {
+      "train_step": 9630,
+      "loss": 0.5189381865239601,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5189381865239601
+    },
+    {
+      "train_step": 9631,
+      "loss": 0.574560610547276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.574560610547276
+    },
+    {
+      "train_step": 9632,
+      "loss": 0.5001196819527893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5001196819527893
+    },
+    {
+      "train_step": 9633,
+      "loss": 0.3614655461419603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3614655461419603
+    },
+    {
+      "train_step": 9634,
+      "loss": 0.5681461828503525,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5681461828503525
+    },
+    {
+      "train_step": 9635,
+      "loss": 0.4938428110176645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4938428110176645
+    },
+    {
+      "train_step": 9636,
+      "loss": 0.5687414170270914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5687414170270914
+    },
+    {
+      "train_step": 9637,
+      "loss": 0.3769624437780099,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3769624437780099
+    },
+    {
+      "train_step": 9638,
+      "loss": 0.5761765593846151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5761765593846151
+    },
+    {
+      "train_step": 9639,
+      "loss": 0.5363649954617815,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5363649954617815
+    },
+    {
+      "train_step": 9640,
+      "loss": 0.3678359677982847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3678359677982847
+    },
+    {
+      "train_step": 9641,
+      "loss": 0.6812759170673,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6812759170673
+    },
+    {
+      "train_step": 9642,
+      "loss": 0.7741953597896853,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7741953597896853
+    },
+    {
+      "train_step": 9643,
+      "loss": 0.7227135862673487,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7227135862673487
+    },
+    {
+      "train_step": 9644,
+      "loss": 0.44606725813271514,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.44606725813271514
+    },
+    {
+      "train_step": 9645,
+      "loss": 0.6429886665347802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6429886665347802
+    },
+    {
+      "train_step": 9646,
+      "loss": 0.48593108916782934,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48593108916782934
+    },
+    {
+      "train_step": 9647,
+      "loss": 0.5131325842823488,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5131325842823488
+    },
+    {
+      "train_step": 9648,
+      "loss": 0.6177581672207951,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6177581672207951
+    },
+    {
+      "train_step": 9649,
+      "loss": 0.3419373906132992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3419373906132992
+    },
+    {
+      "train_step": 9650,
+      "loss": 0.6330345415159472,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6330345415159472
+    },
+    {
+      "train_step": 9651,
+      "loss": 0.5927487058290545,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5927487058290545
+    },
+    {
+      "train_step": 9652,
+      "loss": 0.8064365209595591,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8064365209595591
+    },
+    {
+      "train_step": 9653,
+      "loss": 0.7760434826793193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7760434826793193
+    },
+    {
+      "train_step": 9654,
+      "loss": 0.5503434097790683,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5503434097790683
+    },
+    {
+      "train_step": 9655,
+      "loss": 0.4584157836306212,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4584157836306212
+    },
+    {
+      "train_step": 9656,
+      "loss": 0.3987474189075757,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3987474189075757
+    },
+    {
+      "train_step": 9657,
+      "loss": 0.4633369509646674,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4633369509646674
+    },
+    {
+      "train_step": 9658,
+      "loss": 0.6331875710663987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6331875710663987
+    },
+    {
+      "train_step": 9659,
+      "loss": 0.5420439322736083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5420439322736083
+    },
+    {
+      "train_step": 9660,
+      "loss": 0.5635864103120599,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5635864103120599
+    },
+    {
+      "train_step": 9661,
+      "loss": 0.4332756114594479,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4332756114594479
+    },
+    {
+      "train_step": 9662,
+      "loss": 0.4226845425362151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4226845425362151
+    },
+    {
+      "train_step": 9663,
+      "loss": 0.5174868641461005,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5174868641461005
+    },
+    {
+      "train_step": 9664,
+      "loss": 0.5088678590038431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5088678590038431
+    },
+    {
+      "train_step": 9665,
+      "loss": 0.48115562120951844,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48115562120951844
+    },
+    {
+      "train_step": 9666,
+      "loss": 0.5632601947804483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5632601947804483
+    },
+    {
+      "train_step": 9667,
+      "loss": 0.4695566249971963,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4695566249971963
+    },
+    {
+      "train_step": 9668,
+      "loss": 0.23434454253759252,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.23434454253759252
+    },
+    {
+      "train_step": 9669,
+      "loss": 0.5873013146267597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5873013146267597
+    },
+    {
+      "train_step": 9670,
+      "loss": 0.5411395428978858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5411395428978858
+    },
+    {
+      "train_step": 9671,
+      "loss": 0.5495086533121648,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5495086533121648
+    },
+    {
+      "train_step": 9672,
+      "loss": 0.48346346970218107,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48346346970218107
+    },
+    {
+      "train_step": 9673,
+      "loss": 0.5394182238987397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5394182238987397
+    },
+    {
+      "train_step": 9674,
+      "loss": 0.6305133394626211,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6305133394626211
+    },
+    {
+      "train_step": 9675,
+      "loss": 0.8818919198763017,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8818919198763017
+    },
+    {
+      "train_step": 9676,
+      "loss": 0.35811534680043755,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35811534680043755
+    },
+    {
+      "train_step": 9677,
+      "loss": 0.643223894147466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.643223894147466
+    },
+    {
+      "train_step": 9678,
+      "loss": 0.55397437109932,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.55397437109932
+    },
+    {
+      "train_step": 9679,
+      "loss": 0.6433372661093083,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6433372661093083
+    },
+    {
+      "train_step": 9680,
+      "loss": 0.6819852135519459,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6819852135519459
+    },
+    {
+      "train_step": 9681,
+      "loss": 0.6051048222758423,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6051048222758423
+    },
+    {
+      "train_step": 9682,
+      "loss": 0.6460997215711247,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6460997215711247
+    },
+    {
+      "train_step": 9683,
+      "loss": 0.5153794625172319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5153794625172319
+    },
+    {
+      "train_step": 9684,
+      "loss": 0.5453184895326563,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5453184895326563
+    },
+    {
+      "train_step": 9685,
+      "loss": 0.6451773593728954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6451773593728954
+    },
+    {
+      "train_step": 9686,
+      "loss": 0.43600238859750245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43600238859750245
+    },
+    {
+      "train_step": 9687,
+      "loss": 0.3178518103153149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3178518103153149
+    },
+    {
+      "train_step": 9688,
+      "loss": 0.39518027896556435,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39518027896556435
+    },
+    {
+      "train_step": 9689,
+      "loss": 0.5832569522221783,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5832569522221783
+    },
+    {
+      "train_step": 9690,
+      "loss": 0.4311952216535544,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4311952216535544
+    },
+    {
+      "train_step": 9691,
+      "loss": 0.5696238204458267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5696238204458267
+    },
+    {
+      "train_step": 9692,
+      "loss": 0.4616338314936304,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4616338314936304
+    },
+    {
+      "train_step": 9693,
+      "loss": 0.478188669445222,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.478188669445222
+    },
+    {
+      "train_step": 9694,
+      "loss": 0.6985732033749668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6985732033749668
+    },
+    {
+      "train_step": 9695,
+      "loss": 0.46881553685130983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46881553685130983
+    },
+    {
+      "train_step": 9696,
+      "loss": 0.42427605495080206,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42427605495080206
+    },
+    {
+      "train_step": 9697,
+      "loss": 0.6604820710619168,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6604820710619168
+    },
+    {
+      "train_step": 9698,
+      "loss": 0.4581423325029481,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4581423325029481
+    },
+    {
+      "train_step": 9699,
+      "loss": 0.49281411403406283,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49281411403406283
+    },
+    {
+      "train_step": 9700,
+      "loss": 0.8857229948137386,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8857229948137386
+    },
+    {
+      "train_step": 9701,
+      "loss": 0.46031891695889204,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46031891695889204
+    },
+    {
+      "train_step": 9702,
+      "loss": 0.48718635304307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48718635304307
+    },
+    {
+      "train_step": 9703,
+      "loss": 0.4637707342706247,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4637707342706247
+    },
+    {
+      "train_step": 9704,
+      "loss": 0.7328434573282531,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7328434573282531
+    },
+    {
+      "train_step": 9705,
+      "loss": 0.4055688807005466,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4055688807005466
+    },
+    {
+      "train_step": 9706,
+      "loss": 0.3999397605416196,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3999397605416196
+    },
+    {
+      "train_step": 9707,
+      "loss": 0.4664275072978933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4664275072978933
+    },
+    {
+      "train_step": 9708,
+      "loss": 0.7238720314852202,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7238720314852202
+    },
+    {
+      "train_step": 9709,
+      "loss": 0.5310302973178012,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5310302973178012
+    },
+    {
+      "train_step": 9710,
+      "loss": 0.6629821483261442,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6629821483261442
+    },
+    {
+      "train_step": 9711,
+      "loss": 0.5197647785441549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5197647785441549
+    },
+    {
+      "train_step": 9712,
+      "loss": 0.6652846383980779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6652846383980779
+    },
+    {
+      "train_step": 9713,
+      "loss": 0.5732832094816668,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5732832094816668
+    },
+    {
+      "train_step": 9714,
+      "loss": 0.5474433067094515,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5474433067094515
+    },
+    {
+      "train_step": 9715,
+      "loss": 0.5948477691537992,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5948477691537992
+    },
+    {
+      "train_step": 9716,
+      "loss": 0.4300096621210554,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4300096621210554
+    },
+    {
+      "train_step": 9717,
+      "loss": 0.5732181478383438,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5732181478383438
+    },
+    {
+      "train_step": 9718,
+      "loss": 0.6406350479994999,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6406350479994999
+    },
+    {
+      "train_step": 9719,
+      "loss": 0.3847160472436761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3847160472436761
+    },
+    {
+      "train_step": 9720,
+      "loss": 0.5604122912325987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5604122912325987
+    },
+    {
+      "train_step": 9721,
+      "loss": 0.5851445930095346,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5851445930095346
+    },
+    {
+      "train_step": 9722,
+      "loss": 0.48274222878392714,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48274222878392714
+    },
+    {
+      "train_step": 9723,
+      "loss": 0.6274527477518654,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6274527477518654
+    },
+    {
+      "train_step": 9724,
+      "loss": 0.5371000116479052,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5371000116479052
+    },
+    {
+      "train_step": 9725,
+      "loss": 0.557934285814598,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.557934285814598
+    },
+    {
+      "train_step": 9726,
+      "loss": 0.5492311407127349,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5492311407127349
+    },
+    {
+      "train_step": 9727,
+      "loss": 0.5263459883512287,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5263459883512287
+    },
+    {
+      "train_step": 9728,
+      "loss": 0.6925927159813677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6925927159813677
+    },
+    {
+      "train_step": 9729,
+      "loss": 0.5582247323804307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5582247323804307
+    },
+    {
+      "train_step": 9730,
+      "loss": 0.4820738621709445,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4820738621709445
+    },
+    {
+      "train_step": 9731,
+      "loss": 0.49335212693055835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49335212693055835
+    },
+    {
+      "train_step": 9732,
+      "loss": 0.5250002603579893,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5250002603579893
+    },
+    {
+      "train_step": 9733,
+      "loss": 0.6129008404117634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6129008404117634
+    },
+    {
+      "train_step": 9734,
+      "loss": 0.6144948641705557,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6144948641705557
+    },
+    {
+      "train_step": 9735,
+      "loss": 0.577712068345814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.577712068345814
+    },
+    {
+      "train_step": 9736,
+      "loss": 0.5646480625222189,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5646480625222189
+    },
+    {
+      "train_step": 9737,
+      "loss": 0.5495891347212677,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5495891347212677
+    },
+    {
+      "train_step": 9738,
+      "loss": 0.6047302965925517,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6047302965925517
+    },
+    {
+      "train_step": 9739,
+      "loss": 0.39491400661638365,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39491400661638365
+    },
+    {
+      "train_step": 9740,
+      "loss": 0.8517224432171483,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8517224432171483
+    },
+    {
+      "train_step": 9741,
+      "loss": 0.6212066782813379,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6212066782813379
+    },
+    {
+      "train_step": 9742,
+      "loss": 0.54774567004723,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.54774567004723
+    },
+    {
+      "train_step": 9743,
+      "loss": 0.4918383418675485,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4918383418675485
+    },
+    {
+      "train_step": 9744,
+      "loss": 0.7196204534048761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7196204534048761
+    },
+    {
+      "train_step": 9745,
+      "loss": 0.7238074917341115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7238074917341115
+    },
+    {
+      "train_step": 9746,
+      "loss": 0.751962508568873,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.751962508568873
+    },
+    {
+      "train_step": 9747,
+      "loss": 0.5244183290158282,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5244183290158282
+    },
+    {
+      "train_step": 9748,
+      "loss": 0.41470445759269603,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41470445759269603
+    },
+    {
+      "train_step": 9749,
+      "loss": 0.6232485285840884,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6232485285840884
+    },
+    {
+      "train_step": 9750,
+      "loss": 0.6763773781870137,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6763773781870137
+    },
+    {
+      "train_step": 9751,
+      "loss": 0.544170210322575,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.544170210322575
+    },
+    {
+      "train_step": 9752,
+      "loss": 0.5984072508551015,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5984072508551015
+    },
+    {
+      "train_step": 9753,
+      "loss": 0.5176699880171182,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5176699880171182
+    },
+    {
+      "train_step": 9754,
+      "loss": 0.6337606023907234,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6337606023907234
+    },
+    {
+      "train_step": 9755,
+      "loss": 0.6800954594140636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6800954594140636
+    },
+    {
+      "train_step": 9756,
+      "loss": 0.684334280446398,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.684334280446398
+    },
+    {
+      "train_step": 9757,
+      "loss": 0.536165777800635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.536165777800635
+    },
+    {
+      "train_step": 9758,
+      "loss": 0.6017699045216639,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6017699045216639
+    },
+    {
+      "train_step": 9759,
+      "loss": 0.5915357384143011,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5915357384143011
+    },
+    {
+      "train_step": 9760,
+      "loss": 0.6862859483438066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6862859483438066
+    },
+    {
+      "train_step": 9761,
+      "loss": 0.5794964187755847,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5794964187755847
+    },
+    {
+      "train_step": 9762,
+      "loss": 0.4802650736365444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4802650736365444
+    },
+    {
+      "train_step": 9763,
+      "loss": 0.5587633240328267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5587633240328267
+    },
+    {
+      "train_step": 9764,
+      "loss": 0.6153458833540397,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6153458833540397
+    },
+    {
+      "train_step": 9765,
+      "loss": 0.7815267455922447,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7815267455922447
+    },
+    {
+      "train_step": 9766,
+      "loss": 0.7231304811907802,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7231304811907802
+    },
+    {
+      "train_step": 9767,
+      "loss": 0.555760069134855,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.555760069134855
+    },
+    {
+      "train_step": 9768,
+      "loss": 0.43132738177875996,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.43132738177875996
+    },
+    {
+      "train_step": 9769,
+      "loss": 0.5339095892323062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5339095892323062
+    },
+    {
+      "train_step": 9770,
+      "loss": 0.5090949532059983,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5090949532059983
+    },
+    {
+      "train_step": 9771,
+      "loss": 0.5610529352121947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5610529352121947
+    },
+    {
+      "train_step": 9772,
+      "loss": 0.5163274157862512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5163274157862512
+    },
+    {
+      "train_step": 9773,
+      "loss": 0.5167719852880436,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5167719852880436
+    },
+    {
+      "train_step": 9774,
+      "loss": 0.5546579897256821,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5546579897256821
+    },
+    {
+      "train_step": 9775,
+      "loss": 0.3351787759194902,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3351787759194902
+    },
+    {
+      "train_step": 9776,
+      "loss": 0.5673961097569841,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5673961097569841
+    },
+    {
+      "train_step": 9777,
+      "loss": 0.7414623420782285,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7414623420782285
+    },
+    {
+      "train_step": 9778,
+      "loss": 0.6432263282932336,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6432263282932336
+    },
+    {
+      "train_step": 9779,
+      "loss": 0.5294979240926385,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5294979240926385
+    },
+    {
+      "train_step": 9780,
+      "loss": 0.5962799509449901,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5962799509449901
+    },
+    {
+      "train_step": 9781,
+      "loss": 0.6763875818606584,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6763875818606584
+    },
+    {
+      "train_step": 9782,
+      "loss": 0.5128498623486142,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5128498623486142
+    },
+    {
+      "train_step": 9783,
+      "loss": 0.6135222776860823,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6135222776860823
+    },
+    {
+      "train_step": 9784,
+      "loss": 0.6027089841656539,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6027089841656539
+    },
+    {
+      "train_step": 9785,
+      "loss": 0.5439593136141496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5439593136141496
+    },
+    {
+      "train_step": 9786,
+      "loss": 0.6419995720142964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6419995720142964
+    },
+    {
+      "train_step": 9787,
+      "loss": 0.6998679191263034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6998679191263034
+    },
+    {
+      "train_step": 9788,
+      "loss": 0.5215327428754039,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5215327428754039
+    },
+    {
+      "train_step": 9789,
+      "loss": 0.4341017199369956,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4341017199369956
+    },
+    {
+      "train_step": 9790,
+      "loss": 0.5484057672461286,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5484057672461286
+    },
+    {
+      "train_step": 9791,
+      "loss": 0.607752412674342,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.607752412674342
+    },
+    {
+      "train_step": 9792,
+      "loss": 0.739230484340637,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.739230484340637
+    },
+    {
+      "train_step": 9793,
+      "loss": 0.454168896008165,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.454168896008165
+    },
+    {
+      "train_step": 9794,
+      "loss": 0.5085853076586347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5085853076586347
+    },
+    {
+      "train_step": 9795,
+      "loss": 0.7647780743204169,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7647780743204169
+    },
+    {
+      "train_step": 9796,
+      "loss": 0.49089536738954276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49089536738954276
+    },
+    {
+      "train_step": 9797,
+      "loss": 0.3529458604184151,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3529458604184151
+    },
+    {
+      "train_step": 9798,
+      "loss": 0.6294367584804271,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6294367584804271
+    },
+    {
+      "train_step": 9799,
+      "loss": 0.5693121122827164,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5693121122827164
+    },
+    {
+      "train_step": 9800,
+      "loss": 0.7372943112302264,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7372943112302264
+    },
+    {
+      "train_step": 9801,
+      "loss": 0.5534506043698181,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5534506043698181
+    },
+    {
+      "train_step": 9802,
+      "loss": 0.6019819485543056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6019819485543056
+    },
+    {
+      "train_step": 9803,
+      "loss": 0.506011032222924,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.506011032222924
+    },
+    {
+      "train_step": 9804,
+      "loss": 0.4168050199416055,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4168050199416055
+    },
+    {
+      "train_step": 9805,
+      "loss": 0.829550890107031,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.829550890107031
+    },
+    {
+      "train_step": 9806,
+      "loss": 0.6674641569424473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6674641569424473
+    },
+    {
+      "train_step": 9807,
+      "loss": 0.4183204474990724,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4183204474990724
+    },
+    {
+      "train_step": 9808,
+      "loss": 0.5934727506948482,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5934727506948482
+    },
+    {
+      "train_step": 9809,
+      "loss": 0.5309439336178245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5309439336178245
+    },
+    {
+      "train_step": 9810,
+      "loss": 0.4620738239153954,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4620738239153954
+    },
+    {
+      "train_step": 9811,
+      "loss": 0.6980870634386298,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6980870634386298
+    },
+    {
+      "train_step": 9812,
+      "loss": 0.47150606946373347,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47150606946373347
+    },
+    {
+      "train_step": 9813,
+      "loss": 0.4999535793884998,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4999535793884998
+    },
+    {
+      "train_step": 9814,
+      "loss": 0.5527357310865043,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5527357310865043
+    },
+    {
+      "train_step": 9815,
+      "loss": 0.501735024462836,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.501735024462836
+    },
+    {
+      "train_step": 9816,
+      "loss": 0.5753515366115503,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5753515366115503
+    },
+    {
+      "train_step": 9817,
+      "loss": 0.639707473490947,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.639707473490947
+    },
+    {
+      "train_step": 9818,
+      "loss": 0.5346045780511652,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5346045780511652
+    },
+    {
+      "train_step": 9819,
+      "loss": 0.46736010361380004,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46736010361380004
+    },
+    {
+      "train_step": 9820,
+      "loss": 0.759194337212362,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.759194337212362
+    },
+    {
+      "train_step": 9821,
+      "loss": 0.5217999674089056,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5217999674089056
+    },
+    {
+      "train_step": 9822,
+      "loss": 0.5039341875044191,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5039341875044191
+    },
+    {
+      "train_step": 9823,
+      "loss": 0.5712094345690215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5712094345690215
+    },
+    {
+      "train_step": 9824,
+      "loss": 0.709556288213914,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.709556288213914
+    },
+    {
+      "train_step": 9825,
+      "loss": 0.41008296446642634,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41008296446642634
+    },
+    {
+      "train_step": 9826,
+      "loss": 0.492588175822263,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.492588175822263
+    },
+    {
+      "train_step": 9827,
+      "loss": 0.6473624397306389,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6473624397306389
+    },
+    {
+      "train_step": 9828,
+      "loss": 0.5504070640723909,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5504070640723909
+    },
+    {
+      "train_step": 9829,
+      "loss": 0.47880411166120246,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47880411166120246
+    },
+    {
+      "train_step": 9830,
+      "loss": 0.4599137072987942,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4599137072987942
+    },
+    {
+      "train_step": 9831,
+      "loss": 0.6768104420646621,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6768104420646621
+    },
+    {
+      "train_step": 9832,
+      "loss": 0.529987725450115,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.529987725450115
+    },
+    {
+      "train_step": 9833,
+      "loss": 0.5123789647768084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5123789647768084
+    },
+    {
+      "train_step": 9834,
+      "loss": 0.6067705676629833,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6067705676629833
+    },
+    {
+      "train_step": 9835,
+      "loss": 0.5946637465937861,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5946637465937861
+    },
+    {
+      "train_step": 9836,
+      "loss": 0.5456759048673179,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5456759048673179
+    },
+    {
+      "train_step": 9837,
+      "loss": 0.3968690964650762,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3968690964650762
+    },
+    {
+      "train_step": 9838,
+      "loss": 0.60205641201121,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.60205641201121
+    },
+    {
+      "train_step": 9839,
+      "loss": 0.394174831600857,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.394174831600857
+    },
+    {
+      "train_step": 9840,
+      "loss": 0.47906709868191255,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47906709868191255
+    },
+    {
+      "train_step": 9841,
+      "loss": 0.42761649580818156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42761649580818156
+    },
+    {
+      "train_step": 9842,
+      "loss": 0.5961939066447827,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5961939066447827
+    },
+    {
+      "train_step": 9843,
+      "loss": 0.472692283854908,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.472692283854908
+    },
+    {
+      "train_step": 9844,
+      "loss": 0.45423759420904175,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.45423759420904175
+    },
+    {
+      "train_step": 9845,
+      "loss": 0.8973613462166153,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8973613462166153
+    },
+    {
+      "train_step": 9846,
+      "loss": 0.3980750019538156,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3980750019538156
+    },
+    {
+      "train_step": 9847,
+      "loss": 0.6199998237723592,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6199998237723592
+    },
+    {
+      "train_step": 9848,
+      "loss": 0.5060272953710122,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5060272953710122
+    },
+    {
+      "train_step": 9849,
+      "loss": 0.6187989070713033,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6187989070713033
+    },
+    {
+      "train_step": 9850,
+      "loss": 0.6791782786865351,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6791782786865351
+    },
+    {
+      "train_step": 9851,
+      "loss": 0.5363557610561307,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5363557610561307
+    },
+    {
+      "train_step": 9852,
+      "loss": 0.6796184144351084,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6796184144351084
+    },
+    {
+      "train_step": 9853,
+      "loss": 0.3868495840507375,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3868495840507375
+    },
+    {
+      "train_step": 9854,
+      "loss": 0.47441266952664707,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47441266952664707
+    },
+    {
+      "train_step": 9855,
+      "loss": 0.5252250674983473,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5252250674983473
+    },
+    {
+      "train_step": 9856,
+      "loss": 0.5423587843907058,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5423587843907058
+    },
+    {
+      "train_step": 9857,
+      "loss": 0.5846828720333193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5846828720333193
+    },
+    {
+      "train_step": 9858,
+      "loss": 0.39172858885043693,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39172858885043693
+    },
+    {
+      "train_step": 9859,
+      "loss": 0.5758856345243716,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5758856345243716
+    },
+    {
+      "train_step": 9860,
+      "loss": 0.42769067383383874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.42769067383383874
+    },
+    {
+      "train_step": 9861,
+      "loss": 0.6965994890075553,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6965994890075553
+    },
+    {
+      "train_step": 9862,
+      "loss": 0.6935348690138751,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6935348690138751
+    },
+    {
+      "train_step": 9863,
+      "loss": 0.6060438087084976,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6060438087084976
+    },
+    {
+      "train_step": 9864,
+      "loss": 0.6360811693864431,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6360811693864431
+    },
+    {
+      "train_step": 9865,
+      "loss": 0.47810458339467443,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47810458339467443
+    },
+    {
+      "train_step": 9866,
+      "loss": 0.37678907212448953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.37678907212448953
+    },
+    {
+      "train_step": 9867,
+      "loss": 0.5346469717972335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5346469717972335
+    },
+    {
+      "train_step": 9868,
+      "loss": 0.8579779712580625,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8579779712580625
+    },
+    {
+      "train_step": 9869,
+      "loss": 0.5858368510377491,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5858368510377491
+    },
+    {
+      "train_step": 9870,
+      "loss": 0.6322921330607706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6322921330607706
+    },
+    {
+      "train_step": 9871,
+      "loss": 0.49790840470079856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49790840470079856
+    },
+    {
+      "train_step": 9872,
+      "loss": 0.4982609331835711,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4982609331835711
+    },
+    {
+      "train_step": 9873,
+      "loss": 0.496061072340193,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.496061072340193
+    },
+    {
+      "train_step": 9874,
+      "loss": 0.46028024855519933,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.46028024855519933
+    },
+    {
+      "train_step": 9875,
+      "loss": 0.3867583353862319,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3867583353862319
+    },
+    {
+      "train_step": 9876,
+      "loss": 0.5391581724538822,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5391581724538822
+    },
+    {
+      "train_step": 9877,
+      "loss": 0.6563898758916444,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6563898758916444
+    },
+    {
+      "train_step": 9878,
+      "loss": 0.6666802539041858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6666802539041858
+    },
+    {
+      "train_step": 9879,
+      "loss": 0.5458609355487276,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5458609355487276
+    },
+    {
+      "train_step": 9880,
+      "loss": 0.6818846519496788,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6818846519496788
+    },
+    {
+      "train_step": 9881,
+      "loss": 0.49904630497327335,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49904630497327335
+    },
+    {
+      "train_step": 9882,
+      "loss": 0.571534277873925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.571534277873925
+    },
+    {
+      "train_step": 9883,
+      "loss": 0.564504049671404,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.564504049671404
+    },
+    {
+      "train_step": 9884,
+      "loss": 0.7232400310617635,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7232400310617635
+    },
+    {
+      "train_step": 9885,
+      "loss": 0.4396869240866146,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4396869240866146
+    },
+    {
+      "train_step": 9886,
+      "loss": 0.5305393375760601,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5305393375760601
+    },
+    {
+      "train_step": 9887,
+      "loss": 0.4448932378207063,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4448932378207063
+    },
+    {
+      "train_step": 9888,
+      "loss": 0.5946797963429787,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5946797963429787
+    },
+    {
+      "train_step": 9889,
+      "loss": 0.4039306852453096,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4039306852453096
+    },
+    {
+      "train_step": 9890,
+      "loss": 0.33210022942863143,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.33210022942863143
+    },
+    {
+      "train_step": 9891,
+      "loss": 0.6437794971324694,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6437794971324694
+    },
+    {
+      "train_step": 9892,
+      "loss": 0.4883180325075964,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4883180325075964
+    },
+    {
+      "train_step": 9893,
+      "loss": 0.6724775696363925,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6724775696363925
+    },
+    {
+      "train_step": 9894,
+      "loss": 0.4983185118784812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4983185118784812
+    },
+    {
+      "train_step": 9895,
+      "loss": 0.5952839845693977,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5952839845693977
+    },
+    {
+      "train_step": 9896,
+      "loss": 0.6270368819929262,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6270368819929262
+    },
+    {
+      "train_step": 9897,
+      "loss": 0.34314986875090814,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34314986875090814
+    },
+    {
+      "train_step": 9898,
+      "loss": 0.3666018462334066,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3666018462334066
+    },
+    {
+      "train_step": 9899,
+      "loss": 0.5158995725835643,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5158995725835643
+    },
+    {
+      "train_step": 9900,
+      "loss": 0.4112882342921795,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4112882342921795
+    },
+    {
+      "train_step": 9901,
+      "loss": 0.8365581404471041,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8365581404471041
+    },
+    {
+      "train_step": 9902,
+      "loss": 0.5178930878945177,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5178930878945177
+    },
+    {
+      "train_step": 9903,
+      "loss": 0.3322647216156835,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3322647216156835
+    },
+    {
+      "train_step": 9904,
+      "loss": 0.600593374397811,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.600593374397811
+    },
+    {
+      "train_step": 9905,
+      "loss": 0.4797273867869698,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4797273867869698
+    },
+    {
+      "train_step": 9906,
+      "loss": 0.5516435373523403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5516435373523403
+    },
+    {
+      "train_step": 9907,
+      "loss": 0.6470588954009403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6470588954009403
+    },
+    {
+      "train_step": 9908,
+      "loss": 0.48552317543258633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48552317543258633
+    },
+    {
+      "train_step": 9909,
+      "loss": 0.8420359197659957,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.8420359197659957
+    },
+    {
+      "train_step": 9910,
+      "loss": 0.38690220833824207,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.38690220833824207
+    },
+    {
+      "train_step": 9911,
+      "loss": 0.4811309474005593,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4811309474005593
+    },
+    {
+      "train_step": 9912,
+      "loss": 0.39668618016888935,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39668618016888935
+    },
+    {
+      "train_step": 9913,
+      "loss": 0.6085644019552856,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6085644019552856
+    },
+    {
+      "train_step": 9914,
+      "loss": 0.6247620474885522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6247620474885522
+    },
+    {
+      "train_step": 9915,
+      "loss": 0.48840951988181136,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48840951988181136
+    },
+    {
+      "train_step": 9916,
+      "loss": 0.7404764598259163,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7404764598259163
+    },
+    {
+      "train_step": 9917,
+      "loss": 0.6246296287791838,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6246296287791838
+    },
+    {
+      "train_step": 9918,
+      "loss": 0.35091715345645874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.35091715345645874
+    },
+    {
+      "train_step": 9919,
+      "loss": 0.7357630768874549,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7357630768874549
+    },
+    {
+      "train_step": 9920,
+      "loss": 0.4396037742773921,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4396037742773921
+    },
+    {
+      "train_step": 9921,
+      "loss": 0.7004715360315913,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7004715360315913
+    },
+    {
+      "train_step": 9922,
+      "loss": 0.6155505718125763,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6155505718125763
+    },
+    {
+      "train_step": 9923,
+      "loss": 0.49101217926477236,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49101217926477236
+    },
+    {
+      "train_step": 9924,
+      "loss": 0.5053794939167439,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5053794939167439
+    },
+    {
+      "train_step": 9925,
+      "loss": 0.39251678062660805,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.39251678062660805
+    },
+    {
+      "train_step": 9926,
+      "loss": 0.28887689357493046,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.28887689357493046
+    },
+    {
+      "train_step": 9927,
+      "loss": 0.6049033759666377,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6049033759666377
+    },
+    {
+      "train_step": 9928,
+      "loss": 0.48141946483531245,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48141946483531245
+    },
+    {
+      "train_step": 9929,
+      "loss": 0.6435841332810645,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6435841332810645
+    },
+    {
+      "train_step": 9930,
+      "loss": 0.5616810331721874,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5616810331721874
+    },
+    {
+      "train_step": 9931,
+      "loss": 0.5981370921894182,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5981370921894182
+    },
+    {
+      "train_step": 9932,
+      "loss": 0.6839962189035987,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6839962189035987
+    },
+    {
+      "train_step": 9933,
+      "loss": 0.48530804244839953,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48530804244839953
+    },
+    {
+      "train_step": 9934,
+      "loss": 0.3892251716326922,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3892251716326922
+    },
+    {
+      "train_step": 9935,
+      "loss": 0.7797485712379688,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7797485712379688
+    },
+    {
+      "train_step": 9936,
+      "loss": 0.40964980494804776,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40964980494804776
+    },
+    {
+      "train_step": 9937,
+      "loss": 0.458126548877937,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.458126548877937
+    },
+    {
+      "train_step": 9938,
+      "loss": 0.5700778717076858,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5700778717076858
+    },
+    {
+      "train_step": 9939,
+      "loss": 0.6923969415342021,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6923969415342021
+    },
+    {
+      "train_step": 9940,
+      "loss": 0.5266348125414522,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5266348125414522
+    },
+    {
+      "train_step": 9941,
+      "loss": 0.5561779389361647,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5561779389361647
+    },
+    {
+      "train_step": 9942,
+      "loss": 0.4608301951590238,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4608301951590238
+    },
+    {
+      "train_step": 9943,
+      "loss": 0.5287357008473904,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5287357008473904
+    },
+    {
+      "train_step": 9944,
+      "loss": 0.49398829285299267,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49398829285299267
+    },
+    {
+      "train_step": 9945,
+      "loss": 0.4408388024556184,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4408388024556184
+    },
+    {
+      "train_step": 9946,
+      "loss": 0.40031626255421726,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.40031626255421726
+    },
+    {
+      "train_step": 9947,
+      "loss": 0.49475474183997825,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49475474183997825
+    },
+    {
+      "train_step": 9948,
+      "loss": 0.5222238371431048,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5222238371431048
+    },
+    {
+      "train_step": 9949,
+      "loss": 0.5215060760266578,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5215060760266578
+    },
+    {
+      "train_step": 9950,
+      "loss": 0.6116409310820816,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6116409310820816
+    },
+    {
+      "train_step": 9951,
+      "loss": 0.5870658060522062,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5870658060522062
+    },
+    {
+      "train_step": 9952,
+      "loss": 0.3570945735199266,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3570945735199266
+    },
+    {
+      "train_step": 9953,
+      "loss": 0.6171164477658777,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6171164477658777
+    },
+    {
+      "train_step": 9954,
+      "loss": 0.5170353651703622,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5170353651703622
+    },
+    {
+      "train_step": 9955,
+      "loss": 0.610894595078409,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.610894595078409
+    },
+    {
+      "train_step": 9956,
+      "loss": 0.48945862371611215,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.48945862371611215
+    },
+    {
+      "train_step": 9957,
+      "loss": 0.5611856171394328,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5611856171394328
+    },
+    {
+      "train_step": 9958,
+      "loss": 0.5606707786502313,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5606707786502313
+    },
+    {
+      "train_step": 9959,
+      "loss": 0.6632172830156589,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6632172830156589
+    },
+    {
+      "train_step": 9960,
+      "loss": 0.3775862021587636,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3775862021587636
+    },
+    {
+      "train_step": 9961,
+      "loss": 0.4259117711646779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4259117711646779
+    },
+    {
+      "train_step": 9962,
+      "loss": 0.6178690667313191,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6178690667313191
+    },
+    {
+      "train_step": 9963,
+      "loss": 0.6036065033888706,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6036065033888706
+    },
+    {
+      "train_step": 9964,
+      "loss": 0.7600110426018779,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7600110426018779
+    },
+    {
+      "train_step": 9965,
+      "loss": 0.4784097325345547,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4784097325345547
+    },
+    {
+      "train_step": 9966,
+      "loss": 0.5778879388317496,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5778879388317496
+    },
+    {
+      "train_step": 9967,
+      "loss": 0.5246039163519325,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5246039163519325
+    },
+    {
+      "train_step": 9968,
+      "loss": 0.5859806560724149,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5859806560724149
+    },
+    {
+      "train_step": 9969,
+      "loss": 0.6186912698818424,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6186912698818424
+    },
+    {
+      "train_step": 9970,
+      "loss": 0.351740573319609,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.351740573319609
+    },
+    {
+      "train_step": 9971,
+      "loss": 0.49424847317528886,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.49424847317528886
+    },
+    {
+      "train_step": 9972,
+      "loss": 0.47300124992172277,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.47300124992172277
+    },
+    {
+      "train_step": 9973,
+      "loss": 0.4576925970802608,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4576925970802608
+    },
+    {
+      "train_step": 9974,
+      "loss": 0.5046392209107512,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5046392209107512
+    },
+    {
+      "train_step": 9975,
+      "loss": 0.5768395887333417,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5768395887333417
+    },
+    {
+      "train_step": 9976,
+      "loss": 0.5271221297040687,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5271221297040687
+    },
+    {
+      "train_step": 9977,
+      "loss": 0.3714014385081302,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3714014385081302
+    },
+    {
+      "train_step": 9978,
+      "loss": 0.41948622963315363,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.41948622963315363
+    },
+    {
+      "train_step": 9979,
+      "loss": 0.5924367679519063,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5924367679519063
+    },
+    {
+      "train_step": 9980,
+      "loss": 0.6800010957099167,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6800010957099167
+    },
+    {
+      "train_step": 9981,
+      "loss": 0.4502075270006124,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4502075270006124
+    },
+    {
+      "train_step": 9982,
+      "loss": 0.5566553655099529,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5566553655099529
+    },
+    {
+      "train_step": 9983,
+      "loss": 0.5422141494091132,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5422141494091132
+    },
+    {
+      "train_step": 9984,
+      "loss": 0.5264166700846769,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5264166700846769
+    },
+    {
+      "train_step": 9985,
+      "loss": 0.4526153673466812,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4526153673466812
+    },
+    {
+      "train_step": 9986,
+      "loss": 0.4206322732932719,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4206322732932719
+    },
+    {
+      "train_step": 9987,
+      "loss": 0.610902064782186,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.610902064782186
+    },
+    {
+      "train_step": 9988,
+      "loss": 0.5446708922343403,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5446708922343403
+    },
+    {
+      "train_step": 9989,
+      "loss": 0.7255318880287633,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.7255318880287633
+    },
+    {
+      "train_step": 9990,
+      "loss": 0.3136911078129671,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.3136911078129671
+    },
+    {
+      "train_step": 9991,
+      "loss": 0.4700878304784859,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4700878304784859
+    },
+    {
+      "train_step": 9992,
+      "loss": 0.34364911306437357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34364911306437357
+    },
+    {
+      "train_step": 9993,
+      "loss": 0.5790980227797597,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5790980227797597
+    },
+    {
+      "train_step": 9994,
+      "loss": 0.5150725068079468,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.5150725068079468
+    },
+    {
+      "train_step": 9995,
+      "loss": 0.4151347438145761,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4151347438145761
+    },
+    {
+      "train_step": 9996,
+      "loss": 0.4250239150549567,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4250239150549567
+    },
+    {
+      "train_step": 9997,
+      "loss": 0.6377064253833682,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6377064253833682
+    },
+    {
+      "train_step": 9998,
+      "loss": 0.34051030051367615,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.34051030051367615
+    },
+    {
+      "train_step": 9999,
+      "loss": 0.4751579721690357,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.4751579721690357
+    },
+    {
+      "train_step": 10000,
+      "loss": 0.6164826461442034,
+      "clean_loss": 0.0,
+      "noisy_loss_mean": 0.6164826461442034
+    }
+  ],
+  "evals": [
+    {
+      "kind": "initial",
+      "train_step": 0,
+      "acc": 0.517578125,
+      "tok_acc": 0.8276668595679012
+    },
+    {
+      "kind": "task",
+      "train_step": 1000,
+      "acc": 0.646484375,
+      "tok_acc": 0.8738184799382716
+    },
+    {
+      "kind": "task",
+      "train_step": 2000,
+      "acc": 0.63671875,
+      "tok_acc": 0.8706597222222222
+    },
+    {
+      "kind": "task",
+      "train_step": 3000,
+      "acc": 0.6328125,
+      "tok_acc": 0.8692129629629629
+    },
+    {
+      "kind": "task",
+      "train_step": 4000,
+      "acc": 0.63671875,
+      "tok_acc": 0.8744936342592593
+    },
+    {
+      "kind": "task",
+      "train_step": 5000,
+      "acc": 0.666015625,
+      "tok_acc": 0.8807388117283951
+    },
+    {
+      "kind": "task",
+      "train_step": 6000,
+      "acc": 0.65234375,
+      "tok_acc": 0.8770736882716049
+    },
+    {
+      "kind": "task",
+      "train_step": 7000,
+      "acc": 0.646484375,
+      "tok_acc": 0.8756751543209876
+    },
+    {
+      "kind": "task",
+      "train_step": 8000,
+      "acc": 0.63671875,
+      "tok_acc": 0.8682484567901234
+    },
+    {
+      "kind": "task",
+      "train_step": 9000,
+      "acc": 0.646484375,
+      "tok_acc": 0.8732638888888888
+    },
+    {
+      "kind": "task",
+      "train_step": 10000,
+      "acc": 0.666015625,
+      "tok_acc": 0.8807388117283951
+    },
+    {
+      "kind": "final",
+      "train_step": 10000,
+      "acc": 0.666015625,
+      "tok_acc": 0.8807388117283951
+    }
+  ],
+  "final_acc": 0.666015625,
+  "final_tok_acc": 0.8807388117283951
+}
\ No newline at end of file