Struct border_async_trainer::AsyncTrainer

source · [−]

pub struct AsyncTrainer<A, E, R> where
    A: Agent<E, R> + SyncModel,
    E: Env,
    R: ReplayBufferBase,
    R::PushedItem: Send + 'static,  { /* private fields */ }

Expand description

Manages asynchronous training loop in a single machine.

It will be used with ActorManager

Implementations

source

impl<A, E, R> AsyncTrainer<A, E, R> where
 A: Agent<E, R> + SyncModel,
 E: Env,
 R: ReplayBufferBase,
 R::PushedItem: Send + 'static,

source

pub fn build(
 config: &AsyncTrainerConfig,
 agent_config: &A::Config,
 env_config: &E::Config,
 replay_buffer_config: &R::Config,
 r_bulk_pushed_item: Receiver<PushedItemMessage<R::PushedItem>>,
 model_info_sender: Sender<(usize, A::ModelInfo )>,
 stop: Arc<Mutex<bool>>
) -> Self

Creates AsyncTrainer.

source

pub fn train(
 &mut self,
 recorder: &mut impl Recorder,
 guard_init_env: Arc<Mutex<bool>>
) -> AsyncTrainStat

Runs training loop.

Auto Trait Implementations

impl<A, E, R> RefUnwindSafe for AsyncTrainer<A, E, R> where
 A: RefUnwindSafe,
 E: RefUnwindSafe,
 R: RefUnwindSafe,
 <A as Policy<E>>::Config: RefUnwindSafe,
 <E as Env>::Config: RefUnwindSafe,
 <R as ReplayBufferBase>::Config: RefUnwindSafe,

impl<A, E, R> Send for AsyncTrainer<A, E, R> where
 A: Send,
 E: Send,
 R: Send,
 <A as Policy<E>>::Config: Send,
 <E as Env>::Config: Send,
 <R as ReplayBufferBase>::Config: Send,
 <A as SyncModel>::ModelInfo: Send,

impl<A, E, R> Sync for AsyncTrainer<A, E, R> where
 A: Sync,
 E: Sync,
 R: Sync,
 <A as Policy<E>>::Config: Sync,
 <E as Env>::Config: Sync,
 <R as ReplayBufferBase>::Config: Sync,
 <A as SyncModel>::ModelInfo: Send,

impl<A, E, R> Unpin for AsyncTrainer<A, E, R> where
 A: Unpin,
 E: Unpin,
 R: Unpin,
 <A as Policy<E>>::Config: Unpin,
 <E as Env>::Config: Unpin,
 <R as ReplayBufferBase>::Config: Unpin,
 <R as ReplayBufferBase>::PushedItem: Unpin,

impl<A, E, R> UnwindSafe for AsyncTrainer<A, E, R> where
 A: UnwindSafe,
 E: UnwindSafe,
 R: UnwindSafe,
 <A as Policy<E>>::Config: UnwindSafe,
 <E as Env>::Config: UnwindSafe,
 <R as ReplayBufferBase>::Config: UnwindSafe,

Blanket Implementations

source

impl<T> Any for T where
T: 'static + ?Sized,

source

pub fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

source

impl<T> Borrow<T> for T where
T: ?Sized,

const: unstable · source

pub fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

source

impl<T> BorrowMut<T> for T where
T: ?Sized,

const: unstable · source

pub fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

source

impl<T> From<T> for T

const: unstable · source

pub fn from(t: T) -> T

Performs the conversion.

source

impl<T, U> Into for T where
U: From<T>,

const: unstable · source

pub fn into(self) -> U

Performs the conversion.

impl<T> Pointable for T

pub const ALIGN: usize

The alignment of pointer.

type Init = T

The type for initializers.

pub unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more

pub unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more

pub unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more

pub unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more

source

impl<T, U> TryFrom for T where
U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

const: unstable · source

pub fn try_from(value: U) -> Result<T, <T as TryFrom>::Error>

Performs the conversion.

source

impl<T, U> TryInto for T where
U: TryFrom<T>,

type Error = >::Error

The type returned in the event of a conversion error.

const: unstable · source

pub fn try_into(self) -> Result<U, >::Error>

Performs the conversion.

Struct border_async_trainer::AsyncTrainer

Implementations

impl<A, E, R> AsyncTrainer<A, E, R> where
    A: Agent<E, R> + SyncModel,
    E: Env,
    R: ReplayBufferBase,
    R::PushedItem: Send + 'static,

pub fn train(
    &mut self,
    recorder: &mut impl Recorder,
    guard_init_env: Arc<Mutex<bool>>
) -> AsyncTrainStat

