SaveFormat¶
Inheritance Diagram
-
class
ashpy.callbacks.save_callback.
SaveFormat
[source]¶ Bases:
enum.Flag
Save Format enum.
Attributes
Model format (weights and architecture), saved using model.save()
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()
-
save
(model, save_dir, save_sub_format=<SaveSubFormat.TF: 'tf'>)[source]¶ Save the model using the correct format and sub-format.
- Parameters
model (
tf.keras.models.Model
) – model to Save.save_dir (str) – 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
-