From 0ce34e207a1dee3af41cdad22771306a21a7863d Mon Sep 17 00:00:00 2001 From: Chongxiao Cao Date: Thu, 7 Oct 2021 15:04:58 -0700 Subject: [PATCH] Spark/Lightning: fix checkpoint callback dirpath typo Signed-off-by: Chongxiao Cao --- horovod/spark/lightning/remote.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/horovod/spark/lightning/remote.py b/horovod/spark/lightning/remote.py index 35800b79ae..7580b38e36 100644 --- 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