chichuhu

Results 6 issues of chichuhu

高度不是32的数据集,num_workers不设置为0运行不了,显示RuntimeError: DataLoader worker (pid(s) 27292) exited unexpectedly,有好兄弟有解决方案吗

代码里面在把数据传入LSTM网络的时候把batch_size变换到了第1维(0开始),DataParallel默认按照第0维进行拆分和拼接。 所以第一种方法可以临时修改DataParallel中gather函数传入的参数dim改成1,即可解决,但是跑别的代码要记得改回来。 另一种方法可以将模型的最后输出batch_size改到第0维,这样并行就会按照batch_size拼接,然后再变换到第1维。(从代码中看的话传入CTC损失函数的batch_size是第1维)