Индикатор выполнения TQDM показывает много ошибок на linux удаленном сервере - PullRequest
0 голосов
/ 22 февраля 2020

Это мой код.

Я использую tqdm при обучении моего net:

with tqdm(total=total_steps) as pbar:
    for epoch in range(epochs):
        model.train()
        epoch_loss=0
        for i,(image,mask) in enumerate(train_dl):
                 pbar.update(1)
                 optimizer.zero_grad()
                 image = image.to(device).float()
                 pre_mask=model(image)
                 pre_mask=pre_mask.to('cpu')
                 pre_mask=torch.sigmoid(pre_mask)
                 mask_loss=smp.utils.losses.DiceLoss()
                 loss=mask_loss(pre_mask,mask)
                 epoch_loss+=loss
                 loss.backward()
                 optimizer.step()

                 pbar.set_description("epochs:%i"%epoch)
                 pbar.set_postfix(loss=loss.item(),str='h')

Однако есть много проблем с отображением индикатора выполнения

это ошибка

Но я не получил эту ошибку, когда использовал этот код на kaggle, тогда как на удаленном Linux сервере

...