efrainmain commited on
Commit
b15143a
·
verified ·
1 Parent(s): 093ba76

Fix train_loss metadata

Browse files
best_checkpoints.json CHANGED
@@ -2,7 +2,7 @@
2
  {
3
  "epoch": 1,
4
  "val_loss": 192.81560051995353,
5
- "train_loss": Infinity,
6
  "path": "/kaggle/working/chunkformer-ctc-large-vie-finetune/epoch_1_val_loss_192.8156.pt"
7
  }
8
  ]
 
2
  {
3
  "epoch": 1,
4
  "val_loss": 192.81560051995353,
5
+ "train_loss": 326.93824258185526,
6
  "path": "/kaggle/working/chunkformer-ctc-large-vie-finetune/epoch_1_val_loss_192.8156.pt"
7
  }
8
  ]
resume_checkpoint/best_checkpoints.json CHANGED
@@ -2,7 +2,7 @@
2
  {
3
  "epoch": 1,
4
  "val_loss": 192.81560051995353,
5
- "train_loss": Infinity,
6
  "path": "/kaggle/working/chunkformer-ctc-large-vie-finetune/epoch_1_val_loss_192.8156.pt"
7
  }
8
  ]
 
2
  {
3
  "epoch": 1,
4
  "val_loss": 192.81560051995353,
5
+ "train_loss": 326.93824258185526,
6
  "path": "/kaggle/working/chunkformer-ctc-large-vie-finetune/epoch_1_val_loss_192.8156.pt"
7
  }
8
  ]
resume_checkpoint/trainer_state.json CHANGED
@@ -8,7 +8,7 @@
8
  "best_epoch": 1,
9
  "best_val_loss": 192.81560051995353,
10
  "val_loss": 192.81560051995353,
11
- "train_loss": Infinity,
12
  "has_optimizer_state": true,
13
  "has_scheduler_state": true,
14
  "has_scaler_state": true,
@@ -37,5 +37,6 @@
37
  "skip_bad_samples": true,
38
  "max_train_batches": null,
39
  "max_eval_batches": null
40
- }
 
41
  }
 
8
  "best_epoch": 1,
9
  "best_val_loss": 192.81560051995353,
10
  "val_loss": 192.81560051995353,
11
+ "train_loss": 326.93824258185526,
12
  "has_optimizer_state": true,
13
  "has_scheduler_state": true,
14
  "has_scaler_state": true,
 
37
  "skip_bad_samples": true,
38
  "max_train_batches": null,
39
  "max_eval_batches": null
40
+ },
41
+ "best_train_loss": 326.93824258185526
42
  }
trainer_state.json CHANGED
@@ -8,7 +8,7 @@
8
  "best_epoch": 1,
9
  "best_val_loss": 192.81560051995353,
10
  "val_loss": 192.81560051995353,
11
- "train_loss": Infinity,
12
  "has_optimizer_state": true,
13
  "has_scheduler_state": true,
14
  "has_scaler_state": true,
@@ -37,5 +37,6 @@
37
  "skip_bad_samples": true,
38
  "max_train_batches": null,
39
  "max_eval_batches": null
40
- }
 
41
  }
 
8
  "best_epoch": 1,
9
  "best_val_loss": 192.81560051995353,
10
  "val_loss": 192.81560051995353,
11
+ "train_loss": 326.93824258185526,
12
  "has_optimizer_state": true,
13
  "has_scheduler_state": true,
14
  "has_scaler_state": true,
 
37
  "skip_bad_samples": true,
38
  "max_train_batches": null,
39
  "max_eval_batches": null
40
+ },
41
+ "best_train_loss": 326.93824258185526
42
  }