logo
0
0
WeChat Login
FHfanshu<3357264605@qq.com>
Refactor trainer modules and simplify WebUI config editor

Anima Trainer v1.02(中文说明)

英文 README: README.md

项目简介

Anima(Cosmos-Predict2)LoRA/LoKr 训练脚本(Windows)。

  • 训练入口为 anima_train.py,功能按职责拆分到 utils/*
  • 学习率调度与 Hugging Face warmup/decay 语义对齐(constant/constant_with_warmup/linear/cosine)。
  • 支持 radam_schedulefree 优化器。

推荐流程

两种配置方式,一种训练方式:

  1. 打开 WebUI 或直接编辑 TOML。
  2. 在终端运行训练命令。
  3. TensorBoard 默认自动启用。

方式 A:WebUI 编辑配置

  1. 安装依赖:
install_dependencies.bat
  1. 启动 WebUI:
start_webui.bat
  1. 在 WebUI 编辑并保存配置(保存到 config/save/*.toml)。
  2. WebUI 仅负责配置编辑与校验,训练请在终端执行。

说明:

  • WebUI 会实时显示配置语义错误并阻止保存。
  • 例如 radam_schedulefree + cosine 会被严格报错。

方式 B:直接编辑 TOML

  1. 安装依赖:
install_dependencies.bat
  1. 复制模板到 config/save
copy config\save\anima_lora_config.example.toml config\save\my_train.toml
  1. 修改 config\save\my_train.toml(模型路径、数据路径、输出参数)。
  2. 终端运行训练:
.venv\Scripts\python.exe anima_train.py --config .\config\save\my_train.toml

说明:

  • 训练默认开启 TensorBoard 事件写入(--tensorboard-enabled)。
  • CLI 训练默认自动拉起 TensorBoard(--tensorboard-autostart),默认端口 6006,被占用会自动顺延。
  • 可用 --no-tensorboard-autostart 关闭自动拉起。

说明

  • 仓库不包含模型权重文件。
  • 建议本地配置统一放在 config/save/*.toml
  • 使用 config/save/anima_lora_config.example.toml 作为共享模板。
  • radam_schedulefree 要求 lr_scheduler=constant 且 warmup 为 0(lr_warmup_steps=0lr_warmup_ratio=0)。
  • 训练脚本参考了网上流传实现(含上级目录 v1.01 版本);如有侵权请联系删除。