Skip to content

Commit 72cd50d

Browse files
committed
update weight_path
1 parent dd2351e commit 72cd50d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphgallery/nn/models/base_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ def __init__(self, *graph, device="cpu:0", seed=None, name=None, **kwargs):
4848
# log path
4949
# add random integer to avoid duplication
5050
_id = np.random.RandomState(None).randint(100)
51-
self.weight_path = osp.join(osp.expanduser(osp.normpath("/tmp/weight")),
52-
f"{self.name}_{_id}_weights{POSTFIX}")
51+
self.weight_path = osp.join(osp.expanduser(osp.normpath("./")),
52+
f"{self.name}_weights_id_{_id}{POSTFIX}")
5353

5454
def save(self, path=None, as_model=False, overwrite=True, save_format=None, **kwargs):
5555

0 commit comments

Comments
 (0)