Save decoder and encoder

[joao-alcindo]

How can I save the encoder and decoder of the model after pre training?

[ariG23498]

Hey @joao-alcindo thanks for you interest in the work!

You could access the encoder and decoder individually like so 👇

encoder = mae_model.encoder
decoder = mae_model.decoder

Once you have that, you could simple use the encoder.save("encoder.keras") API to save each of them (here encoder) separately. Hope that helps 😄