Method __init__

eval-lipsync/script/SyncNetInstance.py:40–49

(
        self,
        net: torch.nn.Module,
        device: str = "cuda",
        dropout: float = 0,
        num_layers_in_fc_layers: int = 1024,
    )

class SyncNetInstance(torch.nn.Module):
    def __init__(
        self,
        net: torch.nn.Module,
        device: str = "cuda",
        dropout: float = 0,
        num_layers_in_fc_layers: int = 1024,
    ):
        super(SyncNetInstance, self).__init__()
        self.__S__ = net
        self.device = device

    def evaluate(self, opt):
        self.__S__.to(self.device)