[feat] get params

2025-12-31 20:46:59 +08:00 · 2025-12-31 20:46:59 +08:00 · 9236260a4a
parent 288a1d7212
commit 9236260a4a
1 changed files with 1 additions and 1 deletions
--- a/trainer/trainer_utils.py
+++ b/trainer/trainer_utils.py
@ -12,7 +12,7 @@ import torch
 import torch.distributed as dist
 from torch.utils.data import Sampler
 from transformers import AutoTokenizer
-from model.model_minimind_qwen3 import MiniMindForCausalLM
+from model.model_minimind import MiniMindForCausalLM

 def get_model_params(model, config):
    total = sum(p.numel() for p in model.parameters()) / 1e6