model_load.py

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()))