| import torch | |
| ckpt = torch.load("/workspace/trainTTS/model/epochs_2nd_00020.pth", map_location='cpu') | |
| print(f"Checkpoint info:") | |
| print(f" - Epoch: {ckpt.get('epoch', 'N/A')}") | |
| print(f" - Iters: {ckpt.get('iters', 'N/A')}") | |
| print(f" - Val Loss: {ckpt.get('val_loss', 'N/A')}") | |
| print(f" - Has optimizer: {'optimizer' in ckpt}") | |
| print(list(ckpt['net'].keys())) |