上传者: 42154650
|
上传时间: 2022-02-24 09:40:53
|
文件大小: 63.17MB
|
文件类型: -
DRQN-张量流
使用Tensorflow,openai / gym和openai / retro进行深度递归Q学习
该存储库包含用于在 Atari和环境上训练DQN或DRQN的代码。
请注意,到目前为止,在Retro环境中的训练都是完全实验性的,必须包装这些环境以将动作空间减少到每个游戏所有动作的更合理的子空间。 当前实现的包装仅对SEGA Sonic环境有意义。
安装
您可以通过发出以下命令来安装所有依赖项:
pip install -r requirements.txt
这将在没有GPU支持的情况下安装Tensorflow。 但是,我强烈建议使用带有GPU支持的Tensorflow,否则培训将花费很长时间。 有关此主题的更多信息,请参见 。 为了运行复古环境,您必须收集要玩的游戏的rom并将其导入: :
跑步
您可以通过以下方式开始培训:
python main.py --