aialgo_sequential_training.h

Go to the documentation of this file.

 
#ifndef AIALGO_SEQUENTIAL_TRAINING
#define AIALGO_SEQUENTIAL_TRAINING
 
#include "core/aifes_core.h"
#include "core/aifes_math.h"
#include "basic/base/aimath/aimath_basic.h"
#include "basic/base/aimath/aimath_q31.h"
 
uint32_t aialgo_sizeof_training_memory(aimodel_t *model, aiopti_t *optimizer);
 
uint8_t aialgo_schedule_training_memory(aimodel_t *model, aiopti_t *optimizer, void *memory_ptr, uint32_t memory_size);
 
void aialgo_init_model_for_training(aimodel_t *model, aiopti_t *optimizer);
 
void aialgo_backward_model(aimodel_t *model, aitensor_t *target_data);
 
uint8_t aialgo_train_model(aimodel_t *model, aitensor_t *input_tensor, aitensor_t *target_tensor, aiopti_t *optimizer, uint32_t batch_size);
 
uint8_t aialgo_calc_loss_model_f32(aimodel_t *model, aitensor_t *input_data, aitensor_t *target_data, float *result);
 
uint8_t aialgo_calc_loss_model_q31(aimodel_t *model, aitensor_t *input_data, aitensor_t *target_data, aiscalar_q31_t *result);
 
void aialgo_zero_gradients_model(aimodel_t *model, aiopti_t *optimizer);
 
void aialgo_update_params_model(aimodel_t *model, aiopti_t *optimizer);
 
void aialgo_print_loss_specs(ailoss_t *loss);
 
void aialgo_print_optimizer_specs(aiopti_t *opti);
 
void aialgo_initialize_parameters_model(aimodel_t *model);
 
#endif // AIALGO_SEQUENTIAL_TRAINING