Skip to content

Instantly share code, notes, and snippets.

@Steboss89
Created August 3, 2022 20:29
Show Gist options
  • Save Steboss89/38c3f4217e79d35b3dd5d4d017f2fbae to your computer and use it in GitHub Desktop.
Save Steboss89/38c3f4217e79d35b3dd5d4d017f2fbae to your computer and use it in GitHub Desktop.
Start defining the default for a Linear NN
pub struct LinearConfig {
pub ws_init: super::Init,
pub bs_init: Option<super::Init>,
pub bias: bool,
}
impl Default for LinearConfig {
fn default() -> Self {
LinearConfig { ws_init: super::Init::KaimingUniform, bs_init: None, bias: true }
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment