Struct border_async_trainer::AsyncTrainerConfig

pub struct AsyncTrainerConfig {
    pub model_dir: Option<String>,
    pub record_interval: usize,
    pub eval_interval: usize,
    pub max_train_steps: usize,
    pub save_interval: usize,
    pub sync_interval: usize,
}

Configuration of AsyncTrainer

Fields

model_dir: Option<String>

Where to save the trained model.

record_interval: usize

Interval of recording in training steps.

eval_interval: usize

Interval of evaluation in training steps.

max_train_steps: usize

The maximal number of training steps.

save_interval: usize

Interval of saving the model in optimization steps.

sync_interval: usize

Interval of synchronizing model parameters in training steps.

Implementations

impl AsyncTrainerConfig

pub fn load(path: impl AsRef<Path>) -> Result<Self>

Constructs AsyncTrainerConfig from YAML file.

pub fn save(&self, path: impl AsRef<Path>) -> Result<()>

Saves AsyncTrainerConfig.

Trait Implementations

impl Clone for AsyncTrainerConfig

fn clone(&self) -> AsyncTrainerConfig

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for AsyncTrainerConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for AsyncTrainerConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for AsyncTrainerConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>where __S: Serializer,

Serialize this value into the given Serde serializer.