diff --git a/image_ref/main.py b/image_ref/main.py index ed7d6d17ab73c1a983b22b3d2f3228b8ef59d351..2368cdaff12042c31c8cdc44655581368fed1e43 100644 --- a/image_ref/main.py +++ b/image_ref/main.py @@ -238,8 +238,6 @@ def make_prediction_duo(model, data, f_name, f_name2): label = label.cuda() output = model(imaer, imana, img_ref) confidence = soft_max(output) - print(label) - print(confidence) confidence_pred_list[specie].append(confidence[:, 0].data.cpu().numpy()) # Mono class output (only most postive paire) output = torch.argmax(confidence[:, 0])