train: epoch 26, iter:20, loss: 1.2620, lr: 0.0005049999999999998
train: epoch 26, iter:40, loss: 1.2620, lr: 0.0005049999999999998
100%|█████████████████████████████████████████████████████████████| 4/4 [00:09<00:00,  2.26s/it]
val epoch: 26, loss: 1.2497
train: epoch 27, iter:0, loss: 1.2728, lr: 0.0004739186928329897
train: epoch 27, iter:20, loss: 1.2634, lr: 0.0004739186928329897
train: epoch 27, iter:40, loss: 1.2622, lr: 0.0004739186928329897
100%|█████████████████████████████████████████████████████████████| 4/4 [00:09<00:00,  2.30s/it]
val epoch: 27, loss: 1.2498
train: epoch 28, iter:0, loss: 1.2437, lr: 0.0004429600493856692
train: epoch 28, iter:20, loss: 1.2627, lr: 0.0004429600493856692
train: epoch 28, iter:40, loss: 1.2628, lr: 0.0004429600493856692
100%|█████████████████████████████████████████████████████████████| 4/4 [00:08<00:00,  2.21s/it]
val epoch: 28, loss: 1.2497
train: epoch 29, iter:0, loss: 1.2635, lr: 0.0004122462492800661
train: epoch 29, iter:20, loss: 1.2622, lr: 0.0004122462492800661
train: epoch 29, iter:40, loss: 1.2619, lr: 0.0004122462492800661
100%|█████████████████████████████████████████████████████████████| 4/4 [00:08<00:00,  2.23s/it]
val epoch: 29, loss: 1.2496
train: epoch 30, iter:0, loss: 1.2675, lr: 0.0003818985058533967
train: epoch 30, iter:20, loss: 1.2610, lr: 0.0003818985058533967
train: epoch 30, iter:40, loss: 1.2625, lr: 0.0003818985058533967
100%|█████████████████████████████████████████████████████████████| 4/4 [00:09<00:00,  2.25s/it]
Traceback (most recent call last):
  File "/root/EGE-UNet-main11/train.py", line 191, in <module>
    main(config)
  File "/root/EGE-UNet-main11/train.py", line 147, in main
    loss = val_one_epoch(
           ^^^^^^^^^^^^^^
  File "/root/EGE-UNet-main11/engine.py", line 90, in val_one_epoch
    confusion = confusion_matrix(y_true, y_pre)
                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.11/dist-packages/sklearn/utils/_param_validation.py", line 216