Pytorch学习

2019-04-19马春杰杰阅读(269)评论(0)赞(0)

这篇博文主要讲pytorch的基本函数理解。 1 torch.arange(min,max,stride) 类似于python中的range [crayon-5d5968b97403f129483025/] 可以看到，这相当于从1~10，每...

2019-04-07马春杰杰阅读(290)评论(0)赞(0)

使用pytorch的时候，如果我们使用torch.nn.functional中的sigmoid和tanh函数的话 [crayon-5d5968b974342957285066/] 可能会报错： [crayon-5d5968b97434a08...

2019-04-07马春杰杰阅读(315)评论(1)赞(0)

pytorch 进行反向传播有时会报错： [crayon-5d5968b974501651009593/] 这是因为第一次反向传播之后，这个计算图的内存就会被释放掉，这样的话再次进行反向传播就不行了，解决方法就是添加retain_graph...