hookeai.material_model_finder.train.training.save_material_models_state

save_material_models_state(model, state_type='default', epoch=None)

Save material models states to files.

Parameters:

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

• state_type ({'default', 'init', 'epoch', 'best'}, default='default') –

  Saved model state file type. Options:

  ’default’ : Model default state

  ’init’ : Model initial state

  ’epoch’ : Model state of given training epoch

  ’best’ : Model state of best performance

• epoch (int, default=None) – Training epoch corresponding to current model state.