Save the model into a .h5 model
Hi there, I have fine-tuned the ViTForImageClassification and now I save into a folder containing:config.jsontf_model.h5but the point is that the .h5 doesn’t contain the model but just the weights.There is a way to get the whole model in one .h5 model? It’s possible in keras so I think that there is a solution for it!