Fix

Signed-off-by: Joaquin Anton Guirao <[email protected]>

Author: jantonguirao

docs/examples/use_cases/pytorch/efficientnet/image_classification/dataloaders.py

@@ -896,10 +896,9 @@ def get_impl(
             prefetch_factor=prefetch_factor,
         )
 
-        out_memory_format = memory_format if preprocessing_method == "dali" else None
         return (
             PrefetchedWrapper(
-                train_loader, start_epoch, num_classes, one_hot, memory_format=out_memory_format
+                train_loader, start_epoch, num_classes, one_hot
             ),
             len(train_loader),
         )
@@ -979,10 +978,9 @@ def get_impl(
             prefetch_factor=prefetch_factor,
         )
 
-        out_memory_format = memory_format if preprocessing_method == "dali" else None
         return (
             PrefetchedWrapper(
-                val_loader, 0, num_classes, one_hot, memory_format=out_memory_format
+                val_loader, 0, num_classes, one_hot
             ),
             len(val_loader),
         )