Contents

algorithm

The style transfer task’s training algorithm that prioritizes either speed or quality.

Declaration

var algorithm: MLStyleTransfer.ModelParameters.ModelAlgorithmType

Discussion

You choose the task’s training algorithm based on your intended use for this model.

Use MLStyleTransfer.ModelParameters.ModelAlgorithmType.cnnLite to train a model that prioritizes speed to stylize images with low latency, typically for frames of a video stream. Otherwise, select MLStyleTransfer.ModelParameters.ModelAlgorithmType.cnn to train a model that applies a higher-quality stylization, typically for a still image.

See Also

Setting style transfer parameters