pytroch dataloader配合tqdm

代码如下：

from tqdm import tqdm
from torch.utils.data import DataLoadaer
with tqdm(dataloader:DataLoader) as iterator:
            for x, y in iterator:
                x, y = x.to(device), y.to(device)
                # 开始训练或者预测
       

参考链接：https://github.com/qubvel/segmentation_models.pytorch/blob/master/segmentation_models_pytorch/utils/train.py