Add autorestart
OverLordGoldDragon
released this
08 Jul 00:35
·
8 commits
to master
since this release
FEATURE: autorestart
option which automatically handles Warm Restarts by resetting t_cur=0
after total_iterations
iterations.
- Defaults to
True
ifuse_cosine_annealing=True
, elseFalse
- Must use
use_cosine_annealing=True
if usingautorestart=True
Updated README and example.py
.