def _setup_loss(self): if self.loss[TYPE] == 'softmax_cross_entropy': self.train_loss_function = SequenceLoss() elif self.loss[TYPE] == 'sampled_softmax_cross_entropy': self.train_loss_function = SampledSoftmaxCrossEntropyLoss( decoder_obj=self.decoder_obj, num_classes=self.num_classes, feature_loss=self.loss, name='train_loss') else: raise ValueError("Loss type {} is not supported. Valid values are " "'softmax_cross_entropy' or " "'sampled_softmax_cross_entropy'".format( self.loss[TYPE])) self.eval_loss_function = SequenceLossMetric()
def __init__(self, name=None): super(PerplexityMetric, self).__init__(name=name) self.loss_function = SequenceLoss(from_logits=False)
def __init__(self, name=None): super(SequenceLossMetric, self).__init__(name=name) self.loss_function = SequenceLoss()