From 2611a2226479d1e58b0e893355c3486951a629b8 Mon Sep 17 00:00:00 2001
From: zxsanny <zxsanny@gmail.com>
Date: Wed, 5 Mar 2025 10:48:45 +0200
Subject: [PATCH] GPU augmentation. try2

---
 preprocessing-cuda.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/preprocessing-cuda.py b/preprocessing-cuda.py
index 08603cd..f58fb27 100644
--- a/preprocessing-cuda.py
+++ b/preprocessing-cuda.py
@@ -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