pytorch
文章平均质量分 55
wukurua
摸鱼使者
展开
-
pytorch中的parameter与buffer
先上结论:parameter在反向传播会被optimizer.step更新,buffer在反向传播不会被更新parameter和buffer都被保存在model.state_dict()返回的OrderedDict中(这也是模型保存的对象)模型进行设备移动时,模型中注册的参数(parameter和buffer),即model.state_dict()中的内容会同时进行移动咱来解释一下!文章目录先创建这两种参数吧!1.创建parameter2.创建bufferparameter在反向传播会被`o原创 2021-12-24 17:35:31 · 976 阅读 · 1 评论 -
解析torch.norm函数
官方文档:https://pytorch.org/docs/stable/generated/torch.norm.html?highlight=norm#torch.norm函数:norm(input, p='fro', dim=None, keepdim=False, out=None, dtype=None) Returns the matrix norm or vector norm of a given tensor.参数:input (Tensor) – 输入张量p (.原创 2021-11-26 21:58:07 · 2551 阅读 · 0 评论 -
解决PIL和OpenCV读取显示结果不一致问题
现在我们要读取这张图片使用PIL读取:img_PIL = Image.open(img_path)tensor_trans = transforms.ToTensor()tensor_img = tensor_trans(img_PIL) # 调用ToTensor()的__call__方法使用openCV读取:import cv2img_cv_bgr = cv2.imread(img_path)这里我用tensorboard工具显示图片,出现以下画面:...原创 2021-11-01 23:31:36 · 1326 阅读 · 0 评论 -
python学习两大法宝函数——dir()和help()
参考来源https://www.bilibili.com/video/BV1hE411t7RN?p=4以pytorch为例:把pytorch看作一个工具箱,工具包中有很多小隔间,如图中的1,2,3,4,打开一个小隔间,里面可能又是一些小隔间。也可能是直接能用的工具,如图中3隔间中的a,b,c。那怎么看一个工具箱或者隔间里面有什么东!西?使用dir函数!import torchprint(dir(torch)) # 打开Pytorch包,发现里面有很多小隔间,有一个叫cudaprin.原创 2021-10-31 13:53:18 · 714 阅读 · 0 评论