Restore initial hyperparemeters for MLP

JonathanCrabbe · Jan 31, 2024 · ab5ad87 · ab5ad87
1 parent d96a964
commit ab5ad87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmd/conf/score_model/mlp.yaml b/cmd/conf/score_model/mlp.yaml
@@ -1,6 +1,6 @@
 _target_: fdiff.models.score_models.MLPScoreModule
 _partial_: true
-d_model: 128
+d_model: 72
 d_mlp: 1024
 num_layers: 10
 lr_max: 1.0e-4