Struct AsyncTrainerConfig

pub struct AsyncTrainerConfig {
    pub max_opts: usize,
    pub eval_interval: usize,
    pub flush_record_interval: usize,
    pub record_compute_cost_interval: usize,
    pub record_agent_info_interval: usize,
    pub save_interval: usize,
    pub sync_interval: usize,
    pub warmup_period: usize,
}

Configuration of AsyncTrainer.

Fields

max_opts: usize

The maximum number of optimization steps.

eval_interval: usize

Interval of evaluation in training steps.

flush_record_interval: usize

Interval of flushing records in optimization steps.

record_compute_cost_interval: usize

Interval of recording agent information in optimization steps.

record_agent_info_interval: usize

Interval of recording agent information in optimization steps.

save_interval: usize

Interval of saving the model in optimization steps.

sync_interval: usize

Interval of synchronizing model parameters in training steps.

warmup_period: usize

Warmup period, for filling replay buffer, in environment steps

Implementations

impl AsyncTrainerConfig

pub fn max_opts(self, v: usize) -> Result<Self>

Sets the number of optimization steps.

pub fn eval_interval(self, v: usize) -> Result<Self>

Sets the interval of evaluation in optimization steps.

pub fn record_compute_cost_interval( self, record_compute_cost_interval: usize, ) -> Result<Self>

Sets the interval of computation cost in optimization steps.

pub fn flush_record_interval(self, flush_record_interval: usize) -> Result<Self>

Sets the interval of flushing recordd in optimization steps.

pub fn warmup_period(self, warmup_period: usize) -> Result<Self>

Sets warmup period in environment steps.

pub fn save_interval(self, save_interval: usize) -> Result<Self>

Sets the interval of saving in optimization steps.

pub fn sync_interval(self, sync_interval: usize) -> Result<Self>

Sets the interval of synchronizing model parameters in training steps.

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 Default for AsyncTrainerConfig

fn default() -> Self

There is no special intention behind these initial values.

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.