首页 > 其他分享 >torch的用法总结

torch的用法总结

时间:2023-11-21 23:33:33浏览次数:32  
标签:总结 nn 模型 torch 张量 用法 GPU model

张量操作(Tensors):

创建张量(列表、numpy、0、1)

张量操作(改变形状、索引切片、)

数学操作(加 乘)

查看形状

Autograd(自动求导):

定义可训练参数:torch.nn.Parameter
自动求导上下文管理器:torch.autograd
损失函数:torch.nn.functional 中的函数,如 torch.nn.functional.cross_entropy

神经网络构建:

定义模型类:继承自 torch.nn.Module
定义前向传播方法:forward
层(Layer):如 torch.nn.Linear, torch.nn.Conv2d, torch.nn.ReLU 等

优化器(Optimizers):

梯度下降优化器:torch.optim.SGD, torch.optim.Adam 等
参数更新:optimizer.step()

数据处理:

数据加载器(DataLoader):torch.utils.data.DataLoader
数据集(Dataset):torch.utils.data.Dataset
转换(Transforms):torchvision.transforms

模型训练:

循环迭代训练数据:for epoch in range(num_epochs):
前向传播、损失计算、反向传播、参数更新

模型保存和加载:

保存模型:torch.save(model.state_dict(), 'model.pth')
加载模型:model.load_state_dict(torch.load('model.pth'))

GPU加速:

将模型移到GPU:model.to('cuda')
在GPU上执行张量操作:tensor.cuda()

标签:总结,nn,模型,torch,张量,用法,GPU,model
From: https://www.cnblogs.com/mxleader/p/17847900.html

相关文章

  • 每日总结2023年11月21日
    node:internal/crypto/hash:69  this[kHandle]=new_Hash(algorithm,xofLen);          ^ Error:error:0308010C:digitalenveloperoutines::unsupported这是我从GitHub上下载的项目,在运行前端vue时出现的问题,在网上搜了一下有两个方法,一个......
  • 2023-2024-2 20232404 《网络空间安全导论》第2周学习总结
    教材学习内容总结密码学概述总结:1代换和置换,构造现代成对密码算法最重要核心;2香农是当代密码及信息学的集大成者,大多有关学说无法跳出其框架。密码学基本概念总结:1、RSA是第一个既能用于数据加密也能用于数字签名的算法;2与数学联系紧密,且中国也作出了一定贡献。密码学新......
  • 存储云服务中云硬盘(EVS)的一些归纳与总结
    一、概念即类似于磁盘,一块一块的进行存储。(支持加密,解密)它需要挂载在ECS(弹性云服务器)或者BMS(裸金属服务器)或其他的一些主机上才能发挥它的存储的作用。二、相对于普通的硬盘的优势:规格丰富弹性扩展实时监控安全可靠三、EVS的产品架构就是它在现实中的地位(所处位置),我自......
  • 每日总结
    今天继续书写erp的功能。@Overridepublicvoidwages_get(Wageswages){LocalDatecurrentDate=LocalDate.now();//获取月份intmonth=currentDate.getMonthValue();Stringmonth_now=month+"月";wages.setMonth(m......
  • 几种常见的排序算法总结
    常见的几种排序算法排序算法有很多,比较常见的有:冒泡排序、选择排序、插入排序、希尔排序、归并排序、快速排序、堆排序、计数排序、桶排序、基数排序等。并不是所有的都需要会。本文只会对其中部分算法进行总结。冒泡排序冒泡排序是一种比较简单的排序方法。也比较好理解,但是......
  • go errgroup用法
    草稿errgroup就是在waitgroup基础上封装了一层。我还是不喜欢go的这个控制goroutine的方式。只贴一下代码:funcmain(){ eg,ctx:=errgroup.WithContext(context.Background()) fori:=0;i<10;i++{ i:=i select{ case<-ctx.Done(): fmt.Printf("Task......
  • 11.21每日总结
    实验内容与完成情况:(一) 编程实现以下指定功能,并用Hadoop提供的HBaseShell命令完成相同任务:  List列出hbase的表  查看表中的数据并且向指定表中添加数据。  删除列族。  统计表的行数和删除表。(二) HBase数据库操作1.创建表。     ......
  • drf高级用法之——source
    source的用法(三种): 1.修改前端看到字段key的值(自定义key值)name="三国演义"#修改为book_name="三国演义" 使用代码书写:序列化类(serializer.BookSerializer)中书写:book_name=serializers.CharField(source='name') 2.修改前端看到的value值(同时也可以自......
  • 每日总结11.21
    HBase数据库操作(1)createTable(StringtableName,String[]fields)创建表,参数tableName为表的名称,字符串数组fields为存储记录各个字段名称的数组。要求当HBase已经存在名为tableName的表的时候,先删除原有的表,然后再创建新的表。(2)addRecord(StringtableName,Stringrow,Stri......
  • 第12周linux课堂总结
        这周的linux课程我们学习了存储管理,从连接方式上,存储分为以下3种类型,分别是本地存储、外部存储和网络存储,从工作原理上,硬盘分为固态硬盘和机械硬盘,从硬盘接口上,硬盘分为以下几种类型,IDE——SATA硬盘,SCSI——SAS硬盘,其他——PCIe、FC硬盘,SAS是新一代的SCSI技术,SAS硬盘......