Remove duplicate class DropoutConfig and fix typo in docstring

This commit is contained in:
Lakshit Farswan 2023-04-01 22:57:30 +05:30
parent 78c3235eee
commit ee7f8f7e64

View File

@ -18,7 +18,7 @@ class DropoutConfig(base_config.BaseConfig):
class LayerNormConfig(base_config.BaseConfig): class LayerNormConfig(base_config.BaseConfig):
"""Configruation for the layer normalization.""" """Configuration for the layer normalization."""
epsilon: float = pydantic.Field( epsilon: float = pydantic.Field(
1e-3, description="Small float added to variance to avoid dividing by zero." 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): class MlpConfig(base_config.BaseConfig):
"""Configuration for MLP model.""" """Configuration for MLP model."""