Pytorch学习

Pytorch学习笔记

环境配置

搭建神经网络的三个步骤

1 Define Neural Network定义神经网络 

2  Loss Function定义损失函数

3 Optimization Algorithm优化函数

训练和验证

1 数据读取

#数据读取
#shuffle中,Training：True，Testing：False
dataset=MyDataset(file)
dataloader=DaraLoader(dataset,batch_size,shuffle=True)

#一般的数据集类
from torch.utils.data import Dataset,DataLoader
class MyDataset(Dataset):
    def __init__(self,file):
        self.data= ...
    
    def __getitem__(self,index):
        return self.data[index]
    
    def __len__(self):
        return len(self.data)

Tensors

Pytorch版本的矩阵