未验证 提交 fbd0078c 编写于 作者: R ruri 提交者: GitHub

Fix mobilenet (#3335)

* fix mobilenet bug

* fix inplce warning
上级 38841c3b
......@@ -216,8 +216,3 @@ def MobileNetV1_x1_0():
def MobileNetV1_x0_75():
model = MobileNetV1(scale=0.75)
return model
def MobileNetV1():
model = MobileNetV1(scale=1.0)
return model
......@@ -228,8 +228,3 @@ def MobileNetV2_x1_5():
def MobileNetV2_x2_0():
model = MobileNetV2(scale=2.0)
return model
def MobileNetV2():
model = MobileNetV2(scale=1.0)
return model
......@@ -368,7 +368,8 @@ def best_strategy_compiled(args, program, loss):
return program
else:
build_strategy = fluid.compiler.BuildStrategy()
build_strategy.enable_inplace = True
#Feature will be supported in Fluid v1.6
#build_strategy.enable_inplace = True
exec_strategy = fluid.ExecutionStrategy()
exec_strategy.num_threads = fluid.core.get_cuda_device_count()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册