kohya-ss / kohya-ss/sd-scripts

resume step calculation is flawed

Open
#1,999 4 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

When calculating the steps when resuming i can't correctly set the steps. If i have previously done 5 epochs with 5000 steps and i set the new "max train epochs" to 10, instead of doing another 5000 steps, it does 10000 steps, so in total 15000 steps. If is set the "max train epoch" to 5, it says that steps need to be greater than initial steps...

I fixed it in my script, but this is definetly a bug.

Contributor guide

No contributing guide indexed for this repository

Research direction

Reproduce resume behavior using the reported 5-epoch, 5000-step case, then trace the resume step calculation in the training scripts. Verify that increasing the limit to 10 epochs schedules only the remaining 5000 steps and that setting it to 5 does not trigger an invalid-step error.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
45/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.