pytroch 官方文档研读
文章平均质量分 59
林丿子轩
计算机硕士,中大数字人实验室~两年,平安科技~9个月,金山软件~cv算法研发工程师
展开
-
pytroch 查看卷积过程中的参数和提取的Feature map ,可视化
import numpy as npimport torch.nn as nnimport torchimport cv2import matplotlib.pyplot as pltclass torch.nn.ReLU(inplace=False) [source]参数: inplace-选择是否在本地运算class Model3(nn.Module): def __init__(self): super(Model3, self).__init__()原创 2022-05-11 20:08:33 · 511 阅读 · 0 评论 -
pytorch nn中的卷积层 与空洞卷积
import winntfrom torch.nn import Conv1dimport torch.nn as nnimport torch"""torch.rand(*sizes, out=None) → Tensor 返回一个张量,包含了从区间[0,1)的均匀分布中抽取的一组随机数,形状由可变参数sizes 定义.torch.randn(*sizes, out=None) → Tensor 返回一个张量,包含了从标准正态分布(均值为0,方差为 1,即高斯白噪声)中抽取一组随机数,形状由可变原创 2022-05-11 20:01:35 · 785 阅读 · 0 评论 -
torch中的索引,切片,连接,换位
import numpy as np'''https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/#indexing-slicing-joining-mutating-ops''''https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch/#indexing-slicing-joining-mutating-ops'torch.cat(i转载 2022-05-10 11:31:45 · 1916 阅读 · 0 评论 -
torch 中 模型的保存和恢复
保存和加载模型'''https://pytorch-cn.readthedocs.io/zh/latest/notes/serialization/'''import torchimport torchvisionmodel = torchvision.models.resnet18(pretrained=True)"""---------------保存和加载模型参数(推荐)----------------------------"""torch.save(model.state_dict(转载 2022-05-10 11:31:08 · 329 阅读 · 0 评论