Spaces:
Sleeping
Sleeping
Update models/m5_cfdna/loader.py
Browse files
models/m5_cfdna/loader.py
CHANGED
|
@@ -8,7 +8,7 @@ Usage:
|
|
| 8 |
|
| 9 |
import torch
|
| 10 |
from pathlib import Path
|
| 11 |
-
from model import Pleiades
|
| 12 |
|
| 13 |
# Keys the training code may have used to store the model state dict
|
| 14 |
_MODEL_STATE_KEYS = ("model", "model_state_dict", "state_dict")
|
|
|
|
| 8 |
|
| 9 |
import torch
|
| 10 |
from pathlib import Path
|
| 11 |
+
from models.model import Pleiades
|
| 12 |
|
| 13 |
# Keys the training code may have used to store the model state dict
|
| 14 |
_MODEL_STATE_KEYS = ("model", "model_state_dict", "state_dict")
|