用pytorch训练模型时,torch.data.DataLoader生成载入数据集的类,在用enumerate能正常使用,但是套上tqdm就只有当前是第几个的输出,没有一个漂亮的进度条出来

解决方法

for batch_idx, batch in tqdm(enumerate(myDataLoader), total=len(myDataLoader)):
    pass

因为在自定义数据集的时候,已经定义了--len--方法,所以这样就可以了