Skip to content

Commit f693f4e

Browse files
committed
Merge: [ConvNets/PyT] Fix DALI anti-alias flag
2 parents e976927 + 6fcd5f5 commit f693f4e

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

‎PyTorch/Classification/ConvNets/image_classification/dataloaders.py‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ def __init__(
131131
random_aspect_ratio=[0.75, 4.0/3.0],
132132
random_area=[0.08, 1.0],
133133
num_attempts=100,
134+
antialias=False,
134135
)
135136

136137
self.cmnp=ops.CropMirrorNormalize(
@@ -181,7 +182,10 @@ def __init__(
181182

182183
self.decode=ops.ImageDecoder(device="mixed", output_type=types.RGB)
183184
self.res=ops.Resize(
184-
device="gpu", resize_shorter=size, interp_type=interpolation
185+
device="gpu",
186+
resize_shorter=size,
187+
interp_type=interpolation,
188+
antialias=False,
185189
)
186190
self.cmnp=ops.CropMirrorNormalize(
187191
device="gpu",

0 commit comments

Comments
(0)