Spark/Lightning: fix checkpoint callback dirpath typo (horovod#3204)

Signed-off-by: Chongxiao Cao <chongxiaoc@uber.com> Signed-off-by: weihanmines <weihan13@amd.com>
weihanmines · Dec 10, 2021 · 995346d · 995346d
1 parent a1512da
commit 995346d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/horovod/spark/lightning/remote.py b/horovod/spark/lightning/remote.py
@@ -132,7 +132,7 @@ def train(serialized_model):
             # Otherwise we are seeing hanging in training.
             for cb in callbacks:
                 if isinstance(cb, ModelCheckpoint):
-                    cb.dir_path = ckpt_dir
+                    cb.dirpath = ckpt_dir
                     cb.filename = ckpt_filename
                     _checkpoint_callback = cb
                     require_checkpoint = True