Skip to content

M
models

PaddlePaddle / models
大约 2 年 前同步成功

通知 232
Star 6828
Fork 2962
M
models
提交 20a0107d 编写于 作者: D dengkaipeng
浏览文件
操作

fix train.py not add place

上级 881f62a3
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
fluid/PaddleCV/video/train.py
浏览文件 @ 20a0107d
...@@ -151,7 +151,7 @@ def train(args): ...@@ -151,7 +151,7 @@ def train(args):
"Given pretrain weight dir {} not exist.".format(args.pretrain) "Given pretrain weight dir {} not exist.".format(args.pretrain)
pretrain = args.pretrain or train_model.get_pretrain_weights() pretrain = args.pretrain or train_model.get_pretrain_weights()
if pretrain: 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( train_exe = fluid.ParallelExecutor(
use_cuda=args.use_gpu, use_cuda=args.use_gpu,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或