diff --git a/config/cls.yaml b/config/cls.yaml
index 1d7783d0063871b78c3450073297cf58732286ec..4a2d92fc6a4504519da17de0cf20c23865e0fbe1 100644
--- a/config/cls.yaml
+++ b/config/cls.yaml
@@ -16,4 +16,4 @@ hydra:
 
   sweep:
     dir: log/cls
-    subdir: ${model.name}
\ No newline at end of file
+    subdir: ${model.name}
diff --git a/config/model/Hengshuang.yaml b/config/model/Hengshuang.yaml
index f3cd12e5496c8499abddf41589c501b1b46e6f18..8cbfb74ab6d05466127a713d4f5476ce249add9f 100644
--- a/config/model/Hengshuang.yaml
+++ b/config/model/Hengshuang.yaml
@@ -1,5 +1,4 @@
-# @package _group_
 nneighbor: 16
 nblocks: 4
 transformer_dim: 512
-name: Hengshuang
\ No newline at end of file
+name: Hengshuang
diff --git a/config/model/Menghao.yaml b/config/model/Menghao.yaml
index fea1a8083f378f5d93409de6468f760c7b7c05ec..42981b5c69229d4e4f99f5292ec495ac91a3cbe3 100644
--- a/config/model/Menghao.yaml
+++ b/config/model/Menghao.yaml
@@ -1,2 +1 @@
-# @package _group_
-name: Menghao
\ No newline at end of file
+name: Menghao
diff --git a/config/model/Nico.yaml b/config/model/Nico.yaml
index 9fd6b2e50f84eddcc244a351533f4fab8837e8b5..4f60402690dd40be6a948299c81aa2afe8afc494 100644
--- a/config/model/Nico.yaml
+++ b/config/model/Nico.yaml
@@ -1,4 +1,3 @@
-# @package _group_
 n_head: 8
 m: 4
 k: 64
@@ -6,4 +5,4 @@ global_k: 128
 global_dim: 512
 local_dim: 256
 reduce_dim: 64
-name: Nico
\ No newline at end of file
+name: Nico
diff --git a/config/partseg.yaml b/config/partseg.yaml
index 1e3dd3e59155b058696b7682256a3d92b446da64..dccaca0e0ea6b68140684c389ae9a7d544c751ce 100644
--- a/config/partseg.yaml
+++ b/config/partseg.yaml
@@ -18,4 +18,4 @@ hydra:
 
   sweep:
     dir: log/partseg
-    subdir: ${model.name}
\ No newline at end of file
+    subdir: ${model.name}