WIP. (#8456)

Fix multi-GPU histogram metrics for > 0D tensors.
2026-07-10 11:48:56 +08:00 · 2020-05-15 21:43:27 +02:00
parent 4633d81c39
commit c9435cad43
3 changed files with 7 additions and 6 deletions
@@ -18,7 +18,7 @@ from ray.tune.utils import flatten_dict
 logger = logging.getLogger(__name__)

 tf = None
-VALID_SUMMARY_TYPES = [int, float, np.float32, np.float64, np.int32]
+VALID_SUMMARY_TYPES = [int, float, np.float32, np.float64, np.int32, np.int64]


 class Logger: