burn_central_runtime/
types.rs

1use burn::prelude::Backend;
2use burn_central_core::experiment::ExperimentArgs;
3use derive_more::{Deref, From};
4
5#[derive(From, Deref)]
6pub struct Args<T: ExperimentArgs>(pub T);
7#[derive(Clone, Debug, Deref, From)]
8pub struct MultiDevice<B: Backend>(pub Vec<B::Device>);
9#[derive(Clone, From, Deref)]
10pub struct Model<M>(pub M);
11#[derive(Debug, Deref, From)]
12pub struct In<T>(pub T);
13#[derive(Debug, Deref, From)]
14pub struct Out<T>(pub T);
15#[derive(Debug, Deref, From)]
16pub struct State<T>(pub T);