diff --git a/fluid/object_detection/train.py b/fluid/object_detection/train.py index 1106635dabab26ff70ccad81d477af25819cec17..6e763ea1d4ae1a2579238aa4388bc6425b1400f7 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