问题描述
PyTorch1.2中使用DataParallel的情况下,RNN报错了。部分报错信息如下:
loss: 250.3600 ||: : 3it [00:08, 3.74s/it]/opt/conda/conda-bld/pytorch_1570910687230/work/aten/src/ATen/native/cudnn/RNN.cpp:1238: UserWarning: RNN module weights are not part of single contiguous chunk of memory. This means they need to be compacted at every call, possibly greatly increasing memory usage. To compact weights again call flatten_parameters().
loss: 239.2263 ||: : 4it [00:08, 2.78s/it]/opt/conda/conda-bld/pytor

最低0.47元/天 解锁文章
6万+





