python+pytorch+安装
安装过程
resnet34测试:
import torch
import torch.nn as nn
from torch.autograd import Variable
from torchviz import make_dot
from torchvision import models
dummy_input = Variable(torch.rand(1, 3, 224, ...
原创
2019-11-27 21:10:46 ·
272 阅读 ·
0 评论