Auto Trait Implementations

impl<A, E, R> RefUnwindSafe for AsyncTrainer<A, E, R> where
    A: RefUnwindSafe,
    E: RefUnwindSafe,
    R: RefUnwindSafe,
    <A as Policy<E>>::Config: RefUnwindSafe,
    <E as Env>::Config: RefUnwindSafe,
    <R as ReplayBufferBase>::Config: RefUnwindSafe,

impl<A, E, R> Send for AsyncTrainer<A, E, R> where
    A: Send,
    E: Send,
    R: Send,
    <A as Policy<E>>::Config: Send,
    <E as Env>::Config: Send,
    <R as ReplayBufferBase>::Config: Send,
    <A as SyncModel>::ModelInfo: Send,

impl<A, E, R> Sync for AsyncTrainer<A, E, R> where
    A: Sync,
    E: Sync,
    R: Sync,
    <A as Policy<E>>::Config: Sync,
    <E as Env>::Config: Sync,
    <R as ReplayBufferBase>::Config: Sync,
    <A as SyncModel>::ModelInfo: Send,

impl<A, E, R> Unpin for AsyncTrainer<A, E, R> where
    A: Unpin,
    E: Unpin,
    R: Unpin,
    <A as Policy<E>>::Config: Unpin,
    <E as Env>::Config: Unpin,
    <R as ReplayBufferBase>::Config: Unpin,
    <R as ReplayBufferBase>::PushedItem: Unpin,

impl<A, E, R> UnwindSafe for AsyncTrainer<A, E, R> where
    A: UnwindSafe,
    E: UnwindSafe,
    R: UnwindSafe,
    <A as Policy<E>>::Config: UnwindSafe,
    <E as Env>::Config: UnwindSafe,
    <R as ReplayBufferBase>::Config: UnwindSafe,

Blanket Implementations

impl<T> Any for T where
T: 'static + ?Sized,

pub fn type_id(&self) -> TypeId

impl<T> Borrow<T> for T where
T: ?Sized,

pub fn borrow(&self) -> &T

impl<T> BorrowMut<T> for T where
T: ?Sized,

pub fn borrow_mut(&mut self) -> &mut T

impl<T> From<T> for T

pub fn from(t: T) -> T

impl<T, U> Into<U> for T where
U: From<T>,

pub fn into(self) -> U

impl<T> Pointable for T

pub const ALIGN: usize

type Init = T

pub unsafe fn init(init: <T as Pointable>::Init) -> usize

pub unsafe fn deref<'a>(ptr: usize) -> &'a T

pub unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

pub unsafe fn drop(ptr: usize)

impl<T, U> TryFrom<U> for T where
U: Into<T>,

type Error = Infallible

pub fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for T where
U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

pub fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

impl<V, T> VZip<V> for T where
V: MultiLane<T>,

pub fn vzip(self) -> V

Struct border_async_trainer::AsyncTrainer

Implementations

impl<A, E, R> AsyncTrainer<A, E, R> where A: Agent<E, R> + SyncModel, E: Env, R: ReplayBufferBase, R::PushedItem: Send + 'static,

pub fn build( config: &AsyncTrainerConfig, agent_config: &A::Config, env_config: &E::Config, replay_buffer_config: &R::Config, r_bulk_pushed_item: Receiver<PushedItemMessage<R::PushedItem>>, model_info_sender: Sender<(usize, A::ModelInfo)>, stop: Arc<Mutex<bool>>) -> Self

pub fn train( &mut self, recorder: &mut impl Recorder, guard_init_env: Arc<Mutex<bool>>) -> AsyncTrainStat

Auto Trait Implementations

impl<A, E, R> RefUnwindSafe for AsyncTrainer<A, E, R> where A: RefUnwindSafe, E: RefUnwindSafe, R: RefUnwindSafe, <A as Policy<E>>::Config: RefUnwindSafe, <E as Env>::Config: RefUnwindSafe, <R as ReplayBufferBase>::Config: RefUnwindSafe,

impl<A, E, R> Send for AsyncTrainer<A, E, R> where A: Send, E: Send, R: Send, <A as Policy<E>>::Config: Send, <E as Env>::Config: Send, <R as ReplayBufferBase>::Config: Send, <A as SyncModel>::ModelInfo: Send,

impl<A, E, R> Sync for AsyncTrainer<A, E, R> where A: Sync, E: Sync, R: Sync, <A as Policy<E>>::Config: Sync, <E as Env>::Config: Sync, <R as ReplayBufferBase>::Config: Sync, <A as SyncModel>::ModelInfo: Send,

