Commit ·
f51aa5c
1
Parent(s): a21e922
Prepared for deploy
Browse files
utils.py
CHANGED
|
@@ -23,7 +23,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 23 |
# GAT_HEADS = 2
|
| 24 |
# HIDDEN_CHANNELS = 256
|
| 25 |
|
| 26 |
-
MODEL_PATH = "models/
|
| 27 |
GAT_HEADS = 4
|
| 28 |
HIDDEN_CHANNELS = 128
|
| 29 |
|
|
|
|
| 23 |
# GAT_HEADS = 2
|
| 24 |
# HIDDEN_CHANNELS = 256
|
| 25 |
|
| 26 |
+
MODEL_PATH = "models/model_ep041_attention_mse1.9153.pth"
|
| 27 |
GAT_HEADS = 4
|
| 28 |
HIDDEN_CHANNELS = 128
|
| 29 |
|