Error with the train.ipynb example about sen1floods11 HOT 2 OPEN

Oh, I have encountered the same error. :(

from sen1floods11.

Solved!
Try this in the for loop in the last cell
`for i in range(start, 50):
train_validation_loop(net, optimizer, scheduler, train_loader, valid_loader, 5, i)
epochs.append(i)
x = epochs
plt.plot(x, [l.detach().cpu().numpy() for l in training_losses], label='training losses', color = "#4286f4")
plt.plot(x, [a.detach().cpu().numpy() for a in training_accuracies], label='training accuracy', color = "#0066cc")
plt.plot(x, [iou.detach().cpu().numpy() for iou in training_ious], label='training iou', color = "#003d99")
plt.plot(x, [l.detach().cpu().numpy() for l in valid_losses], label='valid losses', color = "#ff4d4d")
plt.plot(x, [a.detach().cpu().numpy() for a in valid_accuracies],label='valid accuracy', color = "#cc0000")
plt.plot(x, [iou.detach().cpu().numpy() for iou in valid_ious],label='valid iou', color = "#990000")
plt.legend(loc='upper left', bbox_to_anchor=(1.05, 1))
display(plt.show())

print("max valid iou:", max_valid_iou)`

from sen1floods11.