Skip to content

Commit cb4a73b

Browse files
committed
fix fluid_benchmark resnet lr decay
1 parent a56b663 commit cb4a73b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

benchmark/fluid/models/resnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def get_model(args, is_train, main_prog, startup_prog):
207207

208208
total_images = 1281167 / trainer_count
209209

210-
step = int(total_images / args.batch_size + 1)
210+
step = int(total_images / (args.batch_size * args.gpus) + 1)
211211
epochs = [30, 60, 90]
212212
bd = [step * e for e in epochs]
213213
base_lr = args.learning_rate

0 commit comments

Comments
 (0)