readme

lucidrains · Oct 16, 2024 · 0b59eaf · 0b59eaf
1 parent 9fb2977
commit 0b59eaf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -105,6 +105,10 @@ ema = EMA(
     update_model_with_ema_every = 10000 # say 10k steps is 1 epoch
 )
 
+# or you can do it manually at the end of each epoch
+
+ema.update_model_with_ema()
+
 ```
 
 ## Citations