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
Copy file name to clipboardExpand all lines: README.md
+4-3Lines changed: 4 additions & 3 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -5,9 +5,10 @@ Choose your EMA decay rate after training. No need to decide upfront.
5
5
The library uses `sigma_rel` (relative standard deviation) to parameterize EMA decay rates, which relates to the classical EMA decay rate `beta` as follows:
6
6
7
7
```python
8
-
beta =0.9999# Very slow decay -> sigma_rel ≈ 0.05
9
-
beta =0.999# Medium decay -> sigma_rel ≈ 0.15
10
-
beta =0.99# Fast decay -> sigma_rel ≈ 0.28
8
+
beta =0.9999# Very slow decay -> sigma_rel ≈ 0.01
9
+
beta =0.9990# Slow decay -> sigma_rel ≈ 0.03
10
+
beta =0.9900# Medium decay -> sigma_rel ≈ 0.10
11
+
beta =0.9000# Fast decay -> sigma_rel ≈ 0.27
11
12
```
12
13
13
14
This library was adapted from [ema-pytorch](https://github.com/lucidrains/ema-pytorch) by lucidrains.
0 commit comments