423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-14 11:26:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

suppress errors in vision/fair/pytorch3d

Browse Source 
Reviewed By: kjchalup

Differential Revision: D39198333

fbshipit-source-id: 3f4ebcf625215f21d165073837578ff69b05f72d
This commit is contained in:
Pyre Bot Jr
2022-09-01 11:46:55 -07:00
committed by Facebook GitHub Bot
parent d19e6243d0
commit c80e5fd07a
9 changed files with 41 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
projects/implicitron_trainer/experiment.py
View File

@@ -207,6 +207,7 @@ class Experiment(Configurable): # pyre-ignore: 13
val_loader,
) = accelerator.prepare(model, optimizer, train_loader, val_loader)
# pyre-fixme[16]: Optional type has no attribute `is_multisequence`.
if not self.training_loop.evaluator.is_multisequence:
all_train_cameras = self.data_source.all_train_cameras
else:

4
projects/implicitron_trainer/impl/training_loop.py
View File

@@ -29,6 +29,7 @@ from .utils import seed_all_random_engines
logger = logging.getLogger(__name__)
# pyre-fixme[13]: Attribute `evaluator` is never initialized.
class TrainingLoopBase(ReplaceableBase):
"""
Members:
@@ -62,7 +63,7 @@ class TrainingLoopBase(ReplaceableBase):
@registry.register
class ImplicitronTrainingLoop(TrainingLoopBase): # pyre-ignore [13]
class ImplicitronTrainingLoop(TrainingLoopBase):
"""
Members:
eval_only: If True, only run evaluation using the test dataloader.
@@ -137,6 +138,7 @@ class ImplicitronTrainingLoop(TrainingLoopBase): # pyre-ignore [13]
# only run evaluation on the test dataloader
if self.eval_only:
if test_loader is not None:
# pyre-fixme[16]: `Optional` has no attribute `run`.
self.evaluator.run(
all_train_cameras=all_train_cameras,
dataloader=test_loader,