- 博客(2)
- 收藏
- 关注
RSS订阅原创 PyTorch学习（二）
dataformats: 指定图像数据的格式，如 ‘CHW’ 或 ‘HWC’。global_step: 当前步骤的编号，用于在图表中定位数据点。tag: 数据的标签，用于在 TensorBoard 中标识数据。comment: 为日志目录添加注释，方便区分不同的运行结果。log_dir: 指定保存日志的目录，默认为当前目录下的。scalar_value: 要记录的标量值。img_tensor: 要记录的图像张量。global_step: 当前步骤的编号。add_image: 记录图像数据。
2024-08-28 21:54:16
551
原创 PyTorch学习（一）
Dataset类需要实现__init__，__len__和__getitem__三个方法。iter函数将train_dataloader变成一个迭代器，使用next函数可以以此从迭代器中生成一个一个的批次。Dataset是一个抽象类，它的主要作用是封装数据集。你可以将Dataset看作是数据的集合，DataLoader是Dataset的包装器，它的主要作用是提供批量加载数据的能力。功能：返回包含查询对象的所有属性和方法名称的列表。等，它们已经实现了这些方法，可以直接使用。一些内置的Dataset类，比如。
2024-08-27 15:55:26
411
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人