fix trainers.py error with latest version of horovod (0.17.0.post1) (#1306)

0226451c · xiaohanhuang · Yuxin Wu · 047579df · 0226451c
Commit 0226451c authored Aug 22, 2019 by xiaohanhuang Committed by Yuxin Wu Aug 22, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tensorpack/train/trainers.py tensorpack/train/trainers.py +1 -1

No files found.
--- a/tensorpack/train/trainers.py
+++ b/tensorpack/train/trainers.py
@@ -383,7 +383,7 @@ class HorovodTrainer(SingleCostTrainer):
        # lazy import
        import horovod.tensorflow as hvd
        import horovod
-        hvd_version = tuple(map(int, horovod.__version__.split('.')))
+        hvd_version = tuple(map(int, horovod.__version__.split('.')[:3]))
        self.hvd = hvd

        hvd.init()