Update NeuralNet.py

DeepRank · Dec 22, 2021 · 70041ec · 70041ec
1 parent 0a27faa
commit 70041ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeprank/learn/NeuralNet.py b/deeprank/learn/NeuralNet.py
@@ -502,7 +502,7 @@ def load_nn_params(self):
         try:
             self.hit_cutoff = self.state['hit_cutoff']
         except Exception:
-            print(f'No "hit_cutoff" found in {self.pretrained_model}. Please set it in function "test()" when doing benchmark"')
+            logger.warning(f'No "hit_cutoff" found in {self.pretrained_model}. Please set it in function "test()" when doing benchmark"')
 
     def load_data_params(self):
         """Get dataset parameters from a saved model."""