Enum TrainerMsg 

pub enum TrainerMsg<P: ReplicaProtocol> {
    Step {
        batch: Vec<TrainSample>,
        reply: Sender<Result<StepStats, GpuError>>,
    },
    SetReplicas {
        replicas: Vec<ActorRef<P::Msg>>,
    },
}

Variants

Step

Fields

batch: Vec<TrainSample>

reply: Sender<Result<StepStats, GpuError>>

SetReplicas

Set the replica refs after construction. Allows a late-binding pattern when replicas are spawned by another actor.

Fields

replicas: Vec<ActorRef<P::Msg>>