目录
I love pytorch .no session run!
higher level wrappers for working
Define your own nn modules
Dataloader
Pretrained MODELS
import torch
import torchvision
alexnet = torchvision.models.alexnet(pretrained=True)
vgg16 = torchvision.models.vgg16(pretrained=True)
resnet101 = torchvision.models.resnet101(pretrained=True)
Visdom