SaveFormat

Inheritance Diagram

Inheritance diagram of ashpy.callbacks.save_callback.SaveFormat

class ashpy.callbacks.save_callback.SaveFormat[source]

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()[source]

Name of the format.

Return type

str

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

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

Parameters
Return type

None