Jeremy Reizenstein 7be49bf46f allow dots in param_groups ...

Summary:
Allow a module's param_group member to specify overrides to the param groups of its members or their members.
Also logging for param group assignments.

This allows defining `params.basis_matrix` in the param_groups of a voxel_grid.

Reviewed By: shapovalov

Differential Revision: D41080667

fbshipit-source-id: 49f3b0e5b36e496f78701db0699cbb8a7e20c51e

2022-11-07 06:41:40 -08:00

..

__init__.py

Updates to support Accelerate and multigpu training (#37)

2022-07-11 19:29:58 -07:00

model_factory.py

Move load_stats to TrainingLoop

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

optimizer_factory.py

allow dots in param_groups

2022-11-07 06:41:40 -08:00

training_loop.py

load whole dataset in train loop

2022-10-03 08:36:47 -07:00

utils.py

Better seeding of random engines

2022-08-01 10:03:09 -07:00