Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "92b6847ee50be39743ac65ea9cf455ac319840e7"
Commit d42e0d3d authored by Jeremy Reizenstein's avatar Jeremy Reizenstein Committed by Facebook GitHub Bot
Browse files

1GPU for implicitron tests 

Reviewed By: shapovalov

Differential Revision: D38794764

fbshipit-source-id: 140c8a935d760bab8569d903cc52ac3dd73cd553
parent 76234576
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/implicitron/test_forward_pass.py
View file @ d42e0d3d
...@@ -32,7 +32,7 @@ class TestGenericModel(unittest.TestCase): ...@@ -32,7 +32,7 @@ class TestGenericModel(unittest.TestCase):
def test_gm(self): def test_gm(self):
# Simple test of a forward and backward pass of the default GenericModel. # Simple test of a forward and backward pass of the default GenericModel.
device = torch.device("cuda:1") device = torch.device("cuda:0")
expand_args_fields(GenericModel) expand_args_fields(GenericModel)
model = GenericModel(render_image_height=80, render_image_width=80) model = GenericModel(render_image_height=80, render_image_width=80)
model.to(device) model.to(device)
...@@ -116,7 +116,7 @@ class TestGenericModel(unittest.TestCase): ...@@ -116,7 +116,7 @@ class TestGenericModel(unittest.TestCase):
def test_idr(self): def test_idr(self):
# Forward pass of GenericModel with IDR. # Forward pass of GenericModel with IDR.
device = torch.device("cuda:1") device = torch.device("cuda:0")
args = get_default_args(GenericModel) args = get_default_args(GenericModel)
args.renderer_class_type = "SignedDistanceFunctionRenderer" args.renderer_class_type = "SignedDistanceFunctionRenderer"
args.implicit_function_class_type = "IdrFeatureField" args.implicit_function_class_type = "IdrFeatureField"
...@@ -153,7 +153,7 @@ class TestGenericModel(unittest.TestCase): ...@@ -153,7 +153,7 @@ class TestGenericModel(unittest.TestCase):
self.assertGreater(train_preds["objective"].item(), 0) self.assertGreater(train_preds["objective"].item(), 0)
def test_viewpool(self): def test_viewpool(self):
device = torch.device("cuda:1") device = torch.device("cuda:0")
args = get_default_args(GenericModel) args = get_default_args(GenericModel)
args.view_pooler_enabled = True args.view_pooler_enabled = True
args.image_feature_extractor_class_type = "ResNetFeatureExtractor" args.image_feature_extractor_class_type = "ResNetFeatureExtractor"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment