New scheduler: Epochwise OneCycleLR

Signed-off-by: Emanuele Ballarin <emanuele@ballarin.cc>
emaballarin · Aug 17, 2023 · 67b4677 · 67b4677
1 parent 839f400
commit 67b4677
Show file tree

Hide file tree

Showing 5 changed files with 324 additions and 289 deletions.
diff --git a/ebtorch/__init__.py b/ebtorch/__init__.py
@@ -92,6 +92,7 @@
 del ralah_optim
 del wfneal
 del tricyc1c
+del epochwise_onecycle
 
 # Deletions (from .logging)
 del AverageMeter