You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The parameter val_loader2 passed in when using acc, loss_val, precision, recall, auc, sp = evaluate2(val_loader2, net, epoch=epoch_up) in line 271 has a batch_size of 1. Therefore, in line 245, since p has only 1 dimension, it cannot calculate precision_.compute()[1].item(), recall_.compute()[1].item() and reports an error
The text was updated successfully, but these errors were encountered:
The parameter val_loader2 passed in when using
acc, loss_val, precision, recall, auc, sp = evaluate2(val_loader2, net, epoch=epoch_up)
in line 271 has a batch_size of 1. Therefore, in line 245, since p has only 1 dimension, it cannot calculateprecision_.compute()[1].item(), recall_.compute()[1].item()
and reports an errorThe text was updated successfully, but these errors were encountered: