Commit 0d32f19d authored by E144069X's avatar E144069X

Fixed bad use of arguments no_train

parent 945a05bc
......@@ -619,7 +619,7 @@ def run(args):
trainFunc(**kwargsTr)
else:
if not args.no_val:
net.load_state_dict(torch.load("../models/{}/model{}_epoch{}".format(args.exp_id_no_train[0],args.model_id_no_train[1],epoch),map_location="cpu" if not args.cuda else None))
net.load_state_dict(torch.load("../models/{}/model{}_epoch{}".format(args.exp_id_no_train,args.model_id_no_train,epoch),map_location="cpu" if not args.cuda else None))
if not args.no_val:
with torch.no_grad():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment