手电筒向后损失()在帕莱被吊死

2024-10-01 02:31:41 发布

您现在位置:Python中文网/ 问答频道 /正文

我对Memory NetworksMovie 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-qababi任务更复杂。但在我的环境中,训练这个模型通常需要多长时间?有人想通过帕莱训练这个模特吗?恐怕这不是帕莱的虫子。你能建议我继续工作吗?你知道吗

我的环境

  • Ubunt 16.04.03 LTS,64位
  • python 3.6.1(Anaconda 4.4.0(64位))
  • GPU:GTX 1080 ti
  • CPU:Intel(R)Core(TM)i7-7700K CPU@4.20GHz
  • 手电筒。版本:'0.2.0\u 3'

我也是asking developers at ParlAI at their github,但没有回应。你知道吗


Tags: py模型modelbsgputrainqaexamples