kaveh commited on
Commit
fc5c00d
·
1 Parent(s): 3c48411

configs: image-only ablation variants

Browse files
Files changed (1) hide show
  1. experiments/configs.py +8 -0
experiments/configs.py CHANGED
@@ -53,6 +53,14 @@ VARIANTS = {
53
  "s2f_bridge_no_cbam": dict(bridge="cnn"),
54
  "s2f_bridge_plain": dict(bridge="plain"),
55
  "s2f_no_attention_gates": dict(attention_gates=False),
 
 
 
 
 
 
 
 
56
  }
57
 
58
  # Recommended order if GPU time is limited (S2F about 7 h per run on an RTX A6000, U-Net about 2.5 h)
 
53
  "s2f_bridge_no_cbam": dict(bridge="cnn"),
54
  "s2f_bridge_plain": dict(bridge="plain"),
55
  "s2f_no_attention_gates": dict(attention_gates=False),
56
+ # image-only (no substrate channels) counterparts: the substrate channels did not help (2 seeds), so the
57
+ # architecture and adversarial-loss comparisons are repeated on the image-only reference 's2f_no_substrate'
58
+ "unet_no_substrate": dict(arch="unet", lambda_gan=0.0, in_channels=1, substrate_mode="none"),
59
+ "unet_gan_no_substrate": dict(arch="unet", in_channels=1, substrate_mode="none"),
60
+ "s2f_no_substrate_no_gan": dict(in_channels=1, substrate_mode="none", lambda_gan=0.0),
61
+ "s2f_no_substrate_no_multiscale": dict(in_channels=1, substrate_mode="none", use_multi_scale_input=False),
62
+ "s2f_no_substrate_bridge_plain": dict(in_channels=1, substrate_mode="none", bridge="plain"),
63
+ "s2f_no_substrate_no_attention_gates": dict(in_channels=1, substrate_mode="none", attention_gates=False),
64
  }
65
 
66
  # Recommended order if GPU time is limited (S2F about 7 h per run on an RTX A6000, U-Net about 2.5 h)