现实洋葱

文章
8
资源
0
加入时间
2年10月21天

Pytorch——DataSet与DataLoader

在使用 pytorch 构建深度学习相关的项目时,通常需要经过【模型结构】-【损失函数定义】-【数据设置】-【训练代码】-【log、验证、可视化与 checkpoints】。其中,【数据设置】往往因为项目/任务的不同,需要自定义合适的DataLoader(数据加载器)。本文即将介绍 torch.utils.data 中的 Dataset 与 Dataloader 的基本用法,以 Unpaire...