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