423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 20:02:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix link in generic_model.py (#38)

Browse Source 
Summary: Pull Request resolved: https://github.com/fairinternal/pytorch3d/pull/38

Reviewed By: ignacio-rocco

Differential Revision: D37415027

Pulled By: bottler

fbshipit-source-id: 9b17049e4762506cd5c152fd6e244d5f0d97855b
This commit is contained in:
Ignacio Rocco 2022-06-24 06:41:06 -07:00 committed by Facebook GitHub Bot
parent 879495d38f
commit 0e4c53c612
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch3d/implicitron/models/generic_model.py
View File

@ -121,7 +121,7 @@ class GenericModel(ImplicitronModelBase, torch.nn.Module): # pyre-ignore: 13
this sequence of steps. Currently, steps 1, 3, 4, 5, 6 this sequence of steps. Currently, steps 1, 3, 4, 5, 6
can be customized by intializing a subclass of the appropriate can be customized by intializing a subclass of the appropriate
baseclass and adding the newly created module to the registry. baseclass and adding the newly created module to the registry.
Please see https://github.com/fairinternal/pytorch3d/blob/co3d/projects/implicitron_trainer/README.md#custom-plugins Please see https://github.com/facebookresearch/pytorch3d/blob/main/projects/implicitron_trainer/README.md#custom-plugins
for more details on how to create and register a custom component. for more details on how to create and register a custom component.
In the config .yaml files for experiments, the parameters below are In the config .yaml files for experiments, the parameters below are
@ -189,7 +189,7 @@ class GenericModel(ImplicitronModelBase, torch.nn.Module): # pyre-ignore: 13
log_vars: A list of variable names which should be logged. log_vars: A list of variable names which should be logged.
The names should correspond to a subset of the keys of the The names should correspond to a subset of the keys of the
dict `preds` output by the `forward` function. dict `preds` output by the `forward` function.
""" """ # noqa: B950
mask_images: bool = True mask_images: bool = True
mask_depths: bool = True mask_depths: bool = True