Loss Calculation Problem

[clkinder1]

I'm having a problem with the trainer where:

Traceback (most recent call last): File "C:\Users\Carsten Kinderknecht\Documents\GitHub\pytorch-lanenet\lanenet\train.py", line 156, in <module> main() File "C:\Users\Carsten Kinderknecht\Documents\GitHub\pytorch-lanenet\lanenet\train.py", line 144, in main train_iou = train(train_loader, model, optimizer, epoch) File "C:\Users\Carsten Kinderknecht\Documents\GitHub\pytorch-lanenet\lanenet\train.py", line 68, in train total_loss, binary_loss, instance_loss, out, train_iou = compute_loss(net_output, binary_label, instance_label) File "C:\Users\Carsten Kinderknecht\AppData\Local\Programs\Python\Python310\lib\site-packages\lanenet-0.1.0-py3.10.egg\lanenet\model\model.py", line 75, in compute_loss File "C:\Users\Carsten Kinderknecht\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\Carsten Kinderknecht\AppData\Local\Programs\Python\Python310\lib\site-packages\lanenet-0.1.0-py3.10.egg\lanenet\model\loss.py", line 33, in forward File "C:\Users\Carsten Kinderknecht\AppData\Local\Programs\Python\Python310\lib\site-packages\lanenet-0.1.0-py3.10.egg\lanenet\model\loss.py", line 71, in _discriminative_loss File "C:\Users\Carsten Kinderknecht\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\functional.py", line 1472, in norm return _VF.frobenius_norm(input, _dim, keepdim=keepdim) IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

Any help would be appreciated