justheuristic
|
98531ce9ac
Support PyTorch 2.0.0 (#559)
|
2 éve |
Max Ryabinin
|
dac8940c32
Fix shape validation in GradientAverager (#481)
|
3 éve |
justheuristic
|
35851c8ce9
fix use_local_updates in optimizer (#468)
|
3 éve |
Artem Chumachenko
|
8387718c2a
Add PowerSGD for compressed gradient averaging (#432)
|
3 éve |
Max Ryabinin
|
b1507687ae
Use hivemind.Optimizer in examples/albert (#426)
|
3 éve |
justheuristic
|
5d31c3ba9d
Implement core functionality of hivemind.Optimizer (#403)
|
3 éve |
justheuristic
|
d883387e37
Add ProgressTracker (#408)
|
3 éve |
justheuristic
|
02eee9292d
[hivemind.Optimizer] TrainingStateAverager (#407)
|
3 éve |
justheuristic
|
8fa0a8e6ae
Add GradientAverager with support for delayed averaging (#404)
|
3 éve |