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

load whole dataset in train loop

Browse Source 
Summary: Loads the whole dataset and moves it to the device and sends it to for sampling to enable full dataset heterogeneous raysampling.

Reviewed By: bottler

Differential Revision: D39263009

fbshipit-source-id: c527537dfc5f50116849656c9e171e868f6845b1
This commit is contained in:
Darijan Gudelj
2022-10-03 08:36:47 -07:00
committed by Facebook GitHub Bot
parent c311a4cbb9
commit 37bd280d19
3 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch3d/implicitron/models/generic_model.py
View File

@@ -389,7 +389,8 @@ class GenericModel(ImplicitronModelBase): # pyre-ignore: 13
)
# (1) Sample rendering rays with the ray sampler.
ray_bundle: ImplicitronRayBundle = self.raysampler( # pyre-fixme[29]
# pyre-ignore[29]
ray_bundle: ImplicitronRayBundle = self.raysampler(
target_cameras,
evaluation_mode,
mask=mask_crop[:n_targets]