![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
知行SUN
这个作者很懒,什么都没留下…
展开
-
动手深度学习Pytorch笔记
Pytorch 学习笔记数据操作数据操作原创 2019-12-13 15:45:17 · 505 阅读 · 0 评论 -
机器学习系列之线性回归
线性回归线性回归的从零开始实现线性回归的简洁实现线性回归的从零开始实现import torchfrom IPython import displayfrom matplotlib import pyplot as pltimport numpy as npimport randomnum_inputs = 2num_examples = 1000w = [2, -3.4]b ...原创 2019-12-13 14:19:09 · 191 阅读 · 0 评论 -
pytorch 计算flops、params和fps
pytorch 计算flops和paramsfrom thop import profileimport torchfrom resnet_18 import Resnet_18, resnet18model = Resnet_18()input = torch.randn(1, 3, 256, 256)flops, params = profile(model, inputs = ...原创 2019-10-22 11:46:34 · 11017 阅读 · 7 评论 -
Pytorch 可视化工具之tensorboard
tensorboard可视化resnet18可视化resnet18#-*-coding:utf-8-*-import torchimport torchvisionfrom torch.autograd import Variablefrom tensorboardX import SummaryWriter# 模拟输入数据input_data = Variable(torch....原创 2019-07-13 10:29:21 · 445 阅读 · 0 评论 -
torch.stack(), torch.cat()用法详解
torch.stack用法详解 针对二维数据dim=0,1,2dim=0dim=1dim=2针对三维数据dim=0,1,2,3dim=0或1 是一样的dim=2dim=3针对二维数据dim=0,1,2dim=0dim=1dim=2针对三维数据dim=0,1,2,3dim=0或1 是一样的dim=2dim=3所以可以看出dim是根据数据的维度从最后一维往前取d,d-...原创 2019-07-09 11:11:36 · 18310 阅读 · 3 评论