Struct ActorCriticNetwork 

pub struct ActorCriticNetwork { /* private fields */ }

Actor-Critic network for PPO

Implementations

impl ActorCriticNetwork

pub fn new( state_dim: usize, num_actions: usize, num_params: usize, vb: VarBuilder<'_>, ) -> Result<Self>

pub fn forward( &self, state: &Tensor, _training: bool, ) -> Result<(Tensor, Tensor, Tensor, Tensor)>

pub fn save_to_file(&self, path: &Path, metadata: ModelMetadata) -> Result<()>

Save PPO model to file with metadata

pub fn load_from_file( path: &Path, state_dim: usize, num_actions: usize, num_params: usize, device: &Device, ) -> Result<(Self, VarMap)>

Load PPO model from file - returns network and varmap

pub fn load_with_device( path: &Path, state_dim: usize, num_actions: usize, num_params: usize, device: &Device, ) -> Result<(Self, VarMap)>

Update load_with_device to use load_from_file