cutcutcodec.core.nn.model.enhancement.train.train

cutcutcodec.core.nn.model.enhancement.train.train(dataset: VideoDataset | Path | str | bytes, model: CNN | None = None, log: Path | str | bytes | None = None)[source]

Train the model.

Parameters

datasetVideoDataset or pathlike: The dataset containing the videos and dataaug or the folder.
modelCNN, optional: Use the default constructor if not provided.
logpathlike: The directory to store the logs.

Examples

>>> from cutcutcodec.core.nn.model.enhancement.train import train
>>> # train("~/dataset/video_xiph")
>>>