asjc-classification
/

scibert_multilabel_asjc_classifier

@@ -124,7 +124,6 @@ for example_data in all_articles:
     inputs = {key: value.to(device) for key, value in inputs.items()}
     # Make predictions with the model
-    model.eval()            # Set the model to evaluation mode
     with torch.no_grad():   # No gradient computation
         outs = model(**inputs)
         logits = outs[0]    # Raw predictions (logits)

     inputs = {key: value.to(device) for key, value in inputs.items()}
     # Make predictions with the model
     with torch.no_grad():   # No gradient computation
         outs = model(**inputs)
         logits = outs[0]    # Raw predictions (logits)