def _create_squared_loss(self, prev_layer, layer_name):
with tf.variable_scope(layer_name) as scope:
input_tensor, class_tensor, box_tensor = self.placeholder
loss = tf.reduce_mean(tf.reduce_sum(tf.squared_difference(box_tensor,prev_layer),reduction_indices=[1]))
return loss
评论列表
文章目录