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

[feat] update trainer

pull/107/head
jingyaogong 2025-10-28 23:34:16 +08:00
parent 2ce5c31170
commit 778aeb2946
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trainer/train_pretrain_vlm.py
View File

@ -16,7 +16,7 @@ from torch.utils.data import DataLoader, DistributedSampler
from transformers import AutoTokenizer from transformers import AutoTokenizer
from model.model_vlm import MiniMindVLM, VLMConfig from model.model_vlm import MiniMindVLM, VLMConfig
from dataset.lm_dataset import VLMDataset from dataset.lm_dataset import VLMDataset
from trainer.trainer_utils import * from trainer.trainer_utils import get_lr, Logger, is_main_process, init_distributed_mode, setup_seed, init_vlm_model, vlm_checkpoint, SkipBatchSampler
warnings.filterwarnings('ignore') warnings.filterwarnings('ignore')

2
trainer/train_sft_vlm.py
View File

@ -16,7 +16,7 @@ from torch.utils.data import DataLoader, DistributedSampler
from transformers import AutoTokenizer from transformers import AutoTokenizer
from model.model_vlm import MiniMindVLM, VLMConfig from model.model_vlm import MiniMindVLM, VLMConfig
from dataset.lm_dataset import VLMDataset from dataset.lm_dataset import VLMDataset
from trainer.trainer_utils import * from trainer.trainer_utils import get_lr, Logger, is_main_process, init_distributed_mode, setup_seed, init_vlm_model, vlm_checkpoint, SkipBatchSampler
warnings.filterwarnings('ignore') warnings.filterwarnings('ignore')