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

dataparallel fix

Browse Source 
Summary: Attempt to overcome flaky test

Reviewed By: patricklabatut

Differential Revision: D31895560

fbshipit-source-id: 1ecbb1782b0eafe132f88425c48487c2d0e10d2d
This commit is contained in:
Jeremy Reizenstein 2021-10-26 14:32:34 -07:00 committed by Facebook GitHub Bot
parent bee31c48d3
commit 73a14d7266
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_render_multigpu.py
View File

@ -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):