RealmOA Library

oa::ml::training::ItRolloutTraining

Coordinator for alternating fixed-capacity collection and optimizer phases.

StRust public API

Declaration

pub struct ItRolloutTraining<'engine, 'hooks> { /* private fields */ }
Public Rust declaration.

Public Methods

pub fn begin_rollout(&mut self, rollout: &mut RolloutBuffer) -> Result<()>
pub fn finalize_rollout( &mut self, rollout: &mut RolloutBuffer, config: GaeConfig, ) -> Result<()>
pub fn abort_rollout(&mut self, rollout: &mut RolloutBuffer) -> Result<()>
pub fn begin_update(&mut self) -> Result<bool>
pub fn complete_update(&mut self, loss: &Matrix) -> Result<()>
pub fn is_done(&self) -> bool
pub const fn phase(&self) -> RolloutTrainingPhase
pub const fn rollout_index(&self) -> u64
pub const fn update_epoch(&self) -> u64
pub const fn config(&self) -> &ItRolloutTrainingConfig
pub const fn update_loop(&self) -> &ItTraining<'engine, 'hooks>
pub fn update_loop_mut(&mut self) -> &mut ItTraining<'engine, 'hooks>
pub fn finish(self) -> Result<TrainingSnapshot>

Associated Functions

pub fn new( engine: &'engine Engine, optimizer: &'engine mut dyn Optimizer, config: ItRolloutTrainingConfig, ) -> Result<Self>

Public Method Documentation

pub fn begin_rollout(&mut self, rollout: &mut RolloutBuffer) -> Result<()>
No public source comment is attached to this declaration.
pub fn finalize_rollout( &mut self, rollout: &mut RolloutBuffer, config: GaeConfig, ) -> Result<()>
No public source comment is attached to this declaration.
pub fn abort_rollout(&mut self, rollout: &mut RolloutBuffer) -> Result<()>
No public source comment is attached to this declaration.
pub fn begin_update( &mut self ) -> Result<bool>
No public source comment is attached to this declaration.
pub fn complete_update(&mut self, loss: &Matrix) -> Result<()>
No public source comment is attached to this declaration.
pub fn is_done( &self ) -> bool
No public source comment is attached to this declaration.
pub const fn phase( &self ) -> RolloutTrainingPhase
No public source comment is attached to this declaration.
pub const fn rollout_index( &self ) -> u64
No public source comment is attached to this declaration.
pub const fn update_epoch( &self ) -> u64
No public source comment is attached to this declaration.
pub const fn config( &self ) -> &ItRolloutTrainingConfig
No public source comment is attached to this declaration.
pub const fn update_loop( &self ) -> &ItTraining<'engine, 'hooks>
No public source comment is attached to this declaration.
pub fn update_loop_mut( &mut self ) -> &mut ItTraining<'engine, 'hooks>
No public source comment is attached to this declaration.
pub fn finish( self ) -> Result<TrainingSnapshot>
No public source comment is attached to this declaration.
pub fn new( engine: &'engine Engine, optimizer: &'engine mut dyn Optimizer, config: ItRolloutTrainingConfig ) -> Result<Self>
No public source comment is attached to this declaration.