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

clean IF args

Browse Source 
Summary: continued - avoid duplicate inputs

Reviewed By: davnov134

Differential Revision: D38248827

fbshipit-source-id: 91ed398e304496a936f66e7a70ab3d189eeb5c70
This commit is contained in:
Jeremy Reizenstein
2022-08-03 12:37:31 -07:00
committed by Facebook GitHub Bot
parent 078846d166
commit 46e82efb4e
6 changed files with 65 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/implicitron/data/overrides.yaml
View File

@@ -103,7 +103,6 @@ view_pooler_args:
weight_by_ray_angle_gamma: 1.0
min_ray_angle_weight: 0.1
implicit_function_IdrFeatureField_args:
feature_vector_size: 3
d_in: 3
d_out: 1
dims:
@@ -121,6 +120,5 @@ implicit_function_IdrFeatureField_args:
weight_norm: true
n_harmonic_functions_xyz: 1729
pooled_feature_dim: 0
encoding_dim: 0
view_metrics_ViewMetrics_args: {}
regularization_metrics_RegularizationMetrics_args: {}

5
tests/implicitron/test_srn.py
View File

@@ -55,9 +55,10 @@ class TestSRN(TestCaseMixin, unittest.TestCase):
def test_srn_hypernet_implicit_function(self):
# TODO investigate: If latent_dim_hypernet=0, why does this crash and dump core?
latent_dim_hypernet = 39
hypernet_args = {"latent_dim_hypernet": latent_dim_hypernet}
device = torch.device("cuda:0")
implicit_function = SRNHyperNetImplicitFunction(hypernet_args=hypernet_args)
implicit_function = SRNHyperNetImplicitFunction(
latent_dim_hypernet=latent_dim_hypernet
)
implicit_function.to(device)
global_code = torch.rand(_BATCH_SIZE, latent_dim_hypernet, device=device)
bundle = self._get_bundle(device=device)