Skip to content

S
seminar-breakout
Commit f002bfb9 authored by Chao Fang's avatar Chao Fang Committed by Yuxin Wu
Browse files
Options

fix bug of tf version check in fasterrcnn (#773)

parent 883ef79f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/FasterRCNN/train.py
View file @ f002bfb9
......@@ -62,7 +62,7 @@ def get_model_output_names():
def get_model():
if config.MODE_FPN:
if get_tf_version() < 1.6:
if get_tf_version_number() < 1.6:
logger.warn("FPN has chances to crash in TF<1.6, due to a TF issue.")
return ResNetFPNModel()
else:
......
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