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

Fix inversion between fine and coarse implicit_functions

Browse Source 
Summary: Fine implicit function was called before the coarse implicit function.

Reviewed By: shapovalov

Differential Revision: D46224224

fbshipit-source-id: 6b1cc00cc823d3ea7a5b42774c9ec3b73a69edb5
This commit is contained in:
Emilien Garreau
2023-05-26 08:33:54 -07:00
committed by Facebook GitHub Bot
parent e0c3ca97ff
commit 35badc0892
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/implicitron/models/test_overfit_model.py
View File

@@ -109,9 +109,9 @@ class TestOverfitModel(unittest.TestCase):
# Adapt the mapping from generic model to overfit model
mapping_om_from_gm = {
key.replace("_implicit_functions.0._fn", "implicit_function").replace(
"_implicit_functions.1._fn", "coarse_implicit_function"
): val
key.replace(
"_implicit_functions.0._fn", "coarse_implicit_function"
).replace("_implicit_functions.1._fn", "implicit_function"): val
for key, val in generic_model.state_dict().items()
}
# Copy parameters from generic_model to overfit_model