GPU augmentation. try2

2026-04-22 07:06:36 +00:00 · 2025-03-05 10:48:45 +02:00
parent 5bd519b968
commit 2611a22264
1 changed files with 2 additions and 2 deletions
@@ -62,8 +62,8 @@ class DataLoader:
                aug_image = fn.random_resized_crop(
                    images,
                    device='gpu',
-                    min_scale=0.8,
-                    max_scale=1.0
+                    size=(images.shape[1], images.shape[2]),
+                    random_area=(0.8, 1.0)
                )

                # Apply multiple random augmentations