刚开始用脚本训练,但是出了很多问题。
收藏
我原本是在notebook上训练的,然后想转到脚本里单机多卡。我照着教程里的做法把三处改动添加了上去
1.import paddle.distributed as dist #第1处改动,import库
2.dist.init_parallel_env()
3.包装模型modelG=paddle.DataParallel(modelG)
for i in range(3):
modelDs[i]=paddle.DataParallel(modelDs[i])
for i in range(3):
modelDs[i]=paddle.DataParallel(modelDs[i])
但是提交任务后就报这个错
fail to perform transform [] with error: We only support 'to_tensor()' in dynamic graph mode, please call 'paddle.disable_static()' to enter dynamic graph mode. and stack
好像是因为我在dataloader里用了to_tensor的原因,那么我改怎么做呢,是删掉to_tensor直接用numpy的array进行整个训练过程吗?
0
收藏
请登录后评论
dataloader的__getitem__()会自动加to_tensor,不用自己调用,直接用ndarray就好
学习了
学习了
学习了