用pytorch踩过的坑

背景

帮师妹改毕设代码，第一次接触pytorch，没看手册，直接开肛，遇到些坑，在这里记录一下，大佬勿喷。

坑1：进入网络训练的数据必须归一化

如果数据没有归一化，可能得到的loss会成为负数，在此参考了Crazy_Omais的一段归一化代码

def data_in_one(inputdata):
    min = np.nanmin(inputdata)
    max = np.nanmax(inputdata)
    outputdata = (inputdata-min)/(max-min)
    return outputdata

坑2：torchvision.transforms.ToTensor()

torchvision.transforms.ToTensor()不能用于处理一维数据，如果要处理的话，可以使用torch.from_numpy()

def __getitem__(self, index):
    data = self.datas[:]