![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 63
云端一散仙
主要在做图像,点云方向,欢迎讨论~
展开
-
ubuntu18.04安装cuda
下载cuDNN Library for Linux,在解压的文件夹中打开终端,文件夹中有cuda文件夹。accept后,将驱动(Driver)取消后,选择Install。查看显卡适配的NVIDIA的驱动。安装recommend推荐的版本。或者直接使用下面命令安装。查看当前cudnn的版本。安装完毕后,需要重启。原创 2023-08-15 14:17:14 · 476 阅读 · 0 评论 -
Pytorch中的hook
由于pytorch的中间变量在完成计算后就会被释放,如果想要获取中间的信息,就需要使用hook相关函数。首先搭建一个深度学习网络指定网络中的conv2层添加hook,hook单独写一个函数来实现相关功能register_hook,对象为Variable对象register_backward_hook,对应反向传播,用来导出指定子模块输入输出张量的梯度,但只可修改输入张量的梯度,输出张量梯度不可修改通过remove函数取消hook...原创 2022-06-16 15:08:51 · 444 阅读 · 0 评论 -
TensorRT
TensorRT Python API ReferenceTorth转为ONNXmodel = model.cuda()model.eval()_, c, h, w = input_sizedummy_input = torch.randn(MAX_BATCH_SIZE, c, h, w, device='cuda')torch.onnx.export(model, dummy_input, "model.onnx", verbose=True, input_names=["input"], o原创 2021-08-26 09:41:54 · 538 阅读 · 0 评论 -
【pytorch】改造model
Mobilenet_v2import torchvision.models as modelsmodel=models.mobilenet_v2()通过查看model的结构,我们需要保留的是model.features,根据需要,修改model.classifier(classifier): Sequential( (0): Dropout(p=0.2, inplace=False) (1): Linear(in_features=1280, out_features=1000.原创 2020-06-19 11:06:59 · 708 阅读 · 0 评论 -
pytorch可视化网络层特征
Inception V3def preprocess_image(cv2im, resize_im=True): """ Processes image for CNNs Args: PIL_img (PIL_img): Image to process resize_im (bool): Resize to 224 or not returns: im_as_var (Pytorch variable): Vari原创 2020-06-12 13:43:37 · 421 阅读 · 0 评论 -
pytorch部分问题
模型参数加载很慢import torchvision.models as modelsinception = models.inception_v3(pretrained=args.pt_imagenet, aux_logits=False)原因根源在于scipy库。torchvision在加载完模型后需要对模型进行参数初始化,用到了scipy中的stats.truncnorm.rvs方法,而该方法在scipy1.4以上版本中速度变慢了100倍,导致torchvision运行时间变长。将scipy原创 2020-06-02 14:16:42 · 606 阅读 · 0 评论