fb3d83430def1d0939126f45fdb96b270933a357,keras_rcnn/layers/losses/_rcnn.py,RCNNRegressionLoss,call,#RCNNRegressionLoss#Any#Any#,39

Before Change

    def call(self, inputs, training=None, **kwargs):
        output, target, labels_target = inputs

        loss = keras.backend.in_train_phase(lambda: self.compute_regression_loss(output, target, labels_target), keras.backend.variable(0), training=training)

        self.add_loss(loss, inputs)

        return output

After Change

        def calculate_loss():
            return self.compute_regression_loss(output, target, labels_target)
        
        loss = tensorflow.cond(keras.backend.not_equal(keras.backend.shape(output)[1], keras.backend.shape(target)[1]), no_loss, calculate_loss)

        self.add_loss(loss, inputs)

Instances