mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-11-19 06:29:22 +01:00
15 lines
361 B
Python
15 lines
361 B
Python
|
from tml.core.config import base_config
|
||
|
|
|
||
|
|
import pydantic
|
||
|
|
|
||
|
|
|
||
|
|
class TwhinDataConfig(base_config.BaseConfig):
|
||
|
|
data_root: str
|
||
|
|
per_replica_batch_size: pydantic.PositiveInt
|
||
|
|
global_negatives: int
|
||
|
|
in_batch_negatives: int
|
||
|
|
limit: pydantic.PositiveInt
|
||
|
|
offset: pydantic.PositiveInt = pydantic.Field(
|
||
|
|
None, description="The offset to start reading from."
|
||
|
|
)
|