saketh11 commited on
Commit
571c672
·
verified ·
1 Parent(s): b39b124

Upload djmgnn-pfas-finetuned model

Browse files
finetuned_model/README.md CHANGED
@@ -93,7 +93,7 @@ model = DJMGNN(
93
  )
94
 
95
  # Load the checkpoint
96
- checkpoint = torch.load("path/to/checkpoint.pt", map_location="cpu")
97
  model.load_state_dict(checkpoint["model_state_dict"])
98
  model.eval()
99
  ```
 
93
  )
94
 
95
  # Load the checkpoint
96
+ checkpoint = torch.load("path/to/pytorch_model.pt", map_location="cpu")
97
  model.load_state_dict(checkpoint["model_state_dict"])
98
  model.eval()
99
  ```
finetuned_model/pytorch_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bac5dcb7ab9f9d59fd77b4ab02f4ce1380cf78d730e5541bdefd7d7d244a2f32
3
+ size 130213414