Struct ModelRegistry 

pub struct ModelRegistry { /* private fields */ }

Model registry for managing all supported architectures

Implementations

impl ModelRegistry

pub fn new() -> Self

pub fn register<F>(&mut self, architecture: &str, factory: F)

where F: ModelFactory + 'static,

Register a model factory

pub fn create_model( &self, architecture: &str, config_json: &str, ) -> Result<Box<dyn Model<Config = Box<dyn ModelConfig>>>>

Create model by architecture name

pub fn supported_architectures(&self) -> Vec<String>

Get all supported architectures

pub fn supports(&self, architecture: &str) -> bool

Check if architecture is supported