PyTorch
PyTorch:python的机器学习框架
特点:
快速进行高维矩阵的计算
自动算微分
三件事情:
model架构
loss function
optimization算法
DataSet:stores data samples and expected values
DataLoader:groups data in batches, enables multiprocessing
1
2
dataset = MyDataset(file)
dataloader = DataLoader(dataset, batch_size, shuffle = True)
This post is licensed under CC BY 4.0 by the author.





