search

kpot / keras-transformer

Keras library for building (Universal) Transformers, facilitating BERT and GPT models
MIT License
534 stars 140 forks source link

How to load the model? #22

Open 990579474 opened 5 years ago

990579474 commented 5 years ago

When I use a model.save() to save the model, I use a keras.models.load_model to load the model and report an error,How to load the model?

990579474 commented 5 years ago

model = load_model( 'model/model', custom_objects={ 'EmbeddingRet': EmbeddingRet, 'TrigPosEmbedding': TrigPosEmbedding, 'MultiHeadAttention': MultiHeadAttention, 'LayerNormalization': LayerNormalization, 'FeedForward': FeedForward, 'EmbeddingSim': EmbeddingSim, })

HernanFAR commented 3 years ago

In case this is the code that complements the above answer

import keras from keras_transformer.transformer import EmbeddingRet, TrigPosEmbedding, MultiHeadAttention, LayerNormalization, FeedForward, EmbeddingSim

model = keras.models.load_model( '/path/t/model', custom_objects = { 'EmbeddingRet': EmbeddingRet, 'TrigPosEmbedding': TrigPosEmbedding, 'MultiHeadAttention': MultiHeadAttention, 'LayerNormalization': LayerNormalization, 'FeedForward': FeedForward, 'EmbeddingSim': EmbeddingSim, } )