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
}