当前位置:   article > 正文

【pytorch】内容链接汇总

【pytorch】内容链接汇总

pytorch报错部分

pytorch函数详解

链接备注
torch.nn.Linear线性变换
torch.nn.MaxPool2d池化
nn.MultiheadAttention多头注意力
torch.cat()拼接
torch.unsqueeze()和torch.squeeze()增删tensor的维度
with torch.no_grad()反向传播时不会自动求导
torch.utils.data.DataLoader封装数据集
torch.nn.DataParallel使用多卡训练
torch.nn.Module.load_state_dict加载模型权重
model.eval()不启用BN和dropout
torchvision.transforms.Compose组合多个图像变换操作

pytorch教程笔记

基于pytorch的课题研究

链接备注
关于DEKR,SWAHR,HigherHRNet预训练模型的相关研究2D人体姿态估计、预训练模型

pytorch安装部署

链接备注
pytorch,torchvision与cuda版本对应关系官方下载指定版本pytorch
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/482091
推荐阅读
相关标签
  

闽ICP备14008679号