Next up is dropout layers
This commit is contained in:
@@ -63,6 +63,12 @@ struct trainable_layer:layer{
|
||||
panic::tensor::real_matrix dweights;
|
||||
panic::tensor::real_vector dbiases;
|
||||
|
||||
panic::types::real_t weight_regularizer_l1;
|
||||
panic::types::real_t weight_regularizer_l2;
|
||||
|
||||
panic::types::real_t bias_regularizer_l1;
|
||||
panic::types::real_t bias_regularizer_l2;
|
||||
|
||||
/**
|
||||
* @brief Previous parameter updates used by momentum SGD.
|
||||
*
|
||||
@@ -82,6 +88,9 @@ struct trainable_layer:layer{
|
||||
panic::tensor::real_vector bias_cache;
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
virtual ~trainable_layer() = default;
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user