Merge pull request #13382 from typhoonzero/fix_fluid_benchmark_resnet_lr

fix fluid_benchmark resnet lr decay
7 years ago · 83035e9d53
parent c9995289f1 cb4a73be01
commit 83035e9d53
1 changed files with 1 additions and 1 deletions
--- a/benchmark/fluid/models/resnet.py
+++ b/benchmark/fluid/models/resnet.py
@ -207,7 +207,7 @@ def get_model(args, is_train, main_prog, startup_prog):

                total_images = 1281167 / trainer_count

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