dataparallel fix

Summary: Attempt to overcome flaky test Reviewed By: patricklabatut Differential Revision: D31895560 fbshipit-source-id: 1ecbb1782b0eafe132f88425c48487c2d0e10d2d

dataparallel fix
Summary: Attempt to overcome flaky test Reviewed By: patricklabatut Differential Revision: D31895560 fbshipit-source-id: 1ecbb1782b0eafe132f88425c48487c2d0e10d2d
73a14d72 · Jeremy Reizenstein · Facebook GitHub Bot · bee31c48 · 73a14d72
Commit 73a14d72 authored Oct 26, 2021 by Jeremy Reizenstein Committed by Facebook GitHub Bot Oct 26, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_render_multigpu.py tests/test_render_multigpu.py +1 -1

No files found.
--- a/tests/test_render_multigpu.py
+++ b/tests/test_render_multigpu.py
@@ -159,8 +159,8 @@ class TestRenderMeshesMultiGPU(TestCaseMixin, unittest.TestCase):
        verts = ico_sphere(3).verts_padded()
        texs = verts.new_ones(verts.shape)
        model = Model()
+        model.to(GPU_LIST[0])
        model = nn.DataParallel(model, device_ids=GPU_LIST)
-        model.to(f"cuda:{model.device_ids[0]}")

        # Test a few iterations
        for _ in range(100):