github
/
minimind
mirror of https://github.com/jingyaogong/minimind.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[feat] get params

pull/621/head
jingyaogong 2025-12-31 20:46:59 +08:00
parent 288a1d7212
commit 9236260a4a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trainer/trainer_utils.py
View File

@ -12,7 +12,7 @@ import torch
import torch.distributed as dist import torch.distributed as dist
from torch.utils.data import Sampler from torch.utils.data import Sampler
from transformers import AutoTokenizer from transformers import AutoTokenizer
from model.model_minimind_qwen3 import MiniMindForCausalLM from model.model_minimind import MiniMindForCausalLM
def get_model_params(model, config): def get_model_params(model, config):
total = sum(p.numel() for p in model.parameters()) / 1e6 total = sum(p.numel() for p in model.parameters()) / 1e6