ModelScope还是会溢出，会不会跟 transformers、torch 等版本有关？

"参考以下图片
这两处改一下哈
def cfg_modify_fn(cfg):
cfg.train.max_iter = 5 #训练迭代步数
cfg.train.dataloader.batch_size_per_gpu = 1
cfg.train.dataloader.workers_per_gpu = 1
cfg.train.optimizer.lr = 1e-4
cfg.train.lr_scheduler.warmup_step = 1
cfg.train.checkpoint_interval = 5000

cfg.evaluation.interval = 5000
cfg.evaluation.dataloader.batch_size_per_gpu = 1
cfg.evaluation.dataloader.workers_per_gpu = 1

cfg.train.work_dir = './output' #训练输出目录
cfg.train.num_gpus = 0 #训练GPU个数, 0表示不使用GPU, 使用的是CPU
return cfg

trainer = build_trainer(
Trainers.action_detection,
dict(
model_id=model_id,
train_dataset=train_dataset,
test_dataset=train_dataset,
cfg_modify_fn=cfg_modify_fn))
此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”"