Skip to content

S
seminar-breakout
Commit 0aa92de5 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

[MaskRCNN] Allow EVAL_PERIOD=0 to disable evaluation (#1112)

parent 167fa123
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
examples/FasterRCNN/train.py
View file @ 0aa92de5
...@@ -487,10 +487,12 @@ if __name__ == '__main__': ...@@ -487,10 +487,12 @@ if __name__ == '__main__':
PeakMemoryTracker(), PeakMemoryTracker(),
EstimatedTimeLeft(median=True), EstimatedTimeLeft(median=True),
SessionRunTimeout(60000).set_chief_only(True), # 1 minute timeout SessionRunTimeout(60000).set_chief_only(True), # 1 minute timeout
] + [
EvalCallback(dataset, *MODEL.get_inference_tensor_names(), args.logdir)
for dataset in cfg.DATA.VAL
] ]
if cfg.TRAIN.EVAL_PERIOD > 0:
callbacks.extend([
EvalCallback(dataset, *MODEL.get_inference_tensor_names(), args.logdir)
for dataset in cfg.DATA.VAL
])
if not is_horovod: if not is_horovod:
callbacks.append(GPUUtilizationTracker()) callbacks.append(GPUUtilizationTracker())
......
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