From a57749ef6073f56563e88b0df88d7b4b19de5de8 Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Thu, 27 Feb 2025 19:30:53 +0800 Subject: [PATCH] update swanlab log --- examples/wanvideo/train_wan_t2v.py | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/examples/wanvideo/train_wan_t2v.py b/examples/wanvideo/train_wan_t2v.py index 47ee8ef..7eb0afa 100644 --- a/examples/wanvideo/train_wan_t2v.py +++ b/examples/wanvideo/train_wan_t2v.py @@ -432,23 +432,10 @@ def data_process(args): tile_size=(args.tile_size_height, args.tile_size_width), tile_stride=(args.tile_stride_height, args.tile_stride_width), ) - swanlab_logger = None - if args.use_swanlab: - from swanlab.integration.pytorch_lightning import SwanLabLogger - swanlab_config = {"UPPERFRAMEWORK": "DiffSynth-Studio"} - swanlab_config.update(vars(args)) - swanlab_logger = SwanLabLogger( - project=args.swanlab_project, - name=args.swanlab_name, - config=swanlab_config, - mode=args.swanlab_mode, - logdir=args.swanlab_logdir, - ) trainer = pl.Trainer( accelerator="gpu", devices="auto", default_root_dir=args.output_path, - logger=[swanlab_logger], ) trainer.test(model, dataloader)