Skip to content

Add optimizer-provider and pg_collection hooks to setup_model_and_optimizer - #5332

Closed
yashaswikarnati wants to merge 1 commit into
NVIDIA:mainfrom
yashaswikarnati:ykarnati/upstream-trainloop-tl-opt
Closed

Add optimizer-provider and pg_collection hooks to setup_model_and_optimizer#5332
yashaswikarnati wants to merge 1 commit into
NVIDIA:mainfrom
yashaswikarnati:ykarnati/upstream-trainloop-tl-opt

training: allow injecting optimizer factory and pg_collection into setup

b3913cc
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jun 14, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO