torch学习
White Jiang
这个作者很懒,什么都没留下…
展开
-
pytorch模型转换为torch2trt模型
安装TensorRT安装torch2trt转换模型torch转torch2trt减少显存占用,建议在模型转换完成后,将模型保存,使用时直接加载转换后的模型。模型不使用.eval(),可能会报[TRT] [E] Could not implicitly convert NumPy data type: i64 to TensorRT.错误model = torchvision.models.mobilenet_v2(pretrained=True).eval().cuda()x = torch.o原创 2022-04-04 17:04:46 · 2038 阅读 · 2 评论 -
自定义网络后加载相同网络结构的预训练参数
网络结构相同但参数关键字不同如提示以下错误:Missing key(s) in state_dict: xxxxxxxxxxUnexpected key(s) in state_dict:xxxxxxxxxx# 自定义模型的参数字典model_dict = model.state_dict()# 模型参数的关键字列表model_keys = []# 将模型的关键字添加到列表中for k, v in model_dict.items(): model_keys.append(k)# 我复转载 2022-03-09 21:14:42 · 164 阅读 · 0 评论 -
pytorch学习
pytorch学习torch.from_numpyTensorDatasetDataLoadertorch.from_numpy作用:获取一个大小与numpy数组相同的张量用法:npa = numpy.array([1, 0, 1])torch_tensor = torch.from_numpy(npa)TensorDatasetTensorDataset实现对tensor的打包,通过tensor的第一维度进行索引,因此输入的两个tensor的第一维度必须相等,我一般用来他将data和la原创 2021-11-23 15:02:40 · 1196 阅读 · 1 评论