Update experiment scripts

2 years ago · 87596b5f1f
parent 0f06d5d1ce
commit 87596b5f1f
19 changed files with 58 additions and 70 deletions
--- a/examples/rs_research/configs/levircd/bit.yaml
+++ b/examples/rs_research/configs/levircd/bit.yaml
@ -1,6 +0,0 @@
-_base_: ./levircd.yaml
-
-save_dir: ./exp/levircd/bit/
-
-model: !Node
-    type: BIT
--- a/examples/rs_research/configs/levircd/custom_model/custom_model_c.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/custom_model_c.yaml
@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_c/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: c
--- a/examples/rs_research/configs/levircd/custom_model/custom_model_cs.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/custom_model_cs.yaml
@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_cs/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: cs
--- a/examples/rs_research/configs/levircd/custom_model/custom_model_cst.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/custom_model_cst.yaml
@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_cst/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: cst
--- a/examples/rs_research/configs/levircd/custom_model/custom_model_ct.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/custom_model_ct.yaml
@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_ct/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: ct
--- a/examples/rs_research/configs/levircd/custom_model/custom_model_s.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/custom_model_s.yaml
@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_s/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: s
--- a/examples/rs_research/configs/levircd/custom_model/custom_model_st.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/custom_model_st.yaml
@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_st/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: st
--- a/examples/rs_research/configs/levircd/custom_model/custom_model_t.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/custom_model_t.yaml
@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_t/
+
+model: !Node
+    type: CustomTrainer
+    args:
+        att_types: t
--- a/examples/rs_research/configs/levircd/custom_model/iterative_bit_iter2.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/iterative_bit_iter2.yaml
@ -1,11 +0,0 @@
-_base_: ../levircd.yaml
-
-save_dir: ./exp/levircd/custom_model/iter2/
-
-model: !Node
-    type: IterativeBIT
-    args:
-        num_iters: 2
-        num_classes: 2
-        bit_kwargs:
-            in_channels: 3
--- a/examples/rs_research/configs/levircd/custom_model/iterative_bit_iter3.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/iterative_bit_iter3.yaml
@ -1,11 +0,0 @@
-_base_: ../levircd.yaml
-
-save_dir: ./exp/levircd/custom_model/iter3/
-
-model: !Node
-    type: IterativeBIT
-    args:
-        num_iters: 3
-        num_classes: 2
-        bit_kwargs:
-            in_channels: 3
--- a/examples/rs_research/configs/levircd/custom_model/iterative_bit_iter4.yaml
+++ b/examples/rs_research/configs/levircd/custom_model/iterative_bit_iter4.yaml
@ -1,11 +0,0 @@
-_base_: ../levircd.yaml
-
-save_dir: ./exp/levircd/custom_model/iter4/
-
-model: !Node
-    type: IterativeBIT
-    args:
-        num_iters: 4
-        num_classes: 2
-        bit_kwargs:
-            in_channels: 3
--- a/examples/rs_research/configs/levircd/fc_ef.yaml
+++ b/examples/rs_research/configs/levircd/fc_ef.yaml
@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_ef/

 model: !Node
    type: FCEarlyFusion
-    args:
-        use_dropout: True
--- a/examples/rs_research/configs/levircd/fc_siam_conc.yaml
+++ b/examples/rs_research/configs/levircd/fc_siam_conc.yaml
@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_siam_conc/

 model: !Node
    type: FCSiamConc
-    args:
-        use_dropout: True
--- a/examples/rs_research/configs/levircd/fc_siam_diff.yaml
+++ b/examples/rs_research/configs/levircd/fc_siam_diff.yaml
@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_siam_diff/

 model: !Node
    type: FCSiamDiff
-    args:
-        use_dropout: True
--- a/examples/rs_research/configs/levircd/stanet.yaml
+++ b/examples/rs_research/configs/levircd/stanet.yaml
@ -1,6 +0,0 @@
-_base_: ./levircd.yaml
-
-save_dir: ./exp/levircd/stanet/
-
-model: !Node
-    type: STANet
--- a/examples/rs_research/configs/svcd/bit.yaml
+++ b/examples/rs_research/configs/svcd/bit.yaml
@ -1,6 +0,0 @@
-_base_: ./svcd.yaml
-
-save_dir: ./exp/svcd/bit/
-
-model: !Node
-    type: BIT
--- a/examples/rs_research/configs/svcd/custom_model.yaml
+++ b/examples/rs_research/configs/svcd/custom_model.yaml
@ -3,9 +3,4 @@ _base_: ./svcd.yaml
 save_dir: ./exp/svcd/custom_model/

 model: !Node
-    type: IterativeBIT
-    args:
-        num_iters: 2
-        num_classes: 2
-        bit_kwargs:
-            in_channels: 3
+    type: CustomTrainer
--- a/examples/rs_research/configs/svcd/stanet.yaml
+++ b/examples/rs_research/configs/svcd/stanet.yaml
@ -1,6 +0,0 @@
-_base_: ./svcd.yaml
-
-save_dir: ./exp/svcd/stanet/
-
-model: !Node
-    type: STANet
--- a/examples/rs_research/scripts/run_parameter_analysis.sh
+++ b/examples/rs_research/scripts/run_parameter_analysis.sh
@ -3,7 +3,7 @@
 set -e 

 CONFIG_DIR='configs/levircd/custom_model'
-LOG_DIR='exp/logs/parameter_analysis'
+LOG_DIR='exp/logs/ablation'

 mkdir -p "${LOG_DIR}"