From ee7f8f7e64c9861f43a98f11382ae78da560bf5f Mon Sep 17 00:00:00 2001 From: Lakshit Farswan Date: Sat, 1 Apr 2023 22:57:30 +0530 Subject: [PATCH] Remove duplicate class DropoutConfig and fix typo in docstring --- projects/home/recap/model/config.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/projects/home/recap/model/config.py b/projects/home/recap/model/config.py index 47d0640..407b7c8 100644 --- a/projects/home/recap/model/config.py +++ b/projects/home/recap/model/config.py @@ -18,7 +18,7 @@ class DropoutConfig(base_config.BaseConfig): class LayerNormConfig(base_config.BaseConfig): - """Configruation for the layer normalization.""" + """Configuration for the layer normalization.""" epsilon: float = pydantic.Field( 1e-3, description="Small float added to variance to avoid dividing by zero." @@ -110,14 +110,6 @@ class FeaturizationConfig(base_config.BaseConfig): ) -class DropoutConfig(base_config.BaseConfig): - """Configuration for the dropout layer.""" - - rate: pydantic.PositiveFloat = pydantic.Field( - 0.1, description="Fraction of inputs to be dropped." - ) - - class MlpConfig(base_config.BaseConfig): """Configuration for MLP model."""