pytorch学习记录

net = LeNet()   # 实例化一个类模型
import ipdb; ipdb.set_trace()
print(net)
net.load_state_dict(torch.load('Lenet.pth'))  # 加载模型数据

## tensor变量转numpy
im = Image.open('cat.jpg')
im = transform(im)  # [C, H, W]
im.numpy()  ## tensor变量转numpy 

## numpy转tensor
sub_ts = torch.from_numpy(sub_img) 

下图是pytorch目前包含的数据集

https://github.com/pytorch/vision/tree/master/torchvision/datasets