Skip to content

Commit 56d707c

Browse files
authored
return false if load_and_assign fail to find a model (#919)
return false if load_and_assign fail to find a model
1 parent 00da491 commit 56d707c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayer/files/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1711,7 +1711,7 @@ def load_and_assign_npz(sess=None, name=None, network=None):
17111711
raise ValueError("session is None.")
17121712
if not os.path.exists(name):
17131713
logging.error("file {} doesn't exist.".format(name))
1714-
return
1714+
return False
17151715
else:
17161716
params = load_npz(name=name)
17171717
assign_params(sess, params, network)
@@ -1766,7 +1766,7 @@ def load_and_assign_npz_dict(name='model.npz', sess=None):
17661766

17671767
if not os.path.exists(name):
17681768
logging.error("file {} doesn't exist.".format(name))
1769-
return
1769+
return False
17701770

17711771
params = np.load(name)
17721772
if len(params.keys()) != len(set(params.keys())):

0 commit comments

Comments
 (0)