#4040·horovod

Replace tf.train.SessionRunHook by tf.compat.v1.train.SessionRunHook ?

Author: whatdhackCreated May 1, 2024Updated Oct 8, 2024
Labelsbug

Environment:

  1. Framework: TensorFlow,
  2. Framework version: 2.16
  3. Horovod version: 0.28.1
  4. MPI version:
  5. CUDA version: 12.2
  6. NCCL version:
  7. Python version: 3.11.8
  8. Spark / PySpark version:
  9. Ray version:
  10. OS and version:
  11. GCC version:
  12. CMake version:

Bug report: The line https://github.com/horovod/horovod/blob/master/horovod/tensorflow/__init__.py#L510 looks incorrect. Should tf.train.SessionRunHook be replaced by tf.compat.v1.train.SessionRunHook ?