Saving and loading the weights and the architecture of a model