impl<A, E, R> Unpin for AsyncTrainer<A, E, R> where A: Unpin, E: Unpin, R: Unpin, <A as Policy<E>>::Config: Unpin, <E as Env>::Config: Unpin, <R as ReplayBufferBase>::Config: Unpin, <R as ReplayBufferBase>::PushedItem: Unpin,

impl<A, E, R> UnwindSafe for AsyncTrainer<A, E, R> where A: UnwindSafe, E: UnwindSafe, R: UnwindSafe, <A as Policy<E>>::Config: UnwindSafe, <E as Env>::Config: UnwindSafe, <R as ReplayBufferBase>::Config: UnwindSafe,

Blanket Implementations

impl<T> Any for T where T: 'static + ?Sized,

pub fn type_id(&self) -> TypeId

impl<T> Borrow<T> for T where T: ?Sized,

pub fn borrow(&self) -> &T

impl<T> BorrowMut<T> for T where T: ?Sized,

pub fn borrow_mut(&mut self) -> &mut T

impl<T> From<T> for T

pub fn from(t: T) -> T

impl<T, U> Into<U> for T where U: From<T>,

pub fn into(self) -> U

impl<T> Pointable for T

pub const ALIGN: usize

type Init = T

pub unsafe fn init(init: <T as Pointable>::Init) -> usize

pub unsafe fn deref<'a>(ptr: usize) -> &'a T

pub unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

pub unsafe fn drop(ptr: usize)

impl<T, U> TryFrom<U> for T where U: Into<T>,

type Error = Infallible

pub fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for T where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

pub fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

impl<V, T> VZip<V> for T where V: MultiLane<T>,

pub fn vzip(self) -> V

impl<A, E, R> AsyncTrainer<A, E, R> where
A: Agent<E, R> + SyncModel,
E: Env,
R: ReplayBufferBase,
R::PushedItem: Send + 'static,

pub fn build(
config: &AsyncTrainerConfig,
agent_config: &A::Config,
env_config: &E::Config,
replay_buffer_config: &R::Config,
r_bulk_pushed_item: Receiver<PushedItemMessage<R::PushedItem>>,
model_info_sender: Sender<(usize, A::ModelInfo )>,
stop: Arc<Mutex<bool>>
) -> Self

pub fn train(
&mut self,
recorder: &mut impl Recorder,
guard_init_env: Arc<Mutex<bool>>
) -> AsyncTrainStat

impl<A, E, R> RefUnwindSafe for AsyncTrainer<A, E, R> where
A: RefUnwindSafe,
E: RefUnwindSafe,
R: RefUnwindSafe,
<A as Policy<E>>::Config: RefUnwindSafe,
<E as Env>::Config: RefUnwindSafe,
<R as ReplayBufferBase>::Config: RefUnwindSafe,

impl<A, E, R> Send for AsyncTrainer<A, E, R> where
A: Send,
E: Send,
R: Send,
<A as Policy<E>>::Config: Send,
<E as Env>::Config: Send,
<R as ReplayBufferBase>::Config: Send,
<A as SyncModel>::ModelInfo: Send,

impl<A, E, R> Sync for AsyncTrainer<A, E, R> where
A: Sync,
E: Sync,
R: Sync,
<A as Policy<E>>::Config: Sync,
<E as Env>::Config: Sync,
<R as ReplayBufferBase>::Config: Sync,
<A as SyncModel>::ModelInfo: Send,

impl<A, E, R> Unpin for AsyncTrainer<A, E, R> where
A: Unpin,
E: Unpin,
R: Unpin,
<A as Policy<E>>::Config: Unpin,
<E as Env>::Config: Unpin,
<R as ReplayBufferBase>::Config: Unpin,
<R as ReplayBufferBase>::PushedItem: Unpin,

impl<A, E, R> UnwindSafe for AsyncTrainer<A, E, R> where
A: UnwindSafe,
E: UnwindSafe,
R: UnwindSafe,
<A as Policy<E>>::Config: UnwindSafe,
<E as Env>::Config: UnwindSafe,
<R as ReplayBufferBase>::Config: UnwindSafe,

impl<T> Any for T where
T: 'static + ?Sized,

impl<T> Borrow<T> for T where
T: ?Sized,

impl<T> BorrowMut<T> for T where
T: ?Sized,

impl<T, U> Into<U> for T where
U: From<T>,

impl<T, U> TryFrom<U> for T where
U: Into<T>,

impl<T, U> TryInto<U> for T where
U: TryFrom<T>,

impl<V, T> VZip<V> for T where
V: MultiLane<T>,