用pytorch训练模型时,torch.data.DataLoader生成载入数据集的类,在用enumerate能正常使用,但是套上tqdm就只有当前是第几个的输出,没有一个漂亮的进度条出来
解决方法
for batch_idx, batch in tqdm(enumerate(myDataLoader), total=len(myDataLoader)):
pass
因为在自定义数据集的时候,已经定义了--len--方法,所以这样就可以了
用pytorch训练模型时,torch.data.DataLoader生成载入数据集的类,在用enumerate能正常使用,但是套上tqdm就只有当前是第几个的输出,没有一个漂亮的进度条出来
解决方法
for batch_idx, batch in tqdm(enumerate(myDataLoader), total=len(myDataLoader)):
pass
因为在自定义数据集的时候,已经定义了--len--方法,所以这样就可以了