Pytorch学习

2020-01-12马春杰杰阅读(37)评论(0)赞(0)

使用pytorch，遇到错误

2020-01-08马春杰杰阅读(23)评论(0)赞(0)

运行pytorch的时候，出现报错： [crayon-5e8e07c73ad99562009287/] 报错原因： 这种情况一般就是pytorch与cuda版本不一致，先查看pytorch支持的cuda的版本： [crayon-5e8e07...

2019-12-23马春杰杰阅读(21)评论(0)赞(0)

原来是： [crayon-5e8e07c73af42052572410/] 这时用的是单GPU，我们需要改为多GPU，需要在model.to(device)之前加一句 [crayon-5e8e07c73af46378347721/] 即可。

2019-12-22马春杰杰阅读(20)评论(0)赞(0)

使用pytorch的时候，想要对网络进行可视化，结果报错为： RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) sho...

2019-12-20马春杰杰阅读(20)评论(0)赞(0)

使用Pytorch进行训练的时候，会出现： [crayon-5e8e07c73b4f0987084571/] 解决方法： 这种情况，一般是因为训练标签的问题。

2019-11-21马春杰杰阅读(24)评论(0)赞(0)

这代表将模型加载到指定设备上。 其中，device=torch.device("cpu")代表的使用cpu，而device=torch.device("cuda")则代表的使用GPU。 当我们指定了设备之后，就需要将模型加载到相应设备中，此...

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

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

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

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

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

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