Skip to content

Commit 6bc37c3

Browse files
committed
try not to set trainable attribute
1 parent 72e3be4 commit 6bc37c3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bioimageio/core/model_adapters/_tensorflow_model_adapter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,12 +101,12 @@ def _get_network( # pyright: ignore[reportUnknownParameterType]
101101
if self.use_keras_api:
102102
try:
103103
return tf.keras.layers.TFSMLayer(
104-
weight_file, trainable=False, call_endpoint="serve"
104+
weight_file, call_endpoint="serve"
105105
) # pyright: ignore[reportUnknownVariableType]
106106
except Exception as e:
107107
try:
108108
return tf.keras.layers.TFSMLayer(
109-
weight_file, trainable=False, call_endpoint="serving_default"
109+
weight_file, call_endpoint="serving_default"
110110
) # pyright: ignore[reportUnknownVariableType]
111111
except Exception as ee:
112112
logger.opt(exception=ee).info(

0 commit comments

Comments
 (0)