GitHub 热门项目：钻石

GitHub 项目：钻石 仓库地址：https://github.com/eloialonso/diamond 星级：2099 | 作者：eloialonso 项目描述：DIAMOND（扩散作为环境梦想的模型）是一种在扩散世界模型中训练的强化学习代理。 NeurIPS 2024 聚焦。 =================================================== 自述文件内容： # 世界建模的扩散：Atari 中的视觉细节很重要（NeurIPS 2024 聚焦） [**TL;DR**] 💎 DIAMOND（扩散作为环境梦想的模型）是一种完全在扩散世界模型中训练的强化学习代理。 🌍 [项目页面](https://diamond-wm.github.io) • 🤓 [论文](https://arxiv.org/pdf/2405.12399) • 𝕏 [Atari 线程](https://x.com/EloiAlonso1/status/1793916382779982120) • 𝕏 [CSGO线程](https://x.com/EloiAlonso1/status/1844803606064611771) • 💬 [Discord](https://discord.gg/74vha5RWPg) 使用 [miniconda](https://docs.anaconda.com/free/miniconda/miniconda-install/) 快速安装以尝试我们的[预训练世界模型](#try)： >```重击 >git 克隆 https://github.com/eloialonso/diamond.git >CD钻石 >conda 创建-n 钻石 python=3.10 >康达激活钻石 >pip install -r 要求.txt >``` 对于 Atari（世界模型 + RL 代理） >```重击 >python src/play.py --pretrained >``` 对于 CSGO（仅限世界模式） >```重击 >git 签出 csgo >python src/play.py >``` 然后按“m”来控制（默认播放策略）！ **警告**：Atari ROM 将与依赖项一起下载，这意味着您承认您拥有使用它们的许可证。 ## 反恐精英：全球攻势 **编辑**：查看 [csgo 分支](https://github.com/eloialonso/diamond/tree/csgo) 来尝试我们的 DIAMOND 世界模型在 *Counter-Strike: Global Offective* 上训练！ ````bash git checkout CSGO python src/play.py ```` > 关于 Apple Silicon 的注意事项，您必须使用以下命令为 MPS 后端启用 CPU 回退 > PYTORCH_ENABLE_MPS_FALLBACK=1 python src/play.py ## 快速链接 - [尝试我们的可玩扩散世界模型](#try) - [启动训练](#launch) - [配置](#configuration) - [可视化](#可视化) - [播放模式（默认）](#play_mode) - [数据集模式（添加 `-d`）](#dataset_mode) - [其他选项，播放/数据集模式通用](#other_options) - [运行文件夹结构](#结构) - [结果](#结果) - [引文](#引文) - [制作人员](#制作人员) ## [⬆️](#quick_links) 尝试我们的可玩扩散世界模型 ````bash python src/play.py --pretrained ```` 然后选择一个游戏，在 Atari 100k 上预训练的世界模型和策略将从我们的 [Hugging Face Hub 上的存储库 🤗](https://huggingface.co/eloialonso/diamond) 下载并缓存在您的计算机上。 您可能想尝试的一些事情： - 按“m”更改代理和人员之间的策略（默认情况下正在播放该策略）。 - 按`↑/↓`