Pytorch
Elvirangel
一颗上进的心
展开
-
神经网络中参数数量的计算和输出尺寸的计算
AlexNet网络中张量(图像)尺寸和参数数量卷积神经网络的参数计算神经网络中参数数量的计算在实验中,经常需要计算网络的参数量和预算量。参考:https://github.com/nmhkahn/torchsummaryX运算量和参数量的计算代码 # Calculate params & FLOPs from torchsummaryX import summary dummy_input = torch.zeros(1, 3, 128, 128).c原创 2020-08-31 17:17:49 · 782 阅读 · 0 评论 -
Pytorch 训练的注意点,坑
Pytorch将模型加载到GPU中训练时遇到的坑1. 第一个容易遇到的坑: 你的target标签字典中含有list等非tensor的数据类型,这时候不处理会报错如下:(1)'dict' object has no attribute 'cuda'(2)'list' object has no attribute 'cuda'解决方法:(1)https://blog.csdn.net/york1996/article/details/103164696(2)将list堆叠...转载 2020-08-19 19:58:11 · 1357 阅读 · 0 评论 -
torch常用基础函数
torch常用基础函数https://www.jianshu.com/p/d678c5e44a6btorch本笔记引用自PyTorch中文文档包torch包含了多维疑是的数据结构及基于其上的多种数学操作。1. 张量Tensorstorch.is_tensor(obj):如果obj是一个pytorch张量,则返回Truetorch.is_storage(obj):如果obj是一个pytorch storage对象,则返回Truetorch.numel(inpu.转载 2020-05-20 22:45:30 · 1576 阅读 · 0 评论 -
PyTorch中的Variable变量详解
PyTorch中的Variable变量详解一、了解Variable顾名思义,Variable就是 变量 的意思。实质上也就是可以变化的量,区别于int变量,它是一种可以变化的变量,这正好就符合了反向传播,参数更新的属性。具体来说,在pytorch中的Variable就是一个存放会变化值的地理位置,里面的值会不停发生片花,就像一个装鸡蛋的篮子,鸡蛋数会不断发生变化。那谁是里面的鸡蛋...原创 2020-02-23 16:24:18 · 1075 阅读 · 0 评论 -
pytorch 方法名后加下划线,如 clamp() 与 clamp_()
任何可以改变tensor内容的操作都会在方法名后加一个下划线'_'例如:x.copy_(y), x.t_(), 这俩都会改变x的值在pytorch中,我们需要注意,对于tensor的操作时候是in-place类型。in-place类型是指,但在一个tensor上操作了之后,是直接修改了这个tensor。非in-place类型,是指返回一个新的tensor,而旧的tensor并不修...原创 2020-02-15 20:18:46 · 1107 阅读 · 0 评论 -
PyTorch中 tensor.detach() 和 tensor.data 的区别
PyTorch0.4中,.data 仍保留,但建议使用 .detach(),区别在于:.data 返回和 x 的相同数据 tensor, 但不会加入到x的计算历史里,且require s_grad = False, 这样有些时候是不安全的, 因为 x.data 不能被 autograd 追踪求微分 。.detach() 返回相同数据的 tensor ,且 requires_grad=F...转载 2020-02-15 16:12:49 · 417 阅读 · 0 评论 -
Pytorch——model.train 和 model.eval
转载:https://blog.csdn.net/weixin_42018112/article/details/91403816目录连接(1) 数据处理(2) 搭建和自定义网络(3) 使用训练好的模型测试自己图片(4) 视频数据的处理(5) PyTorch源码修改之增加ConvLSTM层(6) 梯度反向传递(BackPropogate)的理解(7) 模型的训练和测试、保存和加载...转载 2019-10-13 20:06:10 · 575 阅读 · 1 评论 -
Pytorch使用GPU或者CPU训练的切换
代码示例转自: pytorch tips:使pytorch代码在不改动情况在有GPU自动在GPU运行1. 定义训练方式:方法一:device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu")model=model.to(device)x=x.to(device)y=y.to(device)方法二:...原创 2019-10-12 11:41:01 · 11628 阅读 · 0 评论 -
pytorch中Dataloader()中的num_workers设置问题
pytorch中Dataloader()中的num_workers设置问题:如果num_workers的值大于0,要在运行的部分放进__main__()函数里,才不会有错:import numpy as npimport torchfrom torch.autograd import Variableimport torch.nn.functionalimport matplot...原创 2019-09-20 20:23:31 · 13844 阅读 · 9 评论 -
PyTorch官方文档
最近正在学习Pytorch的相关知识,每次查找里面某些函数的参数定义啥的也不太方便,故在这里留下torch的官方文档,以备平时学习总结之需。见:Pytorch英文文档pytorch 中文文档:Pytorch中文文档当然,在pycharm中查找某个函数的具体结构时也可以ctrl+click...原创 2019-09-19 11:24:35 · 815 阅读 · 0 评论