Skip to content

Comments

feat: add GPU patch augmentation, dataset preprocessing, and augmenta…#67

Merged
skaliy merged 1 commit intomainfrom
aug-dev
Feb 17, 2026
Merged

feat: add GPU patch augmentation, dataset preprocessing, and augmenta…#67
skaliy merged 1 commit intomainfrom
aug-dev

Conversation

@skaliy
Copy link
Member

@skaliy skaliy commented Feb 17, 2026

…tion refactoring

  • Add GpuPatchAugmentation for GPU-batched augmentation on patch tensors
  • Add preprocess_dataset utility for offline dataset preprocessing with parallel workers
  • Refactor suggest_patch_augmentations to share logic via _compute_patch_aug_params
  • Add gpu_augmentation parameter to MedPatchDataLoader (mutually exclusive with patch_tfms)
  • Add tutorial documentation for GPU augmentation workflow

…tion refactoring

- Add GpuPatchAugmentation for GPU-batched augmentation on patch tensors
- Add preprocess_dataset utility for offline dataset preprocessing with parallel workers
- Refactor suggest_patch_augmentations to share logic via _compute_patch_aug_params
- Add gpu_augmentation parameter to MedPatchDataLoader (mutually exclusive with patch_tfms)
- Add tutorial documentation for GPU augmentation workflow
@skaliy skaliy merged commit 76757cd into main Feb 17, 2026
2 checks passed
@skaliy skaliy deleted the aug-dev branch February 17, 2026 14:02
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