update weight_path

EdisonLeeeee · Oct 21, 2020 · 72cd50d · 72cd50d
1 parent dd2351e
commit 72cd50d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphgallery/nn/models/base_model.py b/graphgallery/nn/models/base_model.py
@@ -48,8 +48,8 @@ def __init__(self, *graph, device="cpu:0", seed=None, name=None, **kwargs):
         # log path
         # add random integer to avoid duplication
         _id = np.random.RandomState(None).randint(100)
-        self.weight_path = osp.join(osp.expanduser(osp.normpath("/tmp/weight")),
-                                    f"{self.name}_{_id}_weights{POSTFIX}")
+        self.weight_path = osp.join(osp.expanduser(osp.normpath("./")),
+                                    f"{self.name}_weights_id_{_id}{POSTFIX}")
 
     def save(self, path=None, as_model=False, overwrite=True, save_format=None, **kwargs):