1. 数据加载器与小批量训练(DataLoader & Mini-Batch Training)
- 使用小批量训练(mini-batch training)将大数据集划分为多个小批量,每次仅将一个小批量的数据加载到GPU中进行训练。PyTorch的
DataLoader
可以帮助你实现这一点,它能够在训练过程中动态地加载小批量的数据,并将其送入模型。
示例代码:
from torch.utils.data import DataLoader, Dataset
class MyDataset(Dataset):