Common tasks in machine learning with pytorch
Working with data
PyTorch has two primitives to work with data:
torch.utils.data.DataLoader and torch.utils.data.Dataset.
Dataset stores the samples and their corresponding labels
DataLoader wraps an iterable around the Dataset.
import torch
from torch i
原创
2022-03-25 21:30:33 ·
75 阅读 ·
0 评论