data_lst

mabaorui · May 16, 2021 · fa751d1 · fa751d1
1 parent 34db8e9
commit fa751d1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/NeuralPull.py b/NeuralPull.py
@@ -274,7 +274,10 @@ def av_dist_None(array1, array2):
 
 #loss = tf.losses.huber_loss(points_target, g_points)
 #loss = chamfer_distance_tf_None(point_target_near, g_points)
-loss = chamfer_distance_tf_None(points_target, g_points)
+#loss = chamfer_distance_tf_None(points_target, g_points)
+l2_loss = tf.norm((points_target-g_points), axis=-1)
+print('l2_loss:',l2_loss)
+loss = tf.reduce_mean(l2_loss)
 
 
 t_vars = tf.trainable_variables()