PyTorch的Dataset 和TorchData API的比较

CSDN头条 2年前 (2022) csdnhot
1.4K 0

内容简介:

深度神经网络需要很长时间来训练。训练速度受模型的复杂性、批大小、GPU、训练数据集的大小等因素的影响。
在PyTorch中,torch.utils.data.Dataset和torch.utils.data.DataLoader通常用于加载数据集和生成批处理。但是从版本1.11开始,PyTorch引入了TorchData库,它实现了一种不同的加载数据集的方法。

在本文中,我们将比较数据集比较大的情况下这两两种方法是如何工作的。我们以CelebA和DigiFace1M的面部图像为例。表1显...

查看原文

? PyTorch的Dataset 和TorchData API的比较

版权声明:csdnhot 发表于 2022-12-27 16:52:58。
转载请注明:PyTorch的Dataset 和TorchData API的比较 | 程序员导航网

暂无评论

您必须登录才能参与评论!
立即登录
暂无评论...