Это мой код.
Я использую 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 сервере