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

move targets

Browse Source 
Summary: Move testing targets from pytorch3d/tests/TARGETS to pytorch3d/TARGETS.

Reviewed By: shapovalov

Differential Revision: D36186940

fbshipit-source-id: a4c52c4d99351f885e2b0bf870532d530324039b
This commit is contained in:
Jeremy Reizenstein
2022-05-25 06:16:03 -07:00
committed by Facebook GitHub Bot
parent f625fe1f8b
commit 34f648ede0
121 changed files with 231 additions and 239 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_render_multigpu.py
View File

@@ -8,7 +8,6 @@ import unittest
import torch
import torch.nn as nn
from common_testing import get_random_cuda_device, TestCaseMixin
from pytorch3d.renderer import (
AlphaCompositor,
BlendParams,
@@ -28,6 +27,8 @@ from pytorch3d.renderer.cameras import FoVPerspectiveCameras, look_at_view_trans
from pytorch3d.structures import Meshes, Pointclouds
from pytorch3d.utils.ico_sphere import ico_sphere
from .common_testing import get_random_cuda_device, TestCaseMixin
# Set the number of GPUS you want to test with
NUM_GPUS = 3