tfm.core.train_utils.write_model_params

Writes the model parameters and shapes to a file.

model A model instance.
output_path Output file path.