Trait relearn::torch::agents::critics::BuildCritic

pub trait BuildCritic {
    type Critic: Critic;

    fn build_critic(
        &self,
        in_dim: usize,
        discount_factor: f64,
        device: Device
    ) -> Self::Critic;
}

Build a Critic.

Required Associated Types

type Critic: Critic

Required Methods

fn build_critic(
    &self,
    in_dim: usize,
    discount_factor: f64,
    device: Device
) -> Self::Critic

Implementors

impl BuildCritic for RewardToGoConfig

type Critic = RewardToGo

impl<MB, OC> BuildCritic for ValuesOptConfig<MB, OC> where
    MB: BuildModule,
    MB::Module: SeqPacked,
    OC: BuildOptimizer,
    OC::Optimizer: Optimizer, 

type Critic = ValuesOpt<<MB as BuildModule>::Module, <OC as BuildOptimizer>::Optimizer>