Pytorch训练常用方法总结

本文总结了PyTorch深度学习训练过程中的实用技巧,包括TensorBoardX的训练过程可视化、自动求导与反向传播的实现、矩阵运算中的transpose用法以及如何利用钩子保存中间变量。通过这些内容,帮助读者更好地理解和优化PyTorch模型的训练流程。
摘要由CSDN通过智能技术生成

写在前面

该篇博客用来记录深度学习训练过程中的小trick 以及常用的容易犯错的内容 持续更新

训练过程可视化

TensorBoardX 记录训练过程 以及训练过程的分析

Pytorch中自动求导和反向传播

1.生成tensor torch.tensor
由于在pytorch 0.4.0 版本中 tensor 和 Variable 合并 因此不需要在给每个tensor 外部包装一个Variable类型 现在每一个tensor就是一个可以自动求导的Variable 变量
在初始化一个tensor的时候需要注意

a=torch.tensor([[1,2,3],[4,5,6]],dtype=torch.float32,required_grad=True)  # 只有float类型的数据才能够具有导数  
a= torch.Tensor([[1,2,3],[4,5,6]]) 注意tensor 和Tensor 的区别  此时 时不具有dtype 和required_grad属性的

使用 torch.Tensor 是所有其他类型的tensor的基类 其他类型的tensor是torch.Tensor的实例
torch.Tensor 返回的是一个空的tensor
torch.tensor 需要传入data数据
torch.Tensor() 和 torch.tensor(( )) 表示相同的意思

torch.FloatTensor(2,3) 构建一个23 Float类型的张量
torch.DoubleTensor(2,3) 构建一个2
3 Double类型的张量
torch.ByteTensor(2,3) 构建一个23 Byte类型的张量
torch.CharTensor(2,3) 构

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值