Delete egs/ami/ASR/xlsr_transducer/test.py
Browse files
egs/ami/ASR/xlsr_transducer/test.py
DELETED
|
@@ -1,9 +0,0 @@
|
|
| 1 |
-
import torch
|
| 2 |
-
checkpoint = torch.load("xlsr_transducer/exp_16gb/best-train-loss.pt")
|
| 3 |
-
print("Checkpoint keys:", checkpoint.keys())
|
| 4 |
-
if 'epoch' in checkpoint:
|
| 5 |
-
print(f"Epoch: {checkpoint['epoch']}")
|
| 6 |
-
if 'batch_idx_train' in checkpoint:
|
| 7 |
-
print(f"Batch idx: {checkpoint['batch_idx_train']}")
|
| 8 |
-
if 'best_train_loss' in checkpoint:
|
| 9 |
-
print(f"Best train loss: {checkpoint['best_train_loss']}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|