torch
zhaosuyuan
这个作者很懒,什么都没留下…
展开
-
torch获取每层权重
pytorch读取每层权重。原创 2023-07-05 11:17:07 · 753 阅读 · 0 评论 -
torch转onnx
torch转onnx。原创 2023-07-04 17:53:34 · 107 阅读 · 0 评论 -
pytorch前向inference
import torchfrom PIL import Imagefrom torchvision import transforms as T######数据均值方差预处理transforms = T.Compose([T.Resize(size=(288, 144)),T.ToTensor(),T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])######数据按要求读取path=‘demo.jpg’src = Im原创 2021-04-21 16:52:33 · 465 阅读 · 0 评论 -
torch torchvision cuda python对应版本
torch torchvision cuda python1.7.1 0.8.2 >=3.61.7.0 0.8.1 >=3.61.7.0 0.8.0 >=3.61.6.0 0.7.0 10.2 >=3.61.5.1 0.6.1 10.1 >=3.51.5.0 0.6.0 10.1 >=3.51.4.0 0.5.0 10.1 ==2.7, &原创 2021-04-21 11:42:20 · 2630 阅读 · 1 评论 -
torch 网络参数查看,参数量统计
以LeNet为例:import torch from torchsummary import summary原创 2021-04-20 10:37:52 · 1374 阅读 · 0 评论 -
torch.nn.DataParallel保存,单GPU加载
@TOCtorch.nn.DataParallel保存,单GPU加载model.load_state_dict[checkpoint]1.torch.nn.DataParallel 保存,torch.nn.DataParallel加载保存torch.nn.DataParatllel(model).cuda() 含有mudule参数torch.save(model.state_dict(), ‘demo.pth’)加载torch.nn.DataParatllel(model).cuda()c原创 2021-04-13 10:40:03 · 1834 阅读 · 0 评论