gaotingquan 1 рік тому
батько
коміт
c81e7918cd
1 змінених файлів з 1 додано та 1 видалено
  1. 1 1
      paddlex/modules/base/evaluator.py

+ 1 - 1
paddlex/modules/base/evaluator.py

@@ -93,7 +93,7 @@ class BaseEvaluator(ABC, metaclass=AutoRegisterABCMetaClass):
             return False
             return False
         for metric in metrics:
         for metric in metrics:
             val = metrics[metric]
             val = metrics[metric]
-            if not isinstance(val, float):
+            if not isinstance(val, (float, int)):
                 return False
                 return False
         return True
         return True