github pytorch/vision v0.1.6
Add models and modelzoo, some bugfixes

latest releases: v0.18.0, v0.18.0-rc4, v0.18.0-rc3...
7 years ago

New Features

  • Add torchvision.models: Definitions and pre-trained models for common vision models
    • ResNet, AlexNet, VGG models added with downloadable pre-trained weights
  • adding padding to RandomCrop. Also add transforms.Pad
  • Add MNIST dataset

Performance Fixes

  • Fixing performance of LSUN Dataset

Bug Fixes

  • Some Python3 fixes
  • Bug fixes in save_image, add single channel support

Don't miss a new vision release

NewReleases is sending notifications on new releases.