pytorch
scut_salmon
这个作者很懒,什么都没留下…
展开
-
torch x = x.view(-1, ...)理解
在torch里面,view函数相当于numpy的reshape,来看几个例子:a = torch.arange(1, 17) # a's shape is (16,)a.view(4, 4) # output belowtensor([[ 1, 2, 3, 4], [ 5, 6, 7, 8], [ 9, 10, 11, 12], ...原创 2018-09-04 22:21:20 · 81217 阅读 · 12 评论 -
torch代码解析 为什么要使用optimizer.zero_grad()
optimizer.zero_grad()意思是把梯度置零,也就是把loss关于weight的导数变成0.在学习pytorch的时候注意到,对于每个batch大都执行了这样的操作: # zero the parameter gradients optimizer.zero_grad() # forward + backward + optim...原创 2018-09-05 09:58:20 · 153251 阅读 · 47 评论 -
pytorch实现upsample
这里实现的是一个比较偏的upsample,是朋友问我的,顺手记录下来。实现的效果如下:输入:输出:实现的是按相同值内插。实现方法是meshgrid,这个函数很好用,可以在输入x坐标和y坐标的情况下,返回由其组成的所有可能坐标组合。具体实现如下:...原创 2019-08-11 16:16:14 · 3452 阅读 · 0 评论