keep experiment for submission

quqixun · Jan 27, 2023 · 15fac8f · 15fac8f
1 parent 2a85c44
commit 15fac8f
Show file tree

Hide file tree

Showing 16 changed files with 34 additions and 601 deletions.
diff --git a/configs/swin_unetr/exp1.yaml b/configs/swin_unetr/exp1.yaml
@@ -10,33 +10,36 @@ loader:
  val_batch: 4
  num_workers: 16
  pin_memory: true
- apply_augment: false
+ apply_augment: true
  s1_index_list: all
  s2_index_list: all
  months_list: all
 
 trainer:
- epochs: 50
- accum_iter: 2
- ckpt_freq: 50
+ epochs: 100
+ accum_iter: 1
+ ckpt_freq: 10
  print_freq: 200
- early_stop: null
+ early_stop: 20
 
 loss:
  rec:
  mode: mae
  weight: 1.0
+ sim:
+ mode: ssim
+ weight: 0.2
 
 optimizer:
  mode: adamw
  lr: 0.001
- betas: [0.9, 0.999]
- amsgrad: true
+ betas: [0.9, 0.99]
+ amsgrad: false
  weight_decay: 0.01
 
 scheduler:
  min_lr: 0.0
- warmup: 5
+ warmup: 10
 
 model:
  name: swin_unetr
@@ -48,12 +51,12 @@ model:
  out_channels: 1
  depths: [2, 2, 2, 2]
  num_heads: [3, 6, 12, 24]
- feature_size: 48
+ feature_size: 72
  norm_name: batch
  drop_rate: 0.0
  attn_drop_rate: 0.0
  drop_path_rate: 0.1
- attn_version: v1
+ attn_version: v2
 
 
-...
+...
diff --git a/configs/swin_unetr/exp10.yaml b/configs/swin_unetr/exp10.yaml
diff --git a/configs/swin_unetr/exp14.yaml b/configs/swin_unetr/exp14.yaml
diff --git a/configs/swin_unetr/exp19.yaml b/configs/swin_unetr/exp19.yaml
diff --git a/configs/swin_unetr/exp20.yaml b/configs/swin_unetr/exp20.yaml
diff --git a/configs/swin_unetr/exp21.yaml b/configs/swin_unetr/exp21.yaml