From fc64a31c0a70c5618432f03a231e7ec8c8d1e631 Mon Sep 17 00:00:00 2001 From: forXuyx Date: Sat, 22 Mar 2025 23:31:31 +0800 Subject: [PATCH] =?UTF-8?q?=E4=B8=80=E4=B8=AA=E5=B0=8F=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- train_pretrain_videolm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_pretrain_videolm.py b/train_pretrain_videolm.py index eb90b0b..b75de3e 100644 --- a/train_pretrain_videolm.py +++ b/train_pretrain_videolm.py @@ -86,7 +86,7 @@ def train_epoch(epoch, wandb): if (step + 1) % args.save_interval == 0 and (not ddp or dist.get_rank() == 0): model.eval() moe_path = '_moe' if model_config.use_moe else '' - ckp = f'{args.save_dir}/pretrain_vlm_{model_config.dim}{moe_path}.pth' + ckp = f'{args.save_dir}/pretrain_videolm_{model_config.dim}{moe_path}.pth' if isinstance(model, torch.nn.parallel.DistributedDataParallel): state_dict = model.module.state_dict() else: