diff --git a/PyTorch/Classification/ConvNets/image_classification/dataloaders.py b/PyTorch/Classification/ConvNets/image_classification/dataloaders.py
index 0249aecc4..3c734dc50 100644
--- a/PyTorch/Classification/ConvNets/image_classification/dataloaders.py
+++ b/PyTorch/Classification/ConvNets/image_classification/dataloaders.py
@@ -162,7 +162,13 @@ def __init__(self, batch_size, num_threads, device_id, data_dir, crop, size):
             random_shuffle=False,
         )
 
-        self.decode = ops.ImageDecoder(device="mixed", output_type=types.RGB)
+        self.decode = ops.ImageDecoder(
+            device="mixed", 
+            output_type=types.RGB,
+            device_memory_padding=117440512,
+            host_memory_padding=8388608,
+        )
+        
         self.res = ops.Resize(device="gpu", resize_shorter=size)
         self.cmnp = ops.CropMirrorNormalize(
             device="gpu",