Fix unbuilt status warnings

bayesflow-org · Jun 3, 2024 · dcc4386 · dcc4386
1 parent af1b2e6
commit dcc4386
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/bayesflow/experimental/networks/resnet/hidden_block.py b/bayesflow/experimental/networks/resnet/hidden_block.py
@@ -43,12 +43,17 @@ def call(self, inputs: Tensor, **kwargs):
             x = x + inputs
         return self.activation_fn(x)
 
+    def build(self, input_shape):
+        super().build(input_shape)
+        self(keras.KerasTensor(input_shape))
+
     def get_config(self):
         config = super().get_config()
         config.update({
             "residual": self.residual,
             "spectral_norm": self.spectral_norm,
             "activation_fn": keras.saving.serialize_keras_object(self.activation_fn),
-            "dense_with_dropout": keras.saving.serialize_keras_object(self.dense_with_dropout)
+            "dense": keras.saving.serialize_keras_object(self.dense),
+            "dropout": keras.saving.serialize_keras_object(self.dropout)
         })
         return config
diff --git a/bayesflow/experimental/networks/resnet/resnet.py b/bayesflow/experimental/networks/resnet/resnet.py
@@ -81,3 +81,4 @@ def __init__(
 
     def call(self, inputs: Tensor, **kwargs):
         return self.res_blocks(inputs, **kwargs)
+
Original file line number	Diff line number	Diff line change
Expand Up		@@ -81,3 +81,4 @@ def __init__(

		def call(self, inputs: Tensor, **kwargs):
		return self.res_blocks(inputs, **kwargs)