SaveWeights

minnt.callbacks.SaveWeights

Bases: Callback

A callback that saves model weights to a file after each epoch.

Source code in minnt/callbacks/save_weights.py

class SaveWeights(Callback):
    """A callback that saves model weights to a file after each epoch."""

    def __init__(self, path: str, optimizer_path: str | None = None) -> None:
        """Create the SaveWeights callback.

        Parameters:
          path: A path where weights will be saved using the [minnt.TrainableModule.save_weights][]
            method after each epoch. Note that you can use templates like `{logdir}` and `{epoch[:formatting]}`.
          optimizer_path: An optional path passed to [minnt.TrainableModule.save_weights][] to
            save also the optimizer state; it is relative to `path`.
        """
        self._path = path
        self._optimizer_path = optimizer_path

    def __call__(self, module: "TrainableModule", epoch: int, logs: Logs) -> None:
        module.save_weights(self._path, optimizer_path=self._optimizer_path)

__init__

__init__(path: str, optimizer_path: str | None = None) -> None

Create the SaveWeights callback.

Parameters:

• path (str) –

  A path where weights will be saved using the minnt.TrainableModule.save_weights method after each epoch. Note that you can use templates like {logdir} and {epoch[:formatting]}.

• optimizer_path (str | None, default: None ) –

  An optional path passed to minnt.TrainableModule.save_weights to save also the optimizer state; it is relative to path.

Source code in minnt/callbacks/save_weights.py

def __init__(self, path: str, optimizer_path: str | None = None) -> None:
    """Create the SaveWeights callback.

    Parameters:
      path: A path where weights will be saved using the [minnt.TrainableModule.save_weights][]
        method after each epoch. Note that you can use templates like `{logdir}` and `{epoch[:formatting]}`.
      optimizer_path: An optional path passed to [minnt.TrainableModule.save_weights][] to
        save also the optimizer state; it is relative to `path`.
    """
    self._path = path
    self._optimizer_path = optimizer_path