关于pytorch的一些常见用法总结

以下常见的用法都是在其他博主那学来的,为了方便以后的查阅,现做一个整理。

1.torch.cat (有关tensor的一些处理)

在深度神经网络的设计中,常常需要进行多尺度的特征融合,这时就会涉及到通道的连接,这时就会用到torch.cat

具体参考:

https://blog.csdn.net/qq_39709535/article/details/80803003

https://www.jianshu.com/p/4e57dbe1d281

2.pytorch上下采样函数-interpolate

链接:https://blog.csdn.net/zhaowangbo/article/details/89917884

3.pytorch反卷积操作详解

链接:https://www.jianshu.com/p/01577e86e506

4.Linux系统查看和添加环境变量

链接:https://blog.csdn.net/cym_anhui/article/details/80781539

5.pytorch 保证模型的可重复性

换言之:pytorch代码中加入以下代码的作用。

torch.backends.cudnn.deterministic = True

链接:https://www.cnblogs.com/yongjieShi/p/9614905.html

6.torch.backends.cudnn.benchmark = True 在什么情况下使用?

如果输入模型的所有input-size大小都相同,则使用上述语句可以达到加速训练的效果。如果输入模型的input-size大小不一样,使用上述语句会使得训练效率变低。

7.Pytorch Tensor维度变换

对tenor增加或减少维度,维度变换等操作

https://blog.csdn.net/weicao1990/article/details/93618136

(所以在将数据送入模型前,都会对数据进行处理,使input-size大小相同)

暂时先写那么多,后面再更新。。。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值