mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update --eval-step to --eval-freq

pull/119/head
KaiyangZhou 2018-11-09 21:58:58 +00:00
parent 36baeebb7b
commit 4a2b98c3b7
5 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
args.py
View File

@ -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")

2
train_imgreid_xent.py
View File

@ -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:

2
train_imgreid_xent_htri.py
View File

@ -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:

2
train_vidreid_xent.py
View File

@ -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:

2
train_vidreid_xent_htri.py
View File

@ -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: