This commit is contained in:
Starbeamrainbowlabs 2022-08-31 18:54:03 +01:00
parent e4edc68df5
commit b2a320134e
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -33,8 +33,8 @@ def make_convnext(input_shape, arch_name="convnext_tiny", **kwargs):
)
layer_out = convnext(layer_in, **depths_dims[arch_name], **kwargs)
return tf.keras.Model(
input = layer_in,
output = layer_out
inputs = layer_in,
outputs = layer_out
)