def get_custom_objects(): """Get all custom objects for loading saved models.""" custom_objects = get_encoder_custom_objects() custom_objects['PositionEmbedding'] = PositionEmbedding custom_objects['TokenEmbedding'] = TokenEmbedding custom_objects['EmbeddingSimilarity'] = EmbeddingSimilarity custom_objects['Masked'] = Masked custom_objects['Extract'] = Extract custom_objects['gelu'] = gelu return custom_objects
def get_custom_objects(): """Get all custom objects for loading saved models.""" custom_objects = get_encoder_custom_objects() custom_objects['PositionEmbedding'] = PositionEmbedding custom_objects['TokenEmbedding'] = TokenEmbedding custom_objects['EmbeddingSimilarity'] = EmbeddingSimilarity custom_objects['Masked'] = Masked custom_objects['Extract'] = Extract custom_objects['gelu'] = gelu custom_objects['gelu_tensorflow'] = gelu_tensorflow custom_objects['gelu_fallback'] = gelu_fallback custom_objects['AdamWarmup'] = AdamWarmup return custom_objects