Lightning-AI / Lightning-AI/pytorch-lightning
Custom batch selection for logging
Open
@SkafteNicki is already working on this.
Since Sep 13, 2025.
feature
logging
- Dominant language
- Python
- Stars
- 31.4k
- Forks
- 3.8k
- Avg merge
- 6d 7h
- Merged PRs (30d)
- 6
Description
### Description & Motivation
Need to be able to select the same batch in every logging cycle. For generation pipelines similar to stable diffusion it is very hard to gauge the performance over training if we continue to choose random batches.
### Pitch
User should have selective ability to choose the batch to log which will be constant for all the logging cycles.
### Alternatives
Its possible to load the data again in train_btach_end() or validation_batch_end(), and call logging.
### Additional context
_No response_
cc @lantiga @borda
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.