From c2a0a3e3ba86b46417937d8893f5883573e4ec85 Mon Sep 17 00:00:00 2001 From: Georgia Gkioxari Date: Fri, 21 Feb 2020 15:54:14 -0800 Subject: [PATCH] fix docstring of mesh edge loss Summary: Fix docstring for mesh edge loss Reviewed By: jcjohnson Differential Revision: D20040560 fbshipit-source-id: 01a3ee9473c7d11583684bf4cd200caa1d3f0260 --- pytorch3d/loss/mesh_edge_loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch3d/loss/mesh_edge_loss.py b/pytorch3d/loss/mesh_edge_loss.py index 77eaa9dc..00d235ad 100644 --- a/pytorch3d/loss/mesh_edge_loss.py +++ b/pytorch3d/loss/mesh_edge_loss.py @@ -7,8 +7,8 @@ import torch def mesh_edge_loss(meshes, target_length: float = 0.0): """ Computes mesh edge length regularization loss averaged across all meshes - in a batch. Each edge contributes equally to the final loss, regardless of - numbers of edges per mesh in the batch by weighting each mesh with the + in a batch. Each mesh contributes equally to the final loss, regardless of + the number of edges per mesh in the batch by weighting each mesh with the inverse number of edges. For example, if mesh 3 (out of N) has only E=4 edges, then the loss for each edge in mesh 3 should be multiplied by 1/E to contribute to the final loss.