文件名称:简单
文件大小:732KB
文件格式:ZIP
更新时间:2024-03-02 18:36:05
Python
SimPLe PyTorch SimPLe算法的PyTorch实现用于Arcade Learning Environment的Atari 2600游戏。 基于论文。 SimPLe从Freeway的4个初始帧中预测出将来的46帧。 安装 该程序使用python 3.7和CUDA 10.2(如果已启用),并已在Ubuntu 20.04.1上进行了测试。 运行以下命令以安装依赖项: pip install torch==1.7.0 gym==0.15.7 gym[atari] opencv-python==4.4.0.42 tqdm==4.49.0 numpy==1.16.4 git clone https://github.com/openai/baselines.git cd baselines pip install -e . 安装wandb(可选) 您可以使用跟踪实验: p
【文件预览】:
SimPLe-main
----atari_utils()
--------atari_utils()
----simple()
--------utils.py(5KB)
--------next_frame_predictor.py(14KB)
--------__init__.py(118B)
--------res()
--------__main__.py(9KB)
--------simulated_env.py(2KB)
--------subproc_vec_env.py(18KB)
--------trainer.py(10KB)
--------adafactor.py(8KB)
----LICENSE(1KB)
----README.md(3KB)
----a2c_ppo_acktr()
--------a2c_ppo_acktr()