Fix one gpu training in deeplabv3+ (#1966)

023742e3 · qingqing01 · GitHub · 1af11949 · 023742e3
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

fluid/PaddleCV/deeplabv3+/train.py fluid/PaddleCV/deeplabv3+/train.py +1 -1

未找到文件。
--- a/fluid/PaddleCV/deeplabv3+/train.py
+++ b/fluid/PaddleCV/deeplabv3+/train.py
@@ -184,7 +184,7 @@ if args.parallel:
        build_strategy=build_strategy,
        exec_strategy=exec_strategy)
 else:
-    binary = fluid.compiler.CompiledProgram(main)
+    binary = fluid.compiler.CompiledProgram(tp)

 if args.use_py_reader:
    assert(batch_size % fluid.core.get_cuda_device_count() == 0)