minimind/trainer
baihaochen 49a05a1307 增加mla注意力模型 2026-05-12 18:57:53 +08:00
..
rollout_engine.py [fix] inference bug 2026-05-03 20:48:48 +08:00
train_agent.py [feat] add MLA with decoupled RoPE (DeepSeek-V2 style) 2026-05-12 18:38:47 +08:00
train_distillation.py [feat] add MLA with decoupled RoPE (DeepSeek-V2 style) 2026-05-12 18:38:47 +08:00
train_dpo.py [feat] add MLA with decoupled RoPE (DeepSeek-V2 style) 2026-05-12 18:38:47 +08:00
train_full_sft.py 增加mla注意力模型 2026-05-12 18:57:53 +08:00
train_grpo.py [feat] add MLA with decoupled RoPE (DeepSeek-V2 style) 2026-05-12 18:38:47 +08:00
train_lora.py [feat] add MLA with decoupled RoPE (DeepSeek-V2 style) 2026-05-12 18:38:47 +08:00
train_ppo.py [feat] add MLA with decoupled RoPE (DeepSeek-V2 style) 2026-05-12 18:38:47 +08:00
train_pretrain.py [feat] add MLA with decoupled RoPE (DeepSeek-V2 style) 2026-05-12 18:38:47 +08:00
train_tokenizer.py add Chinese annotations pretrain 2026-05-08 20:44:02 +08:00
trainer_utils.py [feat] add MLA with decoupled RoPE (DeepSeek-V2 style) 2026-05-12 18:38:47 +08:00