Fix # 457 (#469)

Straightforward change of `print` call to `logging.warning`
lanpa · Jul 16, 2019 · 97a57c0 · 97a57c0
1 parent dce4fb1
commit 97a57c0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorboardX/x2num.py b/tensorboardX/x2num.py
@@ -3,14 +3,15 @@
 from __future__ import division
 from __future__ import print_function
 
+import logging
 import numpy as np
 import six
 
 
 def check_nan(array):
     tmp = np.sum(array)
     if np.isnan(tmp) or np.isinf(tmp):
-        print('Warning: NaN or Inf found in input tensor.')
+        logging.warning('NaN or Inf found in input tensor.')
     return array