runtime error

Exit code: 1. Reason: ig={'module': 'keras.src.models.functional', 'class_name': 'Functional', 'config': {}, 'registered_name': 'Functional', 'build_config': {'input_shape': None}, 'name': 'encoder', 'inbound_nodes': [{'args': [{'class_name': '__keras_tensor__', 'config': {'shape': [None, 256, 256, 3], 'dtype': 'float32', 'keras_history': ['input_layer_6', 0, 0]}}], 'kwargs': {'mask': None}}]}. Exception encountered: <class 'keras.src.layers.core.lambda_layer.Lambda'> could not be deserialized properly. Please ensure that components that are Python object instances (layers, models, etc.) returned by `get_config()` are explicitly deserialized in the model's `from_config()` method. config={'module': 'keras.layers', 'class_name': 'Lambda', 'config': {'name': 'lambda', 'trainable': True, 'dtype': {'module': 'keras', 'class_name': 'DTypePolicy', 'config': {'name': 'float32'}, 'registered_name': None, 'shared_object_id': 140683505184992}, 'function': {'module': 'builtins', 'class_name': 'function', 'config': 'fourier_transform', 'registered_name': 'function'}, 'output_shape': {'class_name': '__lambda__', 'config': {'code': '4wEAAAAAAAAAAAAAAAQAAAATAAAA8yIAAACXAGcAfABkAGQBGgCiAXwAZAEZAAAAZAJ6BQAAkQGt\nBlMAKQNO6f/////pAgAAAKkAKQHaAXNzAQAAACD6IS90bXAvaXB5a2VybmVsXzIxODE0LzE4ODY2\nNDIwMS5wefoIPGxhbWJkYT76H2J1aWxkX2VuY29kZXIuPGxvY2Fscz4uPGxhbWJkYT4zAAAAcyAA\nAACAANA/U8ARwDPAQsAW0D9TyBHIMskV0FFSyRnRP1OAAPMAAAAA\n', 'defaults': None, 'closure': None}}, 'arguments': {}}, 'registered_name': None, 'build_config': {'input_shape': [None, 8, 8, 512]}, 'name': 'lambda', 'inbound_nodes': [{'args': [{'class_name': '__keras_tensor__', 'config': {'shape': [None, 8, 8, 512], 'dtype': 'float32', 'keras_history': ['block_5_conv_bn', 0, 0]}}], 'kwargs': {'mask': None}}]}. Exception encountered: Could not locate function 'fourier_transform'. Make sure custom classes are decorated with `@keras.saving.register_keras_serializable()`. Full object config: {'module': 'builtins', 'class_name': 'function', 'config': 'fourier_transform', 'registered_name': 'function'}

Container logs: