From 6a58f175c5e8db518a900b6afc9027e734834aab Mon Sep 17 00:00:00 2001 From: Amir Bawab Date: Fri, 14 Jun 2019 12:40:47 -0400 Subject: [PATCH] Updated some parameters from double to float --- src/nn-builder/src/builtins/activation.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/nn-builder/src/builtins/activation.h b/src/nn-builder/src/builtins/activation.h index 1225ef2..c6f6f9a 100644 --- a/src/nn-builder/src/builtins/activation.h +++ b/src/nn-builder/src/builtins/activation.h @@ -11,9 +11,9 @@ struct ActivationFunction { wabt::Var derivative; }; struct ActivationOptions { - double linear_slope = 1; - double leaky_relu_slope = 0.01; - double elu_slope = 0.01; + float linear_slope = 1; + float leaky_relu_slope = 0.01; + float elu_slope = 0.01; }; class Activation : public Builtin { private: