Add Hydra config for UnimodalLogitsHead prediction head

creatis-myriad · Nov 1, 2023 · 151fb75 · 151fb75
1 parent 1520968
commit 151fb75
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/didactic/config/task/model/prediction_head/unimodal_logits.yaml b/didactic/config/task/model/prediction_head/unimodal_logits.yaml
@@ -0,0 +1,6 @@
+_target_: didactic.models.layers.UnimodalLogitsHead
+_partial_: True
+in_features: ${task.embed_dim}
+backbone_distribution: binomial
+tau: 1
+tau_mode: learn_fn