Parameters for Training.
More...
#include <aifes_express_f32_fnn.h>
|
AIFES_E_loss | loss |
| Loss selection (AIFES_E_loss)
|
|
AIFES_E_optimizer | optimizer |
| Optimizer selection (AIFES_E_optimizer)
|
|
float | learn_rate |
| Learning rate for training (For all optimizers)
|
|
float | sgd_momentum |
| Optional momentum for SGD (Value 0.0f means Momentum off)
|
|
uint32_t | batch_size |
| Batch size (min: 1 -> max: Entire data set)
|
|
uint32_t | epochs |
| Number of desired epochs (If early stopping is on, can be stopped before)
|
|
uint32_t | epochs_loss_print_interval |
| Selection of the interval in which the loss is to be calculated and output via the print function.
|
|
void(* | loss_print_function )(float) |
| Individual print function for the loss. More...
|
|
AIFES_E_early_stopping | early_stopping |
| Switch early stopping on or off.
|
|
float | early_stopping_target_loss |
| If early stopping is switched on, the target loss can be specified here. More...
|
|
Parameters for Training.
The structure includes all parameters for training.
Example:
@ AIfES_E_mse
Mean squared error (MSE)
Definition: aifes_express_f32_fnn.h:53
@ AIfES_E_adam
ADAM.
Definition: aifes_express_f32_fnn.h:62
Parameters for Training.
Definition: aifes_express_f32_fnn.h:141
float learn_rate
Learning rate for training (For all optimizers)
Definition: aifes_express_f32_fnn.h:144
AIFES_E_early_stopping early_stopping
Switch early stopping on or off.
Definition: aifes_express_f32_fnn.h:165
float sgd_momentum
Optional momentum for SGD (Value 0.0f means Momentum off)
Definition: aifes_express_f32_fnn.h:145
void(* loss_print_function)(float)
Individual print function for the loss.
Definition: aifes_express_f32_fnn.h:164
uint32_t batch_size
Batch size (min: 1 -> max: Entire data set)
Definition: aifes_express_f32_fnn.h:146
float early_stopping_target_loss
If early stopping is switched on, the target loss can be specified here.
Definition: aifes_express_f32_fnn.h:166
AIFES_E_optimizer optimizer
Optimizer selection (AIFES_E_optimizer)
Definition: aifes_express_f32_fnn.h:143
uint32_t epochs_loss_print_interval
Selection of the interval in which the loss is to be calculated and output via the print function.
Definition: aifes_express_f32_fnn.h:148
AIFES_E_loss loss
Loss selection (AIFES_E_loss)
Definition: aifes_express_f32_fnn.h:142
uint32_t epochs
Number of desired epochs (If early stopping is on, can be stopped before)
Definition: aifes_express_f32_fnn.h:147
◆ early_stopping_target_loss
float early_stopping_target_loss |
If early stopping is switched on, the target loss can be specified here.
It is only checked in the print interval
◆ loss_print_function
void(* loss_print_function) (float) |
Individual print function for the loss.
The loss is calculated only in the previously selected interval and output via the function to which the pointer points The function must have the following structure
void my_print_function(
float loss)
{
printf(
"Loss: %f\n",
loss);
}
- Parameters
-
tensor | The tensor to print |
The documentation for this struct was generated from the following file: