![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch基础
_天明_
这个作者很懒,什么都没留下…
展开
-
Pytorch中多GPU训练指北
如果我们机子中有很多显卡(例如我们有八张显卡),但我们只想使用0、1、2号显卡,那么我们可以:net = torch.nn.DataParallel(model, device_ids=[0, 1, 2])另一种方法DistributedParallelpython -m torch.distributed.launch --nproc_per_node=你的GPU数量 YOUR_TRAINING_SCRIPT.py (--arg1 --arg2 --arg3 and转载 2020-05-29 15:51:56 · 3546 阅读 · 0 评论 -
用PyTorch训练Imagenet数据集的流程
https://zhuanlan.zhihu.com/p/67919205原创 2020-05-06 15:14:40 · 2332 阅读 · 0 评论 -
参数量,计算量,运行时间测量
现介绍一款自动计算模型参数量、FLOPs、乘加数以及所需内存等数据的轮子:https://github.com/Swall0w/torchstat食用方法如下:1.安装:pip install torchstat2.使用:In [3]: import torchIn [4]: from torchstat import statIn [5]: import torchvision.m...原创 2020-05-03 16:55:32 · 1071 阅读 · 4 评论 -
logger
用Python写代码的时候,在想看的地方写个print xx 就能在控制台上显示打印信息,这样子就能知道它是什么了,但是当我需要看大量的地方或者在一个文件中查看的时候,这时候print就不大方便了,所以Python引入了logging模块来记录我想要的信息。print也可以输入日志,logging相对print来说更好控制输出在哪个地方,怎么输出及控制消息级别来过滤掉那些不需要的信息。这时候,...原创 2020-04-29 15:34:08 · 176 阅读 · 0 评论 -
解释--全连接层输入大小固定
转载 2020-03-25 10:20:13 · 669 阅读 · 0 评论 -
卷积后的图片大小计算
原创 2019-12-02 11:10:31 · 614 阅读 · 0 评论 -
归一化操作的实现
原创 2019-11-14 20:09:10 · 451 阅读 · 0 评论 -
torch.max()用法
torch.max(input) → Tensor返回输入tensor中所有元素的最大值torch.max)(a,0) 返回每一列中最大值的那个元素,且返回索引(返回最大元素在这一列的行索引)torch.max(a,1) 返回每一行中最大值的那个元素,且返回其索引(返回最大元素在这一行的列索引)x: tensor([[0.5285, 0.1247, 0.8332, 0.5485], ...原创 2019-10-24 11:33:33 · 824 阅读 · 0 评论