File size: 425 Bytes
ad16788 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from abc import ABC
from abc import abstractmethod
from typing import Iterator
from typing import Tuple
from torch.utils.data import Sampler
class AbsSampler(Sampler, ABC):
@abstractmethod
def __len__(self) -> int:
raise NotImplementedError
@abstractmethod
def __iter__(self) -> Iterator[Tuple[str, ...]]:
raise NotImplementedError
def generate(self, seed):
return list(self)
|