fail on val
Opened this issue · 0 comments
Qu1n-22 commented
Did it fail because it was too inaccurate?
'''
100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 60/60 [00:19<00:00, 3.13it/s]
Traceback (most recent call last):
File "/public2/home/wangchen/1U/VM-UNet-main/train.py", line 223, in
main(config)
File "/public2/home/wangchen/1U/VM-UNet-main/train.py", line 178, in main
loss = val_one_epoch(
File "/public2/home/wangchen/1U/VM-UNet-main/engine.py", line 91, in val_one_epoch
TN, FP, FN, TP = confusion[0,0], confusion[0,1], confusion[1,0], confusion[1,1]
IndexError: index 1 is out of bounds for axis 1 with size 1
'''