Log In Sign Up
Models:
Daniel Galbraith
DanielG/
plip
0
Downloads: 1
[278d8a]: / reproducibility / embedders / abst.py

Download this file

11 lines (8 with data), 274 Bytes 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
from abc import ABC, abstractmethod



class AbstractEmbedder(ABC):



    @abstractmethod

    def image_embedder(self, images, device, num_workers, batch_size):

        pass



    @abstractmethod

    def text_embedder(self, images, device, num_workers, batch_size):

        pass