Update import for torchvision.transformers

microsoft · Jan 16, 2025 · dd047fa · dd047fa
1 parent 018ece5
commit dd047fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/alexnet_model.py b/tests/unit/alexnet_model.py
@@ -84,7 +84,7 @@ def cast_to_half(x):
 
 def cifar_trainset(fp16=False):
     torchvision = pytest.importorskip("torchvision", minversion="0.5.0")
-    import torchvision.transforms as transforms
+    from torchvision import transformers
 
     transform_list = [
         transforms.ToTensor(),