Jeremy Reizenstein 305cf32f6b Avoid raysampler dict ...

Summary:
A significant speedup (e.g. >2% of a forward pass).

Move NDCMultinomialRaysampler parts of AbstractMaskRaySampler to members instead of living in a dict. The dict was hiding them from the nn.Module system so their _xy_grid members were remaining on the CPU. Therefore they were being copied to the GPU in every forward pass.

(We couldn't easily use a ModuleDict here because the enum keys are not strs.)

Reviewed By: shapovalov

Differential Revision: D39668589

fbshipit-source-id: 719b88e4a08fd7263a284e0ab38189e666bd7e3a

2022-09-21 04:29:44 -07:00

..

feature_extractor

suppress errors in vision/fair/pytorch3d

2022-06-15 06:27:35 -07:00

global_encoder

more globalencoder followup

2022-08-05 03:33:30 -07:00

implicit_function

raybundle input to ImplicitFunctions -> api unification

2022-09-05 06:26:06 -07:00

renderer

Avoid raysampler dict

2022-09-21 04:29:44 -07:00

view_pooler

omegaconf 2.2.2 compatibility

2022-06-24 04:18:01 -07:00

visualization

visualize_reconstruction fixes

2022-09-07 20:10:07 -07:00

__init__.py

add missing __init__.py files

2022-03-24 07:04:38 -07:00

base_model.py

Move load_stats to TrainingLoop

2022-08-02 15:40:53 -07:00

generic_model.py

remove stray "generic_model_args" references

2022-08-18 07:18:55 -07:00

metrics.py

followups to D37592429

2022-07-13 06:07:02 -07:00

model_dbir.py

Replace pluggable components to create a proper Configurable hierarchy.

2022-07-29 17:32:51 -07:00