github
/
minimind
mirror of https://github.com/jingyaogong/minimind.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
minimind/trainer
History
Jiyang Liu 437eae89d6 [feat] add MPS support for Apple Silicon 
On macOS there's no CUDA, so these trainer scripts fall back to CPU. Detect MPS in the --device default and switch mixed precision to the device-agnostic torch.autocast / torch.amp.GradScaler so it works there.

CUDA and CPU paths are unchanged.
 		2026-08-14 01:26:49 +08:00
..
rollout_engine.py [fix] inference bug 2026-05-03 20:48:48 +08:00
train_agent.py [feat] add MPS support for Apple Silicon 2026-08-14 01:26:49 +08:00
train_distillation.py [feat] add MPS support for Apple Silicon 2026-08-14 01:26:49 +08:00
train_dpo.py [feat] add MPS support for Apple Silicon 2026-08-14 01:26:49 +08:00
train_full_sft.py [feat] add MPS support for Apple Silicon 2026-08-14 01:26:49 +08:00
train_grpo.py [feat] add MPS support for Apple Silicon 2026-08-14 01:26:49 +08:00
train_lora.py [feat] add MPS support for Apple Silicon 2026-08-14 01:26:49 +08:00
train_ppo.py [feat] add MPS support for Apple Silicon 2026-08-14 01:26:49 +08:00
train_pretrain.py [feat] add MPS support for Apple Silicon 2026-08-14 01:26:49 +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