From ce67b9df6638a9b607942140e44094b9fce688a0 Mon Sep 17 00:00:00 2001 From: Nandan Manjunatha Date: Sun, 4 Apr 2021 22:09:39 +0530 Subject: [PATCH] minor fix fixed number of classes for coco in config file --- data/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/config.py b/data/config.py index 8999622cc..259739b56 100644 --- a/data/config.py +++ b/data/config.py @@ -27,7 +27,7 @@ } coco = { - 'num_classes': 201, + 'num_classes': 21, 'lr_steps': (280000, 360000, 400000), 'max_iter': 400000, 'feature_maps': [38, 19, 10, 5, 3, 1],