The python code works for building the model, however I cant get an output for saving the model parameters and weights to the txt file.