torchvisionEasy Playground

PyTorch datasets, model architectures, and transforms for vision

Getting started with torchvisionRun locally
Install
pip install torchvision
Python CodeRun locally
Expected Output
# Expected output shown below
# (Run locally with: torchvision)

torchvision is a third-party package. PyTorch datasets, model architectures, and transforms for vision. Install with: pip install torchvision

Challenge

Try modifying the code above to explore different behaviors. Can you extend the example to handle a new use case?