From 1cbbddd871f27909ee03cd11d72ebcab2e25a5f0 Mon Sep 17 00:00:00 2001 From: Xingyuan Bu Date: Thu, 17 May 2018 18:12:03 +0800 Subject: [PATCH] epocs typo (#919) --- fluid/object_detection/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/object_detection/train.py b/fluid/object_detection/train.py index 45cccbbb..28a5a6fb 100644 --- a/fluid/object_detection/train.py +++ b/fluid/object_detection/train.py @@ -217,11 +217,11 @@ def parallel_exe(args, boundaries = [epocs * 12, epocs * 19] elif '2017' in train_file_list: epocs = 118287 / batch_size - boundaries = [epcos * 12, epocs * 19] + boundaries = [epocs * 12, epocs * 19] values = [ learning_rate, learning_rate * 0.5, learning_rate * 0.25 ] - elif data_args.dataset == 'pascalvoc': + elif 'pascalvoc' in data_args.dataset: epocs = 19200 / batch_size boundaries = [epocs * 40, epocs * 60, epocs * 80, epocs * 100] values = [ -- GitLab