https://github.com/hkchengrex/nitrous-ema

Fast and simple post-hoc EMA (Karras et al., 2023) for PyTorch with minimal `.item()` calls. ~78% lower overhead than ema_pytorch.
https://github.com/hkchengrex/nitrous-ema

ema machine-learning pytorch

Last synced: 2 days ago
JSON representation

Fast and simple post-hoc EMA (Karras et al., 2023) for PyTorch with minimal `.item()` calls. ~78% lower overhead than ema_pytorch.

• Host: GitHub
• URL: https://github.com/hkchengrex/nitrous-ema
• Owner: hkchengrex
• License: mit
• Created: 2024-09-05T21:12:45.000Z (28 days ago)
• Default Branch: main
• Last Pushed: 2024-09-06T20:20:40.000Z (27 days ago)
• Last Synced: 2024-09-07T01:40:04.175Z (26 days ago)
• Topics: ema, machine-learning, pytorch
• Language: Python
• Homepage:
• Size: 11.7 KB
• Stars: 1
• Watchers: 2
• Forks: 0
• Open Issues: 0

Awesome Lists containing this project