Upload djmgnn-base model
Browse files- base_model/README.md +1 -1
- base_model/pytorch_model.pt +3 -0
base_model/README.md
CHANGED
|
@@ -93,7 +93,7 @@ model = DJMGNN(
|
|
| 93 |
)
|
| 94 |
|
| 95 |
# Load the checkpoint
|
| 96 |
-
checkpoint = torch.load("path/to/
|
| 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 |
```
|
base_model/pytorch_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:98b44ad053c7752f06c860d6de62c0d73c59988299fdbcdb4aa8111a1450847a
|
| 3 |
+
size 130483624
|