a | b/reproducibility/embedders/abst.py | ||
---|---|---|---|
1 | from abc import ABC, abstractmethod |
||
2 | |||
3 | class AbstractEmbedder(ABC): |
||
4 | |||
5 | @abstractmethod |
||
6 | def image_embedder(self, images, device, num_workers, batch_size): |
||
7 | pass |
||
8 | |||
9 | @abstractmethod |
||
10 | def text_embedder(self, images, device, num_workers, batch_size): |
||
11 | pass |