aiopti_adam_default.h

Go to the documentation of this file.

 
#ifndef AIOPTI_ADAM_DEFAULT
#define AIOPTI_ADAM_DEFAULT
 
#include "basic/base/aiopti/aiopti_adam.h"
 
#include "basic/default/aimath/aimath_f32_default.h"
#include "basic/default/aimath/aimath_q7_default.h"
 
#define AIOPTI_ADAM_F32(learning_rate, beta1, beta2, eps) {{{0,},},learning_rate, beta1, beta2, eps}
#define AIOPTI_ADAM_Q31(learning_rate, beta1, beta2, eps) {{{0,},},learning_rate, beta1, beta2, eps}
 
typedef struct aiopti_adam_f32  aiopti_adam_f32_t;
//typedef struct aiopti_adam_q31    aiopti_adam_q31_t;
 
struct aiopti_adam_f32 {
    aiopti_adam_t base; 
    aiscalar_f32_t learning_rate; 
    aiscalar_f32_t beta1; 
    aiscalar_f32_t beta2; 
    aiscalar_f32_t eps; 
 
    aiscalar_f32_t beta1t; 
    aiscalar_f32_t beta2t; 
    aiscalar_f32_t one_minus_beta1; 
    aiscalar_f32_t one_minus_beta2; 
    aiscalar_f32_t lrt; 
};
 
 
aiopti_t *aiopti_adam_f32_default(aiopti_adam_f32_t *opti);
 
void aiopti_adam_f32_default_begin_step(aiopti_t *self);
 
void aiopti_adam_f32_default_end_step(aiopti_t *self);
 
#endif // AIOPTI_ADAM_DEFAULT