pub struct Config {Show 14 fields
pub d_model: usize,
pub encoder_heads: usize,
pub decoder_heads: usize,
pub decoder_points: usize,
pub num_queries: usize,
pub feat_strides: Vec<usize>,
pub aifi_level: usize,
pub act: Activation,
pub encoder_act: Activation,
pub decoder_act: Activation,
pub layer_norm_eps: f32,
pub pos_temperature: f64,
pub pointer_dim: usize,
pub labels: Vec<String>,
}Expand description
推理要用的配置。
Fields§
§d_model: usize§encoder_heads: usize§decoder_heads: usize§decoder_points: usize§num_queries: usize§feat_strides: Vec<usize>编码器、解码器输入各层特征的步长(相对网络输入)。
aifi_level: usize做 AIFI(层内自注意力)的那一层在 feat_strides 里的下标。
act: ActivationCSP 块、下采样卷积等的激活(activation_function)。
encoder_act: Activation§decoder_act: Activation§layer_norm_eps: f32§pos_temperature: f64§pointer_dim: usize§labels: Vec<String>类别号 → 名字(id2label)。
Implementations§
Trait Implementations§
Auto Trait Implementations§
impl Forget for Config
impl Freeze for Config
impl RefUnwindSafe for Config
impl Send for Config
impl Sync for Config
impl Unpin for Config
impl UnsafeUnpin for Config
impl UnwindSafe for Config
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more