modelcheckpoint

modelcheckpoint

ModelCheckpoint that supports fsspec filesystems.

Classes

ModelCheckpoint

ModelCheckpoint(
    output_location, file_formats=None, **kwargs
)

Bases: ModelCheckpoint

ModelCheckpoint that supports fsspec filesystems. Subclassed and adapted from https://github.com/keras-team/keras/blob/master/keras/callbacks.py

Source code in niceml/dlframeworks/keras/callbacks/modelcheckpoint.py

def __init__(
    self,
    output_location: Union[dict, LocationConfig],
    file_formats: Optional[dict] = None,
    **kwargs,
):
    super().__init__("", **kwargs)
    self.output_location = output_location
    self.file_formats = file_formats or {}

Functions