Replace tf.train.SessionRunHook by tf.compat.v1.train.SessionRunHook ?
Author: whatdhackCreated May 1, 2024Updated Oct 8, 2024
Labelsbug
Environment:
- Framework: TensorFlow,
- Framework version: 2.16
- Horovod version: 0.28.1
- MPI version:
- CUDA version: 12.2
- NCCL version:
- Python version: 3.11.8
- Spark / PySpark version:
- Ray version:
- OS and version:
- GCC version:
- 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 ?
Source: horovod/horovod