![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
胖琦的pytorch
胖琦今天也在摸鱼
今天不学习,明天变垃圾
展开
-
torch数组计算出现NAN
torch数组计算出现NAN原创 2022-12-11 15:24:14 · 1375 阅读 · 1 评论 -
torch.tensor.scatter_()标签转one-hot
1. 实现代码标签(label)的形状是:(batch_size, 1, H, W) 标签(label)one-hot的形状是:(batch_size, n_classes, H, W)def make_one_hot(labels, classes): one_hot = torch.FloatTensor(labels.size()[0], classes, labels.size()[2], labels.size()[3]).zero_().to(labels.device)原创 2022-03-14 10:59:37 · 2039 阅读 · 0 评论 -
load_state_dict报错: Error(s) in loading state_dict for XXX: Missing keys(s) in state_dict: ....
1. 报错内容/opt/tools/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py in load_state_dict(self, state_dict, strict) 1221 1222 if len(error_msgs) > 0:-> 1223 raise RuntimeError('Error(s) in loading state_dict原创 2022-03-10 14:50:40 · 4602 阅读 · 0 评论 -
from albumentations import torch as AT报错处理
问题最近看kaggle大佬的代码,大家数据扩增(Augmentation)的时候都喜欢用albumentations,但是运行from albumentations import torch as AT会报错呀解决办法去git上面看了看albumentations的源代码,现在这句话应该写成from albumentations import pytorch as AT这样就好...原创 2019-10-10 14:36:50 · 1362 阅读 · 1 评论