pytorch如何加载一个保存的model?

pytorch如何加载一个Zoo的model?比如这个:https://s3.amazonaws.com/pytor ... pth51,并且移除这个model文件?
已邀请:
匿名用户

匿名用户

赞同来自:

打开你要引用的model,然后使用 params = torch.load('resnet18-5c106cde.pth').
来获取一个 ResNet-18并且安装他,你应该使用 res18 = torchvision.models.resnet18(pretrained=True).
来移除一个随机图片, 你可以定义一个FloatTensor,封装在一个变量并将其发送到网络。
x = torch.rand(1, 3, 224, 224)
xVar = torch.autograd.Variable(x)
res18(xVar)

-->

Variable containing:
-0.4374 -0.3994 -0.5249 ... -0.5333 1.4113 0.9452
[torch.FloatTensor of size 1x1000]

要回复问题请先登录注册