ruda-nn 0.21.26

Ruda neural network layers, activation modules and losses.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18


use ruda_model::config::Config;

/// The reduction type for the loss.
#[derive(Config, Debug)]
pub enum Reduction {
    /// The mean of the losses will be returned.
    Mean,

    /// The sum of the losses will be returned.
    Sum,

    /// The sum of the losses divided by the batch_size will be returned.
    BatchMean,

    /// The mean of the losses will be returned.
    Auto,
}