Mirrors/the-algorithm-ml
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm-ml.git synced 2024-12-23 06:41:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove duplicate class DropoutConfig and fix typo in docstring

Browse Source
This commit is contained in:
Lakshit Farswan 2023-04-01 22:57:30 +05:30
parent 78c3235eee
commit ee7f8f7e64
1 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
projects/home/recap/model/config.py
View File

@ -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."""