kohya-ss / kohya-ss/sd-scripts

Feature Request: Save N epoch and N steps, comma seperated

Open
#1,727 3 comments 2 reactions 0 assignees View on GitHub
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.