minimind/trainer
shiguoliang 450885c0d5 feat: 添加 macOS Metal (MPS) 设备支持
通过条件分支实现 CUDA/MPS/CPU 三端兼容:
- 新增 get_default_device() 自动检测最优设备
- 混合精度、GradScaler、DDP 按设备类型条件启用
- CUDA 环境行为完全不变,MPS 作为新增分支

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-07-06 10:54:20 +08:00
..
rollout_engine.py [fix] inference bug 2026-05-03 20:48:48 +08:00
train_agent.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00
train_distillation.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00
train_dpo.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00
train_full_sft.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00
train_grpo.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00
train_lora.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00
train_ppo.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00
train_pretrain.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00
train_tokenizer.py [update] minimind-3 2026-03-25 23:57:45 +08:00
trainer_utils.py feat: 添加 macOS Metal (MPS) 设备支持 2026-07-06 10:54:20 +08:00