Pytorch学习

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

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

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

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

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

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