WandB custom logging--which py script do I edit?

[rafstahelin]

I am trying to make some config/log changes to the wandb API call. I am just not sure where kohya_ss training does this, on which py file. Can anyone help?

I will edit this code:
if args.log_wandb:
wandb.init(
project=args.project,
name=args.run_name,
entity=args.wandb_entity,
config=args_to_dict(args),
resume=args.resume_from_wandb,
)

thanks