configs: image-only ablation variants
Browse files- 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)
|