我对Memory Networks和Movie Dialog QA感兴趣。最近facebook宣布了一个名为ParlAI的人工智能培训框架,它支持许多模型和数据集。虽然我在ParlAI上尝试了下面的命令,但是训练在第一次loss.backward()
在memnn.py
停止了。我等了差不多一天,但loss.backward()
还没说完。我已经通过打印调试和[Using Cuda]
打印检查了这一点。事实上,我的GPU工作,因为它使用了一些内存。我查过了。你知道吗
python examples/train_model.py -m memnn -t "#moviedd-qa" -bs 32 --gpu 0 -e 10
然后,我切换到简单的任务,它完成了几分钟。你知道吗
python examples/train_model.py -m memnn -t "babi:task1k:1" -bs 32 --gpu 0 -e 10
我意识到#moviedd-qa
比babi任务更复杂。但在我的环境中,训练这个模型通常需要多长时间?有人想通过帕莱训练这个模特吗?恐怕这不是帕莱的虫子。你能建议我继续工作吗?你知道吗
我的环境
我也是asking developers at ParlAI at their github,但没有回应。你知道吗
目前没有回答
相关问题 更多 >
编程相关推荐