add find_unused_parameters param (#668)

* add find_unused_parameters param * fix default val

add find_unused_parameters param (#668)
* add find_unused_parameters param * fix default val
a6e2114e · littletomatodonkey · GitHub · 794caef2 · a6e2114e
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

tools/train.py tools/train.py +3 -1

未找到文件。
--- a/tools/train.py
+++ b/tools/train.py
@@ -70,7 +70,9 @@ def main(args):
        config, parameter_list=net.parameters())

    if config["use_data_parallel"]:
-        net = paddle.DataParallel(net)
+        find_unused_parameters = config.get("find_unused_parameters", False)
+        net = paddle.DataParallel(
+            net, find_unused_parameters=find_unused_parameters)

    # load model from checkpoint or pretrained model
    init_model(config, net, optimizer)