diff options
author | Jan Buethe <jbuethe@amazon.de> | 2024-01-22 17:12:26 +0300 |
---|---|---|
committer | Jan Buethe <jbuethe@amazon.de> | 2024-01-22 17:12:26 +0300 |
commit | 5f8201c71ee81a9eb7bb302d1bd72fe15678d213 (patch) | |
tree | f9519d812e5b7ff8b500fb3b0dd403834a9a16f6 | |
parent | 6a9831a6b038638266165dc1e9f115678f0b330e (diff) |
OSCE_MAX_RNN_UNITS now derived from osce model parameters
-rw-r--r-- | dnn/nnet.c | 2 | ||||
-rw-r--r-- | dnn/osce.h | 3 | ||||
-rw-r--r-- | dnn/osce_config.h | 2 |
3 files changed, 4 insertions, 3 deletions
@@ -42,7 +42,7 @@ #include "vec.h" #ifdef ENABLE_OSCE -#include "osce_config.h" +#include "osce.h" #endif #ifdef NO_OPTIMIZATIONS @@ -52,10 +52,13 @@ #if !defined(DISABLE_NOLACE) #define OSCE_DEFAULT_METHOD OSCE_METHOD_NOLACE +#define OSCE_MAX_RNN_NEURONS NOLACE_FNET_GRU_STATE_SIZE #elif !defined(DISABLE_LACE) #define OSCE_DEFAULT_METHOD OSCE_METHOD_LACE +#define OSCE_MAX_RNN_NEURONS LACE_FNET_GRU_STATE_SIZE #else #define OSCE_DEFAULT_METHOD OSCE_METHOD_NONE +#define OSCE_MAX_RNN_NEURONS 0 #endif diff --git a/dnn/osce_config.h b/dnn/osce_config.h index 1543b803..0298befb 100644 --- a/dnn/osce_config.h +++ b/dnn/osce_config.h @@ -28,8 +28,6 @@ #ifndef OSCE_CONFIG #define OSCE_CONFIG -#define OSCE_MAX_RNN_NEURONS 256 - #define OSCE_FEATURES_MAX_HISTORY 350 #define OSCE_FEATURE_DIM 93 #define OSCE_MAX_FEATURE_FRAMES 4 |