Skip to content

docs(python): document uint8 GPU image batches - #10811

Closed
YannByron wants to merge 1 commit into
apache:masterfrom
YannByron:codex/cwd-gpu-image-supply
Closed

YannByron wants to merge 1 commit into
apache:masterfrom
YannByron:codex/cwd-gpu-image-supply

Conversation

@YannByron

Copy link
Copy Markdown
Contributor

Contiguous window training examples currently convert eight-bit images to float32 in DataLoader workers. This documents the existing images_to_tensor(..., return_uint8=True) path and GPU-side normalization so image tensors use one byte per pixel through worker collation and host-to-device transfer. Tests cover multi-frame value equivalence and mixed bit depth, where the uint8 path must not silently normalize high-bit-depth frames.

The change is deliberately limited to the documented training configuration; it does not introduce a new sampler or decoder backend. Real-image A10 measurements show the configuration improves Paimon’s own small CNN training proxy, but the same optimization applies to HDF5, so it is not evidence of a Paimon format advantage or the 30% end-to-end target.

Validation: the affected PyPaimon image/window tests passed before the final documentation wording adjustment; git diff --check passes after rebasing onto current master. The final wording change only moves transform configuration into Dataset construction.

Document the existing eight-bit image path for GPU training and cover
its value range for multi-frame and mixed-depth windows. Configure the
transform when constructing the dataset, preserving checkpoint identity.

Co-Authored-By: Codex <noreply@anthropic.com>
AI-Model: GPT-6
Co-Authored-By: Codex <noreply@ai-tool.com>
AI-Contributed/Feature: 23/23
AI-Contributed/UT: 13/13
@YannByron YannByron closed this Oct 26, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant