You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The current manual optimization is not flexible enough.
From discussion with @justusschock and @awaelchli, here is the proposal:
manual_backward shouldn't take any optimizers as they aren't used internally expect in Horovod. To solve this, in Horovod, call synchronise on all optimizers.
In LightningOptimizer, optimizer shoudn't follow accumulated_grad_batches.
Pitch
Alternatives
Additional context
The text was updated successfully, but these errors were encountered:
🚀 Feature
Motivation
The current
manual optimization
is not flexible enough.From discussion with @justusschock and @awaelchli, here is the proposal:
Pitch
Alternatives
Additional context
The text was updated successfully, but these errors were encountered: