mirror of
https://github.com/HumanAIGC/lite-avatar.git
synced 2026-02-04 17:39:19 +08:00
20 lines
425 B
Python
20 lines
425 B
Python
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)
|