Summary: Use logging instead of printing in the internals of implicitron.

Reviewed By: davnov134

Differential Revision: D35247581

fbshipit-source-id: be5ddad5efe1409adbae0575d35ade6112b3be63
This commit is contained in:
Jeremy Reizenstein
2022-04-04 06:53:16 -07:00
committed by Facebook GitHub Bot
parent 6473aa316c
commit 199309fcf7
10 changed files with 65 additions and 35 deletions

View File

@@ -10,6 +10,7 @@ import sys
import unittest
import unittest.mock
if os.environ.get("FB_TEST", False):
from common_testing import get_pytorch3d_dir
else:

View File

@@ -24,6 +24,7 @@ from pytorch3d.implicitron.models.model_dbir import ModelDBIR
from pytorch3d.implicitron.tools.metric_utils import calc_psnr, eval_depth
from pytorch3d.implicitron.tools.utils import dataclass_to_cuda_
if os.environ.get("FB_TEST", False):
from .common_resources import get_skateboard_data, provide_lpips_vgg
else: