PyTorch
文章平均质量分 61
Deepsdu
减肥是人类进步的阶梯,idushens@163.com
展开
-
【论文阅读】OSTrack:Joint Feature Learning and Relation Modeling for Tracking A One-Stream Framework
Ostrack 文章阅读原创 2023-02-12 17:58:45 · 747 阅读 · 2 评论 -
【PyTorch】神经网络工具箱nn
神经网络工具箱nn核心数据结构式Module 表示层\表示一个包含很多层的神经网络Python类的 参数过程nn.Conv2d() 对由几个输入平面组成的输入信号应用二维卷积nn.AvgPool2d() 平均池化nn.ReLU() 激活函数 Sequential() 简化的前馈传播网络 例子中第二种较为简单nn.BatchNorm2d()RNN (vanilla RNN LSTM GRU 以及三种cell)nn.CrossEntropyLoss()交叉熵损失 与普通的layer相同原创 2021-01-29 00:02:28 · 539 阅读 · 0 评论 -
【PyTorch】Autograd
Autogradrequires_grad 需要求导(requires_grad)的tensor即Variable 变量的requires_grad属性默认为False,如果某一个节点requires_grad被设置为True,那么所有依赖它的节点requires_grad都是True。grad_fn 查看反向传播的函数的类型 AddBackward0/MulBackward0next_functions保存grad_fn的输入,是一个tuple,tuple的元素也是Func原创 2021-01-27 23:53:20 · 119 阅读 · 0 评论 -
【PyTorch】---Tensor
常用Tensor用法小结生成 tensortensor(*size) tensor(data)arange(s,e,step) #s-e 步长为steplinspace(s,e,steps) #s-e 均匀steps份rand/randn(size) #均匀/标准分布normal(mean,std)/uniform(from,to)#正态/均匀分布randperm(m) #随机排列 生成0- m-1 的随即排列2. 常用操作tensor.shape=tensor.si原创 2021-01-27 23:47:46 · 174 阅读 · 0 评论