gray___的博客

torch数组计算出现NAN

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)

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

问题 最近看kaggle大佬的代码，大家数据扩增（Augmentation）的时候都喜欢用albumentations，但是运行 from albumentations import torch as AT 会报错呀 解决办法 去git上面看了看albumentations的源代码，现在这句话应该写成 from albumentations import pytorch as AT 这样就好...