hookeai.material_model_finder.train.training.save_parameters_history

save_parameters_history(model, model_parameters_history, model_parameters_bounds)[source]

Save model learnable parameters history record.

Model parameters record file is stored in model_directory under the name parameters_history_record.pkl.

Overwrites existing model parameters record file.

Parameters:
  • model (torch.nn.Module) – Model.

  • model_parameters_history (dict) – Model learnable parameters history. For each model parameter (key, str), store the corresponding training history (item, list).

  • model_parameters_bounds (dict) – Model learnable parameters bounds. For each parameter (key, str), the corresponding bounds are stored as a tuple(lower_bound, upper_bound) (item, tuple).