Skip to content

S
seminar-breakout
Commit 2981c5d4 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix typo in last commit

parent cefc7748
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/FasterRCNN/train.py
View file @ 2981c5d4
......@@ -86,7 +86,7 @@ if __name__ == '__main__':
ScheduledHyperParamSetter('learning_rate', lr_schedule),
GPUMemoryTracker(),
HostMemoryTracker(),
ThroughputTracker(samples_per_epoch=cfg.TRAIN.NUM_GPUS),
ThroughputTracker(samples_per_step=cfg.TRAIN.NUM_GPUS),
EstimatedTimeLeft(median=True),
SessionRunTimeout(60000), # 1 minute timeout
]
......
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