From 778aeb2946fe50ca2d3e8d5511f025f59b026cec Mon Sep 17 00:00:00 2001 From: jingyaogong Date: Tue, 28 Oct 2025 23:34:16 +0800 Subject: [PATCH] [feat] update trainer --- trainer/train_pretrain_vlm.py | 2 +- trainer/train_sft_vlm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/trainer/train_pretrain_vlm.py b/trainer/train_pretrain_vlm.py index 7b361cc..324cddc 100644 --- a/trainer/train_pretrain_vlm.py +++ b/trainer/train_pretrain_vlm.py @@ -16,7 +16,7 @@ from torch.utils.data import DataLoader, DistributedSampler from transformers import AutoTokenizer from model.model_vlm import MiniMindVLM, VLMConfig 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') diff --git a/trainer/train_sft_vlm.py b/trainer/train_sft_vlm.py index f9f6ac4..183a43f 100644 --- a/trainer/train_sft_vlm.py +++ b/trainer/train_sft_vlm.py @@ -16,7 +16,7 @@ from torch.utils.data import DataLoader, DistributedSampler from transformers import AutoTokenizer from model.model_vlm import MiniMindVLM, VLMConfig 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')