Skip to content

Commit

Permalink
fix default width of resnet
Browse files Browse the repository at this point in the history
  • Loading branch information
LarsKue committed Jul 7, 2024
1 parent 3855cb2 commit 1eb03d6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bayesflow/networks/resnet/resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
class ResNet(keras.Layer):
"""Implements a super-simple ResNet"""

def __init__(self, depth: int = 6, width: int = 2, activation: str = "gelu", **kwargs):
def __init__(self, depth: int = 6, width: int = 256, activation: str = "gelu", **kwargs):
super().__init__(**keras_kwargs(kwargs))

self.input_layer = keras.layers.Dense(width)
Expand Down

0 comments on commit 1eb03d6

Please sign in to comment.