TorchVision Datasets: Getting Started

Ben Cook • Posted 2021-10-22 • Code
torchvision datasets mnist 3

The TorchVision datasets subpackage is a convenient utility for accessing well-known public image and video datasets. You can use these tools to start training new computer vision models very quickly. TorchVision Datasets Example To get started, all you have to do is import one of the Dataset classes. Then, instantiate … Read more