From 72cd50da719c6742d1ee5ced928d698aad7c1f98 Mon Sep 17 00:00:00 2001 From: EdisonLeeeee Date: Wed, 21 Oct 2020 11:34:45 +0800 Subject: [PATCH] update weight_path --- graphgallery/nn/models/base_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/graphgallery/nn/models/base_model.py b/graphgallery/nn/models/base_model.py index 6f4c8f36..b4f9bdc1 100755 --- 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):