PyTorch

PyTorch PPO 源码解读 (pytorch-a2c-ppo-acktr-gail)

作者

论文告一段落，今天开始会陆续整理一下之前论文用到的一些代码，做一个后续整理工作，以备之后有需要的时候再用。本文整理一下 PyTorch PPO 源码解读，这份解读对快速理解 PPO 代码的帮助还是挺大的，之前了解过 PPO 但是还没有写过代 […]

Transformer 输入输出维度以及 Pytorch nn.Transformer 记录

作者

这几天花了不少时间在看 Transformer，正好不知道更新什么，就在此记录一下吧。其实我并不想对于 Transformer 这个东西进行深究，只想简单的知道这个东西做了什么事情，有什么优势，以及怎么使用。网上找文档的时候发现基本上所有文 […]

Pytorch 同一个网络同时拥有多个输出的写法

作者

Pytorch 是非常热门的神经网络计算框架，一般来说一个网络对应一个输出，如果我们需要一个网络同时输出多个 output，这在 Pytorch 也是可以做到的。输出之后，可以分别计算 Loss，然后进行 BP。 一、同一个网络输出多个结果 […]

使用 PyTorch 测试 CUDA 是否安装成功

作者

由于老唐之前都是使用同学装好的 GPU 服务器的，一直没有自己装过，今天自己装了一下，发现还挺复杂的。本文先记录一下如何使用 PyTorch 测试 CUDA 是否安装成功，测试 CUDA 是否安装成功还是挺简单的，由于一直使用 PyTorc […]

李宏毅机器学习笔记 (LeeML-Notes) 最全整理 GitHub 和在线阅读

作者

今天老唐看到 GitHub 上有人把李宏毅的机器学习笔记全部整理了出来，随便打开一个看了看是真的非常详细，基本上把讲课内容全都整理下来了，对于不喜欢看视频，而喜欢看文字记录的朋友来说是个非常不错的选择。在此感谢 @Datawhale 的朋友 […]