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

update eval

pull/81/merge
jingyaogong 2025-04-27 18:38:51 +08:00
parent 623a130c34
commit a8572190a8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
eval_vlm.py
View File

@ -16,8 +16,8 @@ def count_parameters(model):
def init_model(lm_config, device): def init_model(lm_config, device):
tokenizer = AutoTokenizer.from_pretrained('./model')
if args.load == 0: if args.load == 0:
tokenizer = AutoTokenizer.from_pretrained('./model')
moe_path = '_moe' if args.use_moe else '' moe_path = '_moe' if args.use_moe else ''
modes = {0: 'pretrain_vlm', 1: 'sft_vlm', 2: 'sft_vlm_multi'} modes = {0: 'pretrain_vlm', 1: 'sft_vlm', 2: 'sft_vlm_multi'}
ckp = f'./{args.out_dir}/{modes[args.model_mode]}_{args.hidden_size}{moe_path}.pth' ckp = f'./{args.out_dir}/{modes[args.model_mode]}_{args.hidden_size}{moe_path}.pth'