Lahiru-LK commited on
Commit
45041d3
·
verified ·
1 Parent(s): e05aa19

Upload 2 files

Browse files
Files changed (2) hide show
  1. .DS_Store +0 -0
  2. inspect_model.py +9 -0
.DS_Store ADDED
Binary file (6.15 kB). View file
 
inspect_model.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+
3
+ # Load the saved model
4
+ checkpoint = torch.load('best_codebert_model.pt', map_location='cpu')
5
+
6
+ # Print the keys to understand the structure
7
+ print("Model keys:")
8
+ for key in list(checkpoint.keys())[:10]: # Print first 10 keys
9
+ print(key)