fix train.py not add place

20a0107d · dengkaipeng · 881f62a3 · 20a0107d
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- a/fluid/PaddleCV/video/train.py
+++ b/fluid/PaddleCV/video/train.py
@@ -151,7 +151,7 @@ def train(args):
                "Given pretrain weight dir {} not exist.".format(args.pretrain)
    pretrain = args.pretrain or train_model.get_pretrain_weights()
    if pretrain:
-        train_model.load_pretrain_params(exe, pretrain, train_prog)
+        train_model.load_pretrain_params(exe, pretrain, train_prog, place)

    train_exe = fluid.ParallelExecutor(
        use_cuda=args.use_gpu,