
# DanceGRPO 使用说明

## 环境
```
pip install -r requirements.txt
```

## 1. 数据预处理

请先完成数据预处理，生成模型训练和推理所需的 embedding 数据。

```sh
bash preprocess.sh
```

该脚本会调用 `preprocess_flux_embedding.py`，输入 JSON 数据，输出处理后的数据到指定目录。你可以根据需要修改 `--json_file` 和 `--output_dir` 参数。

## 2. 推理

推理脚本用于对测试集进行模型推理，生成结果。

```sh
bash infer_ckpt.sh
```

该脚本会调用 `infer_flux_kontext.py`，并使用 `configs/kontext.nj.yaml` 配置文件。你可以根据需要修改 `flux_path`、`test.save_path` 等参数。

## 3. 训练

训练脚本用于启动模型训练，支持多机多卡分布式训练。



```sh
bash finetune_flux_kontext_grpo.sh
```

训练脚本会调用 `train_grpo_flux_kontext.py`，并设置相关训练参数（如 batch size、学习率、训练步数等）。你可以根据实际需求调整脚本中的参数。

---

如需自定义数据或参数，请参考各脚本中的注释和参数说明，或直接修改 [configs/kontext.nj.yaml](configs/kontext.nj.yaml) 配置文件。