Skip to content

Commit 39e3bfc

Browse files
committed
update cem default parameters
1 parent 3810d32 commit 39e3bfc

File tree

3 files changed

+15
-2
lines changed

3 files changed

+15
-2
lines changed

conceptarium/conf/model/_commons.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ latent_encoder_kwargs:
1010
hidden_size: 64
1111
n_layers: 1
1212
activation: leaky_relu
13-
dropout: 0.5
13+
dropout: 0.
1414

1515

1616
# =============================================================

conceptarium/conf/model/cem.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ _target_: "torch_concepts.nn.ConceptEmbeddingModel"
77

88
task_names: ${dataset.default_task_names}
99

10-
exogenous_size: 16
10+
exogenous_size: 4
1111

1212
inference:
1313
_target_: "torch_concepts.nn.DeterministicInference"
Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
defaults:
2+
- _commons
3+
- _self_
4+
5+
_target_: "torch_concepts.nn.ConceptEmbeddingModel_Independent"
6+
7+
task_names: ${dataset.default_task_names}
8+
9+
exogenous_size: 4
10+
11+
inference:
12+
_target_: "torch_concepts.nn.DeterministicInference"
13+
_partial_: true

0 commit comments

Comments
 (0)