2024-11-13 厉飞雨 阅读(11) 评论(0) 赞(2)

在 PyTorch 中，使用 torch.utils.data.DataLoader 类可以实现批量的数据集加载，在我们训练模型中非常常用，其功能也确实比较强度大。由于其参数比较多，我们将会对其用法进行详解。 1. DataLoader 的基本使用 2. DataLoader 的 collate_fn 参数 3. DataLoader 的 sampler 参数 1. Dat...