Linaqruf / kohya-trainer

Adapted from https://note.com/kohya_ss/n/nbf7ce8d80f29 for easier cloning
Apache License 2.0
1.83k stars 300 forks source link

Add WandB support to 1.5 training #279

Open xjdeng opened 1 year ago

xjdeng commented 1 year ago

I used the code from your dev branch for wandb training and adapted it to add the ability to track your training in Weights and Biases. I added "Kohya-ss" as a prefix to the project name so it'll be easier to find but feel free to change that if you'd like.