cutcutcodec.core.nn.start.load

cutcutcodec.core.nn.start.load(model: Module, weights: Path | str | bytes | None = None)[source]

Load the pretrained weights.

Parameters

modeltorch.nn.Module

The model to be loaded.

weightspathlike, optional

The path to the loading weight file with the suffix .pt or .pt.xz