fix yaml modle matchdg to trainer matchdg

marrlab · Dec 18, 2023 · fe002c1 · fe002c1
1 parent 2daf480
commit fe002c1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/benchmark/benchmark_pacs_resnet.yaml b/examples/benchmark/benchmark_pacs_resnet.yaml
@@ -91,7 +91,8 @@ hduva:  # name
 
 
 matchdg:  # name
-  model: matchdg
+  model: erm
+  trainer: matchdg
   shared:
     - gamma_reg
 
@@ -181,7 +182,8 @@ mldg:  # name
 
 
 hduva_matchdg:  # name
-  model: matchhduva
+  model: hduva
+  trainer: matchdg
   zx_dim: 0
 
   shared: