fix fluid_benchmark resnet lr decay

cb4a73be · typhoonzero · a56b6637 · cb4a73be
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

benchmark/fluid/models/resnet.py benchmark/fluid/models/resnet.py +1 -1

未找到文件。
--- 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