turi::style_transfer::Config Struct Reference

#include <toolkits/style_transfer/st_model_trainer.hpp>

Public Attributes
int	num_styles = 1
int	max_iterations = -1
int	batch_size = 1
int	training_image_height = 256
int	training_image_width = 256
int	random_seed = 0

Detailed Description

Model-agnostic parameters for style transfer.

Definition at line 73 of file st_model_trainer.hpp.

Member Data Documentation

batch_size

int turi::style_transfer::Config::batch_size = 1

The number of images to process per training batch.

Definition at line 85 of file st_model_trainer.hpp.

max_iterations

int turi::style_transfer::Config::max_iterations = -1

The target number of training iterations to perform.

If -1, then this target should be computed heuristically.

Definition at line 82 of file st_model_trainer.hpp.

num_styles

int turi::style_transfer::Config::num_styles = 1

Determines the number of style images used during training.

Definition at line 75 of file st_model_trainer.hpp.

random_seed

int turi::style_transfer::Config::random_seed = 0

Random seed used to initialize the model.

Definition at line 94 of file st_model_trainer.hpp.

training_image_height

int turi::style_transfer::Config::training_image_height = 256

The height of images passed into the training backend.

Definition at line 88 of file st_model_trainer.hpp.

training_image_width

int turi::style_transfer::Config::training_image_width = 256

The width of images passed into the training backend.

Definition at line 91 of file st_model_trainer.hpp.

---

The documentation for this struct was generated from the following file:

• toolkits/style_transfer/st_model_trainer.hpp