def _loss_mse(self): sq = tf.squared_difference(self.sigm_network, self.desired_heatmap) loss = self._adjust_loss(sq) return loss