Update ape_deta_vitl_eva02_lsj1024.py

shenyunhang · Apr 7, 2024 · 504a9ed · 504a9ed
1 parent 308db73
commit 504a9ed
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/configs/D3_InstanceSegmentation/ape_deta/ape_deta_vitl_eva02_lsj1024.py b/configs/D3_InstanceSegmentation/ape_deta/ape_deta_vitl_eva02_lsj1024.py
@@ -6,14 +6,14 @@
 )
 from ...common.data.d3_instance_lsj1024 import dataloader
 
-model.model_vision.dataset_prompts = ["expression"]
-model.model_vision.dataset_names = ["d3"]
+model.model_vision.dataset_prompts = ["expression",]
+model.model_vision.dataset_names = ["d3",]
 model.model_vision.dataset_metas = dataloader.train.dataset.names
 
 model.model_vision.num_classes = 256
 model.model_vision.criterion[0].num_classes = 256
 model.model_vision.select_box_nums_for_evaluation = 300
-model.model_vision.select_box_nums_for_evaluation_list = [300, 300]
+model.model_vision.select_box_nums_for_evaluation_list = [300,]
 
 model.model_vision.instance_on = True
 model.model_vision.semantic_on = False