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).