File size: 252 Bytes
45041d3 |
1 2 3 4 5 6 7 8 9 |
import torch
# Load the saved model
checkpoint = torch.load('best_codebert_model.pt', map_location='cpu')
# Print the keys to understand the structure
print("Model keys:")
for key in list(checkpoint.keys())[:10]: # Print first 10 keys
print(key) |