Interface SaveConfig

Options for saving a model.

interface SaveConfig {
includeOptimizer?: boolean;
trainableOnly?: boolean;
}

Properties

includeOptimizer?: boolean

Whether the optimizer will be saved (if exists).

Default: false.

trainableOnly?: boolean

Whether to save only the trainable weights of the model, ignoring the non-trainable ones.