weixin_43466027的博客

之前跑了好几个模型，都需要保存和加载模型信息。为了省得每次都上网找，开一个帖记录一下。1. 保存模型此处只保存模型参数model = torch.nn.Linear(1, 2)torch.save(model.state_dict(), "./model.pth")2. 加载模型model = torch.nn.Linear(1, 2)model.load_state_dict(torch.load("./model.pth"))...

0x01 GPU如何加速深度学习深度学习的实现似乎很复杂，但是其本质上还是一堆高等代数。常用的运算还是比如矩阵加法和矩阵乘法。比如，我们对一个向量套一个sigmoid函数：x=[x1,...,xn]x = [x_1, ... , x_n]x=[x1​,...,xn​]sigmoid(x)=[sigmoid(x1),...,sigmoid(xn)]sigmoid(x) = [sigmoid(x_1), ... , sigmoid(x_n)]sigmoid(x)=[sigmoid(x1​),...,s

整理自官方文档：https://www.oschina.net/action/GoToLink?url=https%3A%2F%2Fpytorch.org%2Fdocs%2Fstable%2Fgenerated%2Ftorch.gather.html%3Fhighlight%3Dgather%23torch.gather0x01 背景在做强化学习DDQN实（CtrlC）验（CtrlV）的时候遇到了一些不是很理解的函数，这里做一些解读。0x02 gather函数gather，直译为聚合，聚集。先.