From ff0b4ab6dff5cc4a9da95cff539f60cf6039e079 Mon Sep 17 00:00:00 2001
From: Lennart <44500208+lbhm@users.noreply.github.com>
Date: Mon, 18 Jan 2021 16:04:25 +0100
Subject: [PATCH] Added memory_padding to ValPipe

---
 .../ConvNets/image_classification/dataloaders.py          | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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",