![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
LCJhust2012
努力让自己变得更好的学渣佳佳~
展开
-
Pytorch 踩坑记录——RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn
Pytorch中, tensor需要定义成require_grad的类型才能求梯度。可以在定义变量的时候加上“requires_grad=True”或者,某tensor, tensor_X = tensor_X.requires_grad_()原创 2019-07-23 17:45:07 · 23886 阅读 · 5 评论 -
Pytorch Tensorboard踩坑记录——“ TypeError: Cannot handle this data type”
在tensorboard中记录图片,即add_image时一直报错“TypeError: Cannot handle this data type”,尝试过将Image类型的数据转换成numpy.array和torch.tensor,均没有得到解决。最后发现是因为待现显示的图片的shape是(H,W,C), 在pytorch中tensor默认是CHW,而PIL中是HWC。在tensorboardX...原创 2019-07-22 10:47:11 · 17695 阅读 · 1 评论