InferenceStep

burn_train

Trait InferenceStep 

Source 
pub trait InferenceStep {
    type Input: Send + 'static;
    type Output: ItemLazy + 'static;

    // Required method
    fn step(&self, item: Self::Input) -> Self::Output;
}
Expand description

Trait to be implemented for validating models.

Required Associated Types§

Source

type Input: Send + 'static

Type of input for an inference step.

Source

type Output: ItemLazy + 'static

Type of output for an inference step.

Required Methods§

Source

fn step(&self, item: Self::Input) -> Self::Output

Runs a validation step.

§Arguments
  • item - The item to validate on.
§Returns

The validation output.

Implementors§