Method __init__

code/train/model_paper.py:124–130

(self, output_size, keep_pro)

        keep_pro: float, the keep probability of the dropout layer
    """
    def __init__(self, output_size, keep_pro):
        super(Audio2Face, self).__init__()
        self.output_size = output_size
        self.keep_pro = keep_pro
        self.FormantLayer = FormantLayer()
        self.ArticulationLayer = ArticulationLayer()
        self.OutputLayer = OutputLayer(self.output_size, self.keep_pro)
        
    def call(self, x):
        x = self.FormantLayer(x)