SLAC 项目安装与使用教程

1. 项目的目录结构及介绍

slac/
├── data/
├── slac/
│   ├── agents/
│   │   └── slac/
│   │       ├── configs/
│   │       ├── examples/
│   │       └── ...
│   ├── ...
├── .gitignore
├── LICENSE
├── README.md
├── requirements.txt
└── setup.py

目录结构介绍

• data/: 用于存放项目的数据文件。
• slac/: 项目的主要代码目录，包含各种模块和子目录。
  • agents/: 包含各种代理的实现，其中 slac/ 是 Stochastic Latent Actor-Critic 的主要实现。
    • configs/: 包含项目的配置文件。
    • examples/: 包含项目的示例代码。
  • 其他文件和目录: 包含项目的其他模块和辅助文件。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目的许可证文件。
• README.md: 项目的介绍和使用说明。
• requirements.txt: 项目的依赖包列表。
• setup.py: 项目的安装脚本。

2. 项目的启动文件介绍

项目的启动文件主要位于 slac/agents/slac/examples/v1/ 目录下，其中 train_eval.py 是主要的启动文件。

train_eval.py

该文件用于启动训练和评估过程。可以通过命令行参数配置不同的实验设置和参数。

CUDA_VISIBLE_DEVICES=0 python slac/agents/slac/examples/v1/train_eval.py \
  --root_dir logs \
  --experiment_name slac \
  --gin_file slac/agents/slac/configs/slac.gin \
  --gin_file slac/agents/slac/configs/dm_control_cheetah_run.gin \
  --gin_param train_eval.gpu_allow_growth=True \
  --gin_param train_eval.sequence_length=8 \
  --gin_param train_eval.action_repeat=2

参数说明

• --root_dir: 指定日志和结果的保存目录。
• --experiment_name: 指定实验的名称。
• --gin_file: 指定使用的配置文件。
• --gin_param: 指定配置文件中的参数。

3. 项目的配置文件介绍

项目的配置文件主要位于 slac/agents/slac/configs/ 目录下，常用的配置文件包括 slac.gin 和 dm_control_cheetah_run.gin。

slac.gin

该配置文件包含了 Stochastic Latent Actor-Critic 的基本配置参数，如学习率、批量大小等。

dm_control_cheetah_run.gin

该配置文件包含了使用 DeepMind Control Suite 进行训练的具体配置参数，如环境设置、动作重复次数等。

配置文件的使用

可以通过 --gin_file 参数指定使用的配置文件，并通过 --gin_param 参数覆盖配置文件中的默认参数。

CUDA_VISIBLE_DEVICES=0 python slac/agents/slac/examples/v1/train_eval.py \
  --root_dir logs \
  --experiment_name slac \
  --gin_file slac/agents/slac/configs/slac.gin \
  --gin_file slac/agents/slac/configs/dm_control_cheetah_run.gin \
  --gin_param train_eval.gpu_allow_growth=True \
  --gin_param train_eval.sequence_length=8 \
  --gin_param train_eval.action_repeat=2

通过这种方式，可以灵活地配置和启动项目的训练和评估过程。