solved the problem that cannot run result() after load_model

zjuwss · Mar 17, 2024 · 1ee480f · 1ee480f
1 parent 748a5df
commit 1ee480f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/gnnwr/models.py b/src/gnnwr/models.py
@@ -579,6 +579,8 @@ def load_model(self, path, use_dict=False, map_location=None):
         else:
             self._model = self._model.cpu()
             self._out = self._out.cpu()
+        self._modelSavePath = os.path.dirname(path)
+        self._modelName = os.path.basename(path).split('/')[-1].split('.')[0]
         self.__istrained = True