Closed xrsrke closed 9 months ago
Support the following callback events in DistributedOptimizer
DistributedOptimizer
Events
on_after_init
on_before_step
on_after_step
on_before_zero_grad
on_after_zero_grad
Support the following callback events in
DistributedOptimizer
Events
on_after_init
on_before_step
on_after_step
on_before_zero_grad
on_after_zero_grad