kohya-ss / kohya-ss/sd-scripts

when using multiple gpu, it works weirdly. More safetensor files are created than the set number.

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

Description

my expectation is,
epochs : 10
there are 10 safetensor files for output.

because there are files as same as i set before with 1 gpu.

but when using multiple gpu, More safetensor files are created than the set number.
in my case, 10 epochs with 4 gpus, it makes me 33 safe tensor files.
30 epochs with 4 gpus, 100 safe tensor files.

is it work properly ?

Contributor guide

No contributing guide indexed for this repository

Research direction

Reproduce the report with 10 epochs on 1 GPU and 4 GPUs, then compare the generated safetensor file counts; the issue reports 10 files on 1 GPU versus 33 on 4 GPUs. Trace the multi-GPU checkpoint-saving behavior from the training entry point, and consider the issue done when the output count matches the configured epochs or the behavior is clearly documented.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
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.