SaveFormat

Inheritance Diagram

---

class ashpy.callbacks.save_callback.SaveFormat

Bases: enum.Flag

Save Format enum.

Attributes

MODEL	Model format (weights and architecture), saved using model.save()
WEIGHTS	Weights format, saved using model.save_weights()

MODEL = 2

Model format (weights and architecture), saved using model.save()

WEIGHTS = 1

Weights format, saved using model.save_weights()

name()

Name of the format.

Return type:str

save(model, save_dir, save_sub_format=<SaveSubFormat.TF: 'tf'>)

Save the model using the correct format and sub-format.

Parameters:

• model (tf.keras.models.Model) – model to Save.
• save_dir (pathlib.Path) – path of the file in which to save the model.
• save_sub_format (ashpy.callbacks.save_callback.SaveSubFormat) – sub-format of the save operation.

Return type:

None