Debugged the error that was causing to save the predictions in the te…

…xt file in a faulty way.
kspruthviraj · Jul 13, 2022 · 6115a34 · 6115a34
1 parent 5ef001c
commit 6115a34
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/utils/model_training.py b/utils/model_training.py
@@ -379,10 +379,8 @@ def run_ensemble_prediction_on_unseen(self, test_main, data_loader, name):
 
             output, prob = cls_predict_on_unseen(data_loader.test_dataloader, self.model)
 
-            output = torch.cat(output)
             prob = torch.cat(prob)
 
-            output = output.cpu().numpy()
             prob = prob.cpu().numpy()
 
             Ensemble_prob.append(prob)