From 1eb03d6f2217393e998d5cbf5819182ae86438de Mon Sep 17 00:00:00 2001 From: lars Date: Sun, 7 Jul 2024 13:28:10 +0200 Subject: [PATCH] fix default width of resnet --- bayesflow/networks/resnet/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bayesflow/networks/resnet/resnet.py b/bayesflow/networks/resnet/resnet.py index 230a37cdc..7677b4a6d 100644 --- a/bayesflow/networks/resnet/resnet.py +++ b/bayesflow/networks/resnet/resnet.py @@ -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)