【文件属性】:
文件名称:retro-contest-sonic:世界模型应用于Open AI Sonic Retro Contest
文件大小:26KB
文件格式:ZIP
更新时间:2021-05-09 16:36:48
reinforcement-learning deep-learning python3 pytorch sonic
声速竞赛
有模型的学生实现的世界模型论文。
正在进行的项目。
去做
正在做
完毕
视觉模型的β-VAE
用于内存模型的MDN-LSTM
控制器模型的CMA-ES
3种模型的训练流水线
人工记录以生成数据
MongoDB存储数据
LSTM和VAE“成功”培训
对CMA-ES给定的一组参数的评估进行多处理
提交学到的代理
长期计划?
更清晰的代码,更加优化和记录
游戏不可知论
继续培训/测试更好的架构
在线培训,而不是使用数据库
如何启动脚本
将模块安装在requirements.txt,pytorch 0.4和mongoDB中
购买或找到Sonic The Hedgehog的ROM,然后将它们与retro-gym一起安装。
完成此步骤后,您将需要训练3个组成部分: python train_vae.py python train_lstm.py --folder=xxx pyth
【文件预览】:
retro-contest-sonic-master
----models()
--------helper.py(5KB)
--------vae.py(3KB)
--------controller.py(400B)
--------lstm.py(2KB)
----const.py(4KB)
----train_vae.py(4KB)
----train_controller.py(7KB)
----test_sonic.py(410B)
----requirements.txt(265B)
----train_lstm.py(5KB)
----.gitignore(62B)
----lib()
--------agent_play.py(4KB)
--------train_utils.py(2KB)
--------visu.py(2KB)
--------dataset.py(2KB)
--------human.py(3KB)
--------controller_utils.py(2KB)
--------env.py(3KB)
--------jerk.py(5KB)
--------play_utils.py(1KB)
----play.py(687B)
----README.md(2KB)
----play_best.py(1KB)
----test_cuda.py(2KB)