fix fluid_benchmark resnet lr decay

7 years ago · cb4a73be01
parent a56b6637d2
commit cb4a73be01
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