Trait TrainingConfiguration

Source

pub trait TrainingConfiguration<T>: NetworkConfig<T> {
    // Required methods
    fn epochs(&self) -> usize;
    fn batch_size(&self) -> usize;

    // Provided methods
    fn learning_rate(&self) -> Option<&T> { ... }
    fn momentum(&self) -> Option<&T> { ... }
    fn weight_decay(&self) -> Option<&T> { ... }
    fn dropout(&self) -> Option<&T> { ... }
    fn decay(&self) -> Option<&T> { ... }
    fn beta1(&self) -> Option<&T> { ... }
    fn beta2(&self) -> Option<&T> { ... }
    fn epsilon(&self) -> Option<&T> { ... }
    fn gradient_clip(&self) -> Option<&T> { ... }
}