forked from nudt_dsp/netrans
10 lines
410 B
Python
10 lines
410 B
Python
from typing import TypeVar, Optional, Iterator
|
|
from . import Sampler, Dataset
|
|
|
|
T_co = TypeVar('T_co', covariant=True)
|
|
class DistributedSampler(Sampler[T_co]):
|
|
def __init__(self, dataset: Dataset, num_replicas: Optional[int]=..., rank: Optional[int]=..., shuffle: bool=...): ...
|
|
def __iter__(self) -> Iterator[int]: ...
|
|
def __len__(self) -> int: ...
|
|
def set_epoch(self, epoch: int) -> None: ...
|