文件名称:wmg_agent:WMG代理
文件大小:47.67MB
文件格式:ZIP
更新时间:2024-05-24 13:05:34
Python
工作记忆图 该存储库随附了Ricky Loynd,Roland Fernandez,Asli Celikyilmaz,Adith Swaminathan和Matthew Hausknecht撰写的ICML 2020。 () WMG是基于Transformer的RL代理,它遵循代表观察到的状态和循环状态的动态向量集。 安装 (该代码已在Windows和Linux上使用Python 3.6进行了测试。) 创建一个新的虚拟环境。 激活虚拟环境。 安装PyTorch 1.3.1。 安装BabyAI(从任何目录): conda安装-c conda-forge python-blosc git clone -b iclr19 cd宝贝 pip install --user --editable。 光盘.. 克隆wmg_agent存储库。 cd wmg_agent 点安装-r r
【文件预览】:
wmg_agent-master
----utils()
--------graph.py(613B)
--------worker.py(11KB)
--------spec_reader.py(4KB)
----agents()
--------a3c.py(6KB)
--------networks()
----environments()
--------sokoban.py(22KB)
--------babyai.py(28KB)
--------pathfinding.py(10KB)
--------gym-sokoban-LICENSE(1KB)
----models()
--------wmg_factored_babyai.pth(8.54MB)
--------gru_factored_babyai.pth(4.34MB)
--------gru_pathfinding.pth(4.35MB)
--------nr_wmg_pathfinding.pth(812KB)
--------wmg_sokoban.pth(17.23MB)
--------wmg_pathfinding.pth(530KB)
--------wmg_flat_babyai.pth(7.72MB)
--------gru_flat_babyai.pth(5.64MB)
----results()
--------README.md(65B)
----images()
--------pathfinding.png(423KB)
--------sokoban2.gif(235KB)
--------sokoban.png(664KB)
--------babyai.png(259KB)
--------overview.png(557KB)
----run.py(569B)
----requirements.txt(78B)
----specs()
--------test_wmg_on_factored_babyai.py(1KB)
--------test_nr_wmg_on_pathfinding.py(960B)
--------render_wmg_on_sokoban.py(1KB)
--------train_gru_on_flat_babyai.py(960B)
--------train_gru_on_factored_babyai.py(983B)
--------train_wmg_on_flat_babyai.py(1KB)
--------test_gru_on_factored_babyai.py(989B)
--------train_gru_on_pathfinding.py(909B)
--------test_wmg_on_sokoban.py(1KB)
--------test_wmg_on_pathfinding.py(976B)
--------train_wmg_on_pathfinding.py(1KB)
--------test_gru_on_pathfinding.py(846B)
--------train_wmg_on_factored_babyai.py(1KB)
--------README.md(5KB)
--------train_wmg_on_sokoban.py(1KB)
--------train_nr_wmg_on_pathfinding.py(1022B)
--------test_wmg_on_flat_babyai.py(1KB)
--------test_gru_on_flat_babyai.py(966B)
----LICENSE(1KB)
----README.md(4KB)
----SECURITY.md(3KB)
----data()
--------boxoban-levels-master()
----NOTICE(2KB)
----CODE_OF_CONDUCT.md(444B)
----.gitignore(1KB)
----_config.yml(26B)