Skip to content

Commit ff0b4ab

Browse files
authored
Added memory_padding to ValPipe
1 parent 0cbadd7 commit ff0b4ab

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

PyTorch/Classification/ConvNets/image_classification/dataloaders.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,13 @@ def __init__(self, batch_size, num_threads, device_id, data_dir, crop, size):
162162
random_shuffle=False,
163163
)
164164

165-
self.decode = ops.ImageDecoder(device="mixed", output_type=types.RGB)
165+
self.decode = ops.ImageDecoder(
166+
device="mixed",
167+
output_type=types.RGB,
168+
device_memory_padding=117440512,
169+
host_memory_padding=8388608,
170+
)
171+
166172
self.res = ops.Resize(device="gpu", resize_shorter=size)
167173
self.cmnp = ops.CropMirrorNormalize(
168174
device="gpu",

0 commit comments

Comments
 (0)