Skip to content

Commit fe238e2

Browse files
authored
Fix a bug in the script run.py
1 parent 6aa3402 commit fe238e2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

script/run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def train_and_validate(cfg, solver):
2828
solver.save("model_epoch_%d.pth" % solver.epoch)
2929
solver.model.split = "valid"
3030
metric = solver.evaluate("valid")
31-
result = metric[args.metric]
31+
result = metric[cfg.metric]
3232
if result > best_result:
3333
best_result = result
3434
best_epoch = solver.epoch
@@ -60,4 +60,4 @@ def test(cfg, solver):
6060
solver = util.build_solver(cfg, dataset)
6161

6262
train_and_validate(cfg, solver)
63-
test(cfg, solver)
63+
test(cfg, solver)

0 commit comments

Comments
 (0)