@@ -1,6 +0,0 @@
-_base_: ./levircd.yaml
-
-save_dir: ./exp/levircd/bit/
-model: !Node
- type: BIT
@@ -0,0 +1,8 @@
+_base_: ../levircd.yaml
+
+save_dir: ./exp/levircd/custom_model/att_c/
+model: !Node
+ type: CustomTrainer
+ args:
+ att_types: c
+save_dir: ./exp/levircd/custom_model/att_cs/
+ att_types: cs
+save_dir: ./exp/levircd/custom_model/att_cst/
+ att_types: cst
+save_dir: ./exp/levircd/custom_model/att_ct/
+ att_types: ct
+save_dir: ./exp/levircd/custom_model/att_s/
+ att_types: s
+save_dir: ./exp/levircd/custom_model/att_st/
+ att_types: st
+save_dir: ./exp/levircd/custom_model/att_t/
+ att_types: t
@@ -1,11 +0,0 @@
-_base_: ../levircd.yaml
-save_dir: ./exp/levircd/custom_model/iter2/
- type: IterativeBIT
- args:
- num_iters: 2
- num_classes: 2
- bit_kwargs:
- in_channels: 3
-save_dir: ./exp/levircd/custom_model/iter3/
- num_iters: 3
-save_dir: ./exp/levircd/custom_model/iter4/
- num_iters: 4
@@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_ef/
model: !Node
type: FCEarlyFusion
- use_dropout: True
@@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_siam_conc/
type: FCSiamConc
@@ -4,5 +4,3 @@ save_dir: ./exp/levircd/fc_siam_diff/
type: FCSiamDiff
-save_dir: ./exp/levircd/stanet/
- type: STANet
-_base_: ./svcd.yaml
-save_dir: ./exp/svcd/bit/
@@ -3,9 +3,4 @@ _base_: ./svcd.yaml
save_dir: ./exp/svcd/custom_model/
-save_dir: ./exp/svcd/stanet/
@@ -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}"