We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6aa3402 commit fe238e2Copy full SHA for fe238e2
script/run.py
@@ -28,7 +28,7 @@ def train_and_validate(cfg, solver):
28
solver.save("model_epoch_%d.pth" % solver.epoch)
29
solver.model.split = "valid"
30
metric = solver.evaluate("valid")
31
- result = metric[args.metric]
+ result = metric[cfg.metric]
32
if result > best_result:
33
best_result = result
34
best_epoch = solver.epoch
@@ -60,4 +60,4 @@ def test(cfg, solver):
60
solver = util.build_solver(cfg, dataset)
61
62
train_and_validate(cfg, solver)
63
- test(cfg, solver)
+ test(cfg, solver)
0 commit comments