PRGC After training one epoch, got an error

After training one epoch, got an error

Open KeranLi opened this issue 1 year ago • 5 comments

Epoch 1/100 100%|####################################################################################################| 89/89 [00:27<00:00, 3.25it/s, loss=0.288, loss_mat=0.000, loss_rel=0.000, loss_seq=0.288] 0%| | 0/2 [00:00<?, ?Batch/s] Traceback (most recent call last): File "train.py", line 224, in train_and_evaluate(model, params, ex_params, args.restore_file) File "train.py", line 153, in train_and_evaluate val_metrics = evaluate(model, val_loader, params, ex_params, mark='Val') File "F:\code\PRGC\evaluate.py", line 92, in evaluate ex_params=ex_params) ValueError: not enough values to unpack (expected 4, got 3)

I'm so confused about this error. because I confirmed the output of the evaluate function is 3. The expected 4 is non-sense.