Update adrd/model/adrd_model.py
Browse filesAdd weights_only=False to torch.load
- adrd/model/adrd_model.py +1 -1
adrd/model/adrd_model.py
CHANGED
@@ -749,7 +749,7 @@ class ADRDModel(BaseEstimator):
|
|
749 |
:type img_dict: _type_, optional
|
750 |
"""
|
751 |
# load state_dict
|
752 |
-
state_dict = torch.load(filepath, map_location=map_location)
|
753 |
|
754 |
# load data modalities
|
755 |
self.src_modalities: dict[str, dict[str, Any]] = state_dict.pop('src_modalities')
|
|
|
749 |
:type img_dict: _type_, optional
|
750 |
"""
|
751 |
# load state_dict
|
752 |
+
state_dict = torch.load(filepath, map_location=map_location, weights_only=False)
|
753 |
|
754 |
# load data modalities
|
755 |
self.src_modalities: dict[str, dict[str, Any]] = state_dict.pop('src_modalities')
|