pytorch
pytorch基础与实战
HopeTiger_only1
这个作者很懒,什么都没留下…
展开
-
多特征融合代码的简单实现
【代码】多特征融合代码的简单实现。原创 2023-04-23 11:52:12 · 1272 阅读 · 1 评论 -
Anaconda创建虚拟环境时报错
Anaconda创建虚拟环境时报错:The channel is not accessible or is invalid.channel name: anconda/pkgs/freechannel url: https://mirrors.tuna.tsinghua.edu.cn/anconda/pkgs/freeerror code: 404解决方案:1、按照命令行中的提示,查看自己的channels2、恢复默认源3、查看当前的channels使用命令conda config原创 2021-10-27 09:35:59 · 432 阅读 · 0 评论 -
pytorch搭建网络模型实战-LeNet5
神经网络模型训练步骤1、数据2、模型3、损失函数4、优化器5、迭代训练6、保存模型基本上所有的网络训练过程都可以按这种过程来实现这里以简单的LeNet-5识别手写体数字为例,让大家看到训练的完整流程:import torchfrom torchvision import datasetsfrom torchvision import transformsfrom torch.utils.data import DataLoaderimport torch.nn as nnimpo原创 2022-05-28 16:32:33 · 513 阅读 · 0 评论 -
torch.nn常用命令
模型的创建构建网络:def __init__(self,): pass拼接网络def forward(): passtorch.nn中常用的几种方法nn.Parameter() # 张量子类,表示可学习参数,如weightnn.Module() # 所有网络基类,管理网络属性nn.functional() # 函数的具体实现nn.init() # 参数初始化方法使用nn.Module注意事项:一个module相当于一次运算,必须实现forwar原创 2022-05-28 14:25:53 · 600 阅读 · 0 评论 -
Pytorch常用命令
一、张量处理1、张量的创建1.1torch.tensor(data, dtype=None, device=None, requires_grad=False, pin_memory=False)功能:从data创建tensordata:数据,list或numpydtype:数据类型,默认与data一致device:所在设备requires_grad:是否需要梯度pin_memory:是否存于锁页内存1.2to原创 2022-05-27 21:24:19 · 853 阅读 · 0 评论