Skip to content
This repository was archived by the owner on Nov 22, 2022. It is now read-only.

Change default batch sampler DisjointMultitaskData to RoundRobinBatchSampler #802

Closed
wants to merge 1 commit into from
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 8 additions & 2 deletions pytext/data/disjoint_multitask_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,13 @@

from pytext.common.constants import BatchContext, Stage
from pytext.config.component import Component, ComponentType, create_component
from pytext.data import BaseBatchSampler, Data, EvalBatchSampler, generator_iterator
from pytext.data import (
BaseBatchSampler,
Data,
EvalBatchSampler,
RoundRobinBatchSampler,
generator_iterator,
)
from pytext.data.data import BatchData


Expand All @@ -26,7 +32,7 @@ class DisjointMultitaskData(Data):
"""

class Config(Component.Config):
sampler: BaseBatchSampler.Config = EvalBatchSampler.Config()
sampler: BaseBatchSampler.Config = RoundRobinBatchSampler.Config()
test_key: Optional[str] = None

@classmethod
Expand Down