Post

PyTorch

PyTorch:python的机器学习框架

特点:

  1. 快速进行高维矩阵的计算

  2. 自动算微分

截屏2026-01-18 15.31.23

三件事情:

model架构

loss function

optimization算法

截屏2026-01-19 11.07.50

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)

截屏2026-01-19 11.12.52

截屏2026-01-19 11.16.32

截屏2026-01-19 11.18.07

截屏2026-01-19 11.20.30

This post is licensed under CC BY 4.0 by the author.

Trending Tags