kohya-ss / kohya-ss/sd-scripts

Log loss on wandb

Open
#1,356 3 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
7.2k
Forks
1.2k
Avg merge
11m
Merged PRs (30d)
2

Description

Hi, I have digged multiple issues and tried to check the commit diff related to the recent addition of a .toml to log informations to wandb, however I only get the system informations and sample images sent. Is there a way to also send the loss, or for example current learning rate ?

Thanks very much.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by tracing the recent .toml-related wandb logging addition and compare the existing system-information and sample-image logging with the training values mentioned here. Determine where loss and current learning rate are available, then verify that both appear in wandb alongside the existing information.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.