pub struct JemConfig {
pub d_input: usize,
pub n_classes: usize,
pub n_hidden: usize,
pub sgld_steps: usize,
pub sgld_step_size: f32,
pub sgld_noise: f32,
pub buffer_size: usize,
}Expand description
Hyper-parameters for the Jem joint energy model.
Fields§
§d_input: usizeInput dimension (feature / latent vector length).
n_classes: usizeNumber of output classes.
Hidden layer dimension (single hidden layer MLP).
sgld_steps: usizeNumber of SGLD steps per Jem::sgld_step call.
sgld_step_size: f32SGLD gradient step size α.
sgld_noise: f32SGLD noise scale σ applied to the Langevin noise term.
buffer_size: usizeNumber of persistent MCMC chains stored in the replay buffer.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for JemConfig
impl RefUnwindSafe for JemConfig
impl Send for JemConfig
impl Sync for JemConfig
impl Unpin for JemConfig
impl UnsafeUnpin for JemConfig
impl UnwindSafe for JemConfig
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