Merge branch 'main' into swanlab-dev

This commit is contained in:
Zhongjie Duan
2025-02-28 16:16:15 +08:00
committed by GitHub
5 changed files with 9 additions and 1 deletions

View File

@@ -269,7 +269,6 @@ def launch_training_task(model, args):
batch_size=args.batch_size,
num_workers=args.dataloader_num_workers
)
# train
trainer = pl.Trainer(
max_epochs=args.max_epochs,