fix reproduction

Owen-Liuyuxuan · Feb 1, 2021 · 3affa28 · 3affa28
1 parent 8139863
commit 3affa28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/config/Yolo3D_example b/config/Yolo3D_example
@@ -151,7 +151,7 @@ head_layer = edict(
     num_features_in=1024,
     num_cls_output=len(cfg.obj_types)+1,
     num_reg_output=12,
-    cls_feature_size=1024,
+    cls_feature_size=512,
     reg_feature_size=1024,
 )
 detector.head = edict(