Giters

horovod / horovod

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

Home Page:http://horovod.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

whatdhack opened this issue · comments

commented

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 ?