+1 to `keep_epochs`, it's much easier for me to remember the functionality with this name.