From cb4a73be010d2314531173e28022e3b5d163c033 Mon Sep 17 00:00:00 2001 From: typhoonzero Date: Thu, 13 Sep 2018 15:45:27 +0800 Subject: [PATCH] fix fluid_benchmark resnet lr decay --- benchmark/fluid/models/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmark/fluid/models/resnet.py b/benchmark/fluid/models/resnet.py index d71b855612a..1b3bfe659c7 100644 --- 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 -- GitLab