Couldn't able to save and load the model after finetuning
Maria-philna opened this issue · 1 comments
Maria-philna commented
In bigbird summarization, I have loaded pretrained model , after that I have performed finetuning on gigaword tensorflow dataset , then I tried to save the model using tf.saved_model.save(model, data_dir=export_dir) and loaded the model using loaded_model = tf.keras.models.load_model("/drive/My Drive/Checkpoint_Summarization/original_saved") and it is throwing
ValueError: Found zero restored functions for caller function.
Anhar2 commented
Hi, I'm trying to fine-tune it as well, and I encountered your issue by chance, could you solve it though?