Skip to content

Commit 3a55f58

Browse files
committed
[fix] demo issue for 'horovod_synchronous' -> 'synchronous'
1 parent 9767fcf commit 3a55f58

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

demo/dynamic_embedding/movielens-100k-sync-estimator-with-horovod/movielens-100k-estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def model_fn(features, labels, mode, params):
122122
sync_hook = HorovodSyncHook(device=device)
123123
optimizer = tf.compat.v1.train.AdamOptimizer(learning_rate=0.001)
124124
optimizer = tfra.dynamic_embedding.DynamicEmbeddingOptimizer(
125-
optimizer, horovod_synchronous=True)
125+
optimizer, synchronous=True)
126126
train_op = optimizer.minimize(
127127
loss, global_step=tf.compat.v1.train.get_or_create_global_step())
128128
return tf.estimator.EstimatorSpec(mode=mode,

0 commit comments

Comments
 (0)