Skip to content

Commit 47f500e

Browse files
committed
fix the bug of scaling learning rate of trained
1 parent 0da233c commit 47f500e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorlayer/distributed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def __init__(
140140

141141
# Adjust learning rate based on number of GPUs.
142142
lr = optimizer_args['learning_rate']
143-
optimizer_args['learning_rate'] = scaling_learning_rate if lr * hvd.size() else lr
143+
optimizer_args['learning_rate'] = lr * hvd.size() if scaling_learning_rate else lr
144144
opt = optimizer(**optimizer_args)
145145

146146
# Add Horovod Distributed Optimizer.

0 commit comments

Comments
 (0)