Struct TiRexConfig 

pub struct TiRexConfig {
    pub patch_size: usize,
    pub num_blocks: usize,
    pub embedding_dim: usize,
    pub num_heads: usize,
    pub input_ff_dim: usize,
    pub ffn_up_dim: usize,
    pub train_ctx_len: usize,
    pub quantiles: Vec<f32>,
    pub num_quantiles: usize,
}

TiRex model configuration (fixed from checkpoint hyper_parameters).

Fields

patch_size: usize

num_blocks: usize

embedding_dim: usize

num_heads: usize

input_ff_dim: usize

ffn_up_dim: usize

train_ctx_len: usize

quantiles: Vec<f32>

num_quantiles: usize

Implementations

impl TiRexConfig

pub fn default_from_ckpt() -> Self

pub fn head_dim(&self) -> usize

pub fn num_patches(&self) -> usize

pub fn output_dim(&self) -> usize

pub fn input_dim(&self) -> usize

Trait Implementations

impl Clone for TiRexConfig

fn clone(&self) -> TiRexConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.