fix docstring of mesh edge loss

Summary: Fix docstring for mesh edge loss

Reviewed By: jcjohnson

Differential Revision: D20040560

fbshipit-source-id: 01a3ee9473c7d11583684bf4cd200caa1d3f0260
This commit is contained in:
Georgia Gkioxari 2020-02-21 15:54:14 -08:00 committed by Facebook Github Bot
parent e491efb81f
commit c2a0a3e3ba

View File

@ -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.