标签：PyTorch

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

赞(13)老唐2021-04-17科研

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

赞(6)老唐2021-02-06科研

Pytorch 是非常注明的机器学习框架，其中的 torch.Tensor 是自带排序的，直接使用 torch.sort() 这个方法即可。排序可以按照升序、降序，可以选择排序的维度，等等。下面介绍一下 Pytorch 中的排序方法。 文章...

赞(0)老唐2020-01-07科研

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

赞(6)老唐2020-01-06科研

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

赞(1)老唐2019-08-17科研

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

赞(3)老唐2019-08-08科研

好久没写 PyTorch 了，记录一下在 PyTorch 中如何让程序使用指定的 GPU，这样能有效的避免多人共用一台服务器的时候互相抢占 GPU 资源。如果不指定，PyTorch 默认会占用所有 GPU，这样是非常不友好的，建议大家都在写...

赞(3)老唐2019-07-25科研