Skip to content

Commit 83035e9

Browse files
authored
Merge pull request #13382 from typhoonzero/fix_fluid_benchmark_resnet_lr
fix fluid_benchmark resnet lr decay
2 parents c999528 + cb4a73b commit 83035e9

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)