def output_types(self): return { "loss": NeuralType(elements_type=LossType()), "mel_loss": NeuralType(elements_type=LossType()), "dur_loss": NeuralType(elements_type=LossType()), "pitch_loss": NeuralType(elements_type=LossType()), }
def output_types(self): return { 'loss': NeuralType(elements_type=LossType()), }
def output_types(self): return { "sc_loss": NeuralType(elements_type=LossType()), "mag_loss": NeuralType(elements_type=LossType()), }
def output_types(self): return { "forward_sum_loss": NeuralType(elements_type=LossType()), }
def output_types(self): return { "loss": NeuralType(elements_type=LossType()), "real_losses": NeuralType(elements_type=LossType()), "fake_losses": NeuralType(elements_type=LossType()), }
def output_types(self): return { "loss": NeuralType(elements_type=LossType()), "gate_target": NeuralType(('B', 'T'), LogitsType()), # Used for evaluation }