NERSC/pytorch-examples

refactor optimizer and learning-rate logic

Opened this issue · 0 comments

Implemented already in heptrkx-gnn-tracking. Add improved support for optimizer choices and LR scaling