theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-23 17:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

update swanlab log

Browse Source
This commit is contained in:
Artiprocher
2025-02-27 19:30:53 +08:00
parent b5c1d33e58
commit a57749ef60
1 changed files with 0 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
examples/wanvideo/train_wan_t2v.py
View File

@@ -432,23 +432,10 @@ def data_process(args):
tile_size=(args.tile_size_height, args.tile_size_width), tile_size=(args.tile_size_height, args.tile_size_width),
tile_stride=(args.tile_stride_height, args.tile_stride_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( trainer = pl.Trainer(
accelerator="gpu", accelerator="gpu",
devices="auto", devices="auto",
default_root_dir=args.output_path, default_root_dir=args.output_path,
logger=[swanlab_logger],
) )
trainer.test(model, dataloader) trainer.test(model, dataloader)