From 8d4ccb19e1f0b5f7c477d08c7883a60f65271af8 Mon Sep 17 00:00:00 2001 From: qingqing01 Date: Wed, 26 Sep 2018 19:02:16 +0800 Subject: [PATCH] Fix args. (#1298) --- fluid/object_detection/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/object_detection/train.py b/fluid/object_detection/train.py index 1106635d..6e763ea1 100644 --- a/fluid/object_detection/train.py +++ b/fluid/object_detection/train.py @@ -126,7 +126,7 @@ def train(args, devices = os.getenv("CUDA_VISIBLE_DEVICES") or "" devices_num = len(devices.split(",")) batch_size = train_params['batch_size'] - epoc_num = train_params['epoch_num'] + epoc_num = train_params['epoc_num'] batch_size_per_device = batch_size // devices_num iters_per_epoc = train_params["train_images"] // batch_size num_workers = 8 -- GitLab