Struct RepformersConfig

Source

pub struct RepformersConfig {Show 28 fields
    pub rcut: f64,
    pub rcut_smth: f64,
    pub sel: usize,
    pub ntypes: usize,
    pub nlayers: usize,
    pub g1_dim: usize,
    pub g2_dim: usize,
    pub axis_neuron: usize,
    pub update_g1_has_conv: bool,
    pub update_g1_has_drrd: bool,
    pub update_g1_has_grrg: bool,
    pub update_g1_has_attn: bool,
    pub update_g2_has_g1g1: bool,
    pub update_g2_has_attn: bool,
    pub update_h2: bool,
    pub attn1_hidden: usize,
    pub attn1_nhead: usize,
    pub attn2_hidden: usize,
    pub attn2_nhead: usize,
    pub attn2_has_gate: bool,
    pub activation_function: String,
    pub update_style: String,
    pub smooth: bool,
    pub use_sqrt_nnei: bool,
    pub g1_out_conv: bool,
    pub g1_out_mlp: bool,
    pub ln_eps: f32,
    pub epsilon: f32,
}

Expand description

Top-level repformer-block config (RepformerArgs in Python).

Fields§

§rcut: f64§rcut_smth: f64§sel: usize§ntypes: usize§nlayers: usize§g1_dim: usize§g2_dim: usize§axis_neuron: usize§update_g1_has_conv: bool§update_g1_has_drrd: bool§update_g1_has_grrg: bool§update_g1_has_attn: bool§update_g2_has_g1g1: bool§update_g2_has_attn: bool§update_h2: bool§attn1_hidden: usize§attn1_nhead: usize§attn2_hidden: usize§attn2_nhead: usize§attn2_has_gate: bool§activation_function: String§update_style: String§smooth: bool§use_sqrt_nnei: bool§g1_out_conv: bool§g1_out_mlp: bool§ln_eps: f32§epsilon: f32

Struct RepformersConfig Copy item path

Fields§

Implementations§

impl RepformersConfig

pub fn ng1(&self) -> usize

pub fn ng2(&self) -> usize

Trait Implementations§

impl Clone for RepformersConfig

fn clone(&self) -> RepformersConfig

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

impl Debug for RepformersConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

impl<'de> Deserialize<'de> for RepformersConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where __D: Deserializer<'de>,

impl Serialize for RepformersConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>where __S: Serializer,

Auto Trait Implementations§

impl Freeze for RepformersConfig

impl RefUnwindSafe for RepformersConfig

impl Send for RepformersConfig

impl Sync for RepformersConfig

impl Unpin for RepformersConfig

impl UnsafeUnpin for RepformersConfig

impl UnwindSafe for RepformersConfig

Blanket Implementations§

impl<T> Any for Twhere T: 'static + ?Sized,

fn type_id(&self) -> TypeId

impl<T> Borrow<T> for Twhere T: ?Sized,

fn borrow(&self) -> &T

impl<T> BorrowMut<T> for Twhere T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

impl<T> CloneToUninit for Twhere T: Clone,

unsafe fn clone_to_uninit(&self, dest: *mut u8)

impl<T> From<T> for T

fn from(t: T) -> T

impl<T, U> Into<U> for Twhere U: From<T>,

fn into(self) -> U

impl<T> ToOwned for Twhere T: Clone,

type Owned = T

fn to_owned(&self) -> T

fn clone_into(&self, target: &mut T)

impl<T, U> TryFrom<U> for Twhere U: Into<T>,

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for Twhere U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

impl<T> DeserializeOwned for Twhere T: for<'de> Deserialize<'de>,

Struct RepformersConfig

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where __D: Deserializer<'de>,

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where S: Serializer,

impl<T> Any for T
where T: 'static + ?Sized,

impl<T> Borrow<T> for T
where T: ?Sized,

impl<T> BorrowMut<T> for T
where T: ?Sized,

impl<T> CloneToUninit for T
where T: Clone,

impl<T, U> Into<U> for T
where U: From<T>,

impl<T> ToOwned for T
where T: Clone,

impl<T, U> TryFrom<U> for T
where U: Into<T>,

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

impl<T> DeserializeOwned for T
where T: for<'de> Deserialize<'de>,