最近在用dataloader写数据集,在使用tqdm的时候遇见了一些问题,经过查找大量的资料,总结一个简单的方法。
首先,先设置网络的输入和输出,假设这两个量已经是tensor类型了。
- 输入:
tensor_x - 输出:
tensor_y
那么导入这个库函数
from torch.utils.data import DataLoader, TensorDataset
dataset = TensorDataset(tensor_x, tensor_y)
loader = DataLoader(dataset=dataset,batch_size=32,shuffle=False,num_workers=2,pin_memory=True)
接下来,开始写循环
for i in range(10):
with tqdm(total=len(loader)) as t:
for step, (batch_x, batch_y) in enumerate

本文介绍了如何在使用PyTorch DataLoader加载数据时,利用tqdm库优雅地跟踪训练进度,包括设置数据集、配置Dataloader并实现带有进度条的循环。
1万+

被折叠的 条评论
为什么被折叠?



