minimind/trainer
广峰 7227c6a21c chore(args): 支持Mac MPS设备作为运行和训练设备
- 修改eval_llm.py脚本支持MPS设备(macOS GPU加速)
- 修改eval_toolcall.py脚本支持MPS设备
- 修改serve_openai_api.py脚本支持MPS设备
- 修改多个训练脚本(train_agent.py、train_distillation.py、train_dpo.py、train_full_sft.py、train_grpo.py、
  train_lora.py、train_ppo.py、train_pretrain.py)支持MPS设备
- 优先判断CUDA,其次判断MPS,最后回退到CPU设备
- 统一设备选择逻辑以适配不同硬件加速环境
2026-05-16 01:26:42 +08:00
..
rollout_engine.py [fix] inference bug 2026-05-03 20:48:48 +08:00
train_agent.py chore(args): 支持Mac MPS设备作为运行和训练设备 2026-05-16 01:26:42 +08:00
train_distillation.py chore(args): 支持Mac MPS设备作为运行和训练设备 2026-05-16 01:26:42 +08:00
train_dpo.py chore(args): 支持Mac MPS设备作为运行和训练设备 2026-05-16 01:26:42 +08:00
train_full_sft.py chore(args): 支持Mac MPS设备作为运行和训练设备 2026-05-16 01:26:42 +08:00
train_grpo.py chore(args): 支持Mac MPS设备作为运行和训练设备 2026-05-16 01:26:42 +08:00
train_lora.py chore(args): 支持Mac MPS设备作为运行和训练设备 2026-05-16 01:26:42 +08:00
train_ppo.py chore(args): 支持Mac MPS设备作为运行和训练设备 2026-05-16 01:26:42 +08:00
train_pretrain.py chore(args): 支持Mac MPS设备作为运行和训练设备 2026-05-16 01:26:42 +08:00
train_tokenizer.py [update] minimind-3 2026-03-25 23:57:45 +08:00
trainer_utils.py [update] minimind-3 2026-03-25 23:57:45 +08:00