Skip to content

Commit 5e34e9c

Browse files
specify chief master behavior in r callbacks for distributed training
1 parent 09de409 commit 5e34e9c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

inst/python/kerastools/callback.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,9 @@ def __init__(self, r_set_context,
4545
self.r_on_test_end = r_on_test_end
4646
self.r_on_train_batch_begin = r_on_train_batch_begin
4747
self.r_on_train_batch_end = r_on_train_batch_end
48+
49+
# requiered when using tf$distributed strategies
50+
self._chief_worker_only = False
4851

4952
def on_epoch_begin(self, epoch, logs=None):
5053
self.r_on_epoch_begin(epoch, logs)

0 commit comments

Comments
 (0)