GPU 计算与 CPU 相比能够快多少？在本文中，我将使用 Python 和 PyTorch 线性变换函数对其进行测试。

机器学习或深度学习模型的训练的目标是成为“通用”模型。这就需要模型没有过度拟合训练数据集，或者换句话说，我们的模型对看不见的数据有很好的了解。数据增强也是避免过度拟合的众多方法之一。...

在pytorch中的多GPU训练一般有2种DataParallel（DP）和DistributedDataParallel（DDP） ，DataParallel是最简单的的单机多卡实现，但是它使用多线程模型，并不能够在多机多卡的环境下使用，所以本文将介绍DistributedDataParallel，DDP...

装饰器是 python 上下文管理器的特定实现。本片文章将通过一个pytorch GPU 调试的示例来说明如何使用它们。虽然它可能不适用于所有情况，但我它们却是非常有用。...

工欲善其事必先利其器。做人工智能，必然少不了基础框架。深度学习领域两大主流框架TensorFlow和PyTorch，一个是来自Google，另一个来自于Facebook. 国内其实也有不少框架，百度的Paddle算是做得最早的，除此之外，还有华为的Min...

众所周知，作为深度学习框架之一的 PyTorch 和其他深度学习框架原理几乎完全一致，都有着自动求导机制，当然也可以说成是自动微分机制。有些时候，我们不想要它自带的求导机制，需要在它的基础之上做些扩展，这个时候我们只需借...

如果你使用conda install pytorch torchvision cudatoolkit=9.0 -c pytorch来安装pytorch，并且添加了清华镜像源，但还是由于网络原因下载失败，你只需要把-c pytorch去掉：...

最早是看了matlab的代码，搭了环境，demo也跑了，就再也没碰过了。之后想自己把测试和训练部分全部跑通，找了个用pytorch写的代码，看的过程中发现自己还是很多细节部分不是很清楚。虽然文章写的很一笔带过，但是看着代码会发现...

PyTorch简介 在2017年1月18日，facebook下的Torch7团队宣布PyTorch开源后就引来了剧烈的反响。PyTorch 是 Torch 在 Python 上的衍生版本。Torch 是一个使用 Lua 语言的神经网络库, Torch 很好用, 但是 Lua 流行度不够,...

在对STN的原论文进行了翻译、理解后，我打算去github上运行下源码，以加深对ST的理解。毕竟，talk is cheap，show me the code！