bugfix(config): fix bug in cfg

LikeLidoA · Mar 14, 2022 · 97284db · 97284db
1 parent 52f6f4b
commit 97284db
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/experiments/pascal/1464/ours/config.yaml b/experiments/pascal/1464/ours/config.yaml
@@ -58,7 +58,9 @@ saver:
   pretrain: ''
 
 criterion:
-  use_weight: False
+  type: CELoss
+  kwargs:
+    use_weight: False
 
 net: # Required.
   num_classes: 21

diff --git a/experiments/pascal/1464/suponly/config.yaml b/experiments/pascal/1464/suponly/config.yaml
@@ -44,7 +44,9 @@ saver:
   pretrain: ''
 
 criterion:
-  use_weight: False
+  type: CELoss
+  kwargs:  
+    use_weight: False
 
 net: # Required.
   num_classes: 21