From 4a2b98c3b780d9fc1149519aef8e1cf2e21d0ea6 Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Fri, 9 Nov 2018 21:58:58 +0000 Subject: [PATCH] update --eval-step to --eval-freq --- args.py | 4 ++-- train_imgreid_xent.py | 2 +- train_imgreid_xent_htri.py | 2 +- train_vidreid_xent.py | 2 +- train_vidreid_xent_htri.py | 2 +- 5 files changed, 6 insertions(+), 6 deletions(-) diff --git a/args.py b/args.py index 9d0e6e3..f8e8893 100644 --- a/args.py +++ b/args.py @@ -125,8 +125,8 @@ def argument_parser(): help="load pretrained weights but ignore layers that don't match in size") parser.add_argument('--evaluate', action='store_true', help="evaluate only") - parser.add_argument('--eval-step', type=int, default=-1, - help="run evaluation for every N epochs (set to -1 to test only in the end)") + parser.add_argument('--eval-freq', type=int, default=-1, + help="evaluation frequency (set to -1 to test only in the end)") parser.add_argument('--start-eval', type=int, default=0, help="start to evaluate after a specific epoch") diff --git a/train_imgreid_xent.py b/train_imgreid_xent.py index c2d0a42..872c39b 100755 --- a/train_imgreid_xent.py +++ b/train_imgreid_xent.py @@ -122,7 +122,7 @@ def main(): scheduler.step() - if (epoch + 1) > args.start_eval and args.eval_step > 0 and (epoch + 1) % args.eval_step == 0 or (epoch + 1) == args.max_epoch: + if (epoch + 1) > args.start_eval and args.eval_freq > 0 and (epoch + 1) % args.eval_freq == 0 or (epoch + 1) == args.max_epoch: print("==> Test") for name in args.target_names: diff --git a/train_imgreid_xent_htri.py b/train_imgreid_xent_htri.py index d43213f..b55fb61 100755 --- a/train_imgreid_xent_htri.py +++ b/train_imgreid_xent_htri.py @@ -125,7 +125,7 @@ def main(): scheduler.step() - if (epoch + 1) > args.start_eval and args.eval_step > 0 and (epoch + 1) % args.eval_step == 0 or (epoch + 1) == args.max_epoch: + if (epoch + 1) > args.start_eval and args.eval_freq > 0 and (epoch + 1) % args.eval_freq == 0 or (epoch + 1) == args.max_epoch: print("==> Test") for name in args.target_names: diff --git a/train_vidreid_xent.py b/train_vidreid_xent.py index 44412bf..4385d64 100755 --- a/train_vidreid_xent.py +++ b/train_vidreid_xent.py @@ -123,7 +123,7 @@ def main(): scheduler.step() - if (epoch + 1) > args.start_eval and args.eval_step > 0 and (epoch + 1) % args.eval_step == 0 or (epoch + 1) == args.max_epoch: + if (epoch + 1) > args.start_eval and args.eval_freq > 0 and (epoch + 1) % args.eval_freq == 0 or (epoch + 1) == args.max_epoch: print("==> Test") for name in args.target_names: diff --git a/train_vidreid_xent_htri.py b/train_vidreid_xent_htri.py index d3e9504..8a3bd11 100755 --- a/train_vidreid_xent_htri.py +++ b/train_vidreid_xent_htri.py @@ -127,7 +127,7 @@ def main(): scheduler.step() - if (epoch + 1) > args.start_eval and args.eval_step > 0 and (epoch + 1) % args.eval_step == 0 or (epoch + 1) == args.max_epoch: + if (epoch + 1) > args.start_eval and args.eval_freq > 0 and (epoch + 1) % args.eval_freq == 0 or (epoch + 1) == args.max_epoch: print("==> Test") for name in args.target_names: