kohya-ss / kohya-ss/sd-scripts
Feature Request: Save N epoch and N steps, comma seperated
- Dominant language
- Python
- Stars
- 7.2k
- Forks
- 1.2k
- Avg merge
- 11m
- Merged PRs (30d)
- 2
Description
We need to be able to save only certain epochs and steps
Like save epoch 30,35,40,45 and no others
or
Save step 300,400,500 and no others
Can you please add this option? Thank you @kohya-ss
This became super important for FLUX training since each checkpoint is 24 GB
This is for saving checkpoints but saving state option this way would be nice as well
Contributor guide
No contributing guide indexed for this repository
Research direction
The issue names no files, tests, or entry points. Locate the checkpoint-saving and training-state-saving paths first, then trace how epoch and step intervals are handled; done means supporting comma-separated epoch or step selections without saving other requested checkpoints, with coverage for both checkpoint and state-saving behavior.
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
- Mostly clear
- Newbie friendliness
- 30/100