Skip to content

Commit

Permalink
Equalize defaults
Browse files Browse the repository at this point in the history
  • Loading branch information
stefanradev93 committed Jul 10, 2024
1 parent 00a123a commit d8428c8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bayesflow/networks/flow_matching/flow_matching.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@

@register_keras_serializable(package="bayesflow.networks")
class FlowMatching(InferenceNetwork):
def __init__(self, subnet: str = "resnet", base_distribution: str = "normal", **kwargs):
def __init__(self, subnet: str = "mlp", base_distribution: str = "normal", **kwargs):
super().__init__(base_distribution=base_distribution, **keras_kwargs(kwargs))
self.subnet = find_network(subnet, **kwargs.get("subnet_kwargs", {}))
self.output_projector = keras.layers.Dense(units=None, bias_initializer="zeros", kernel_initializer="zeros")
Expand Down

0 comments on commit d8428c8

Please sign in to comment.