Now alpha rescales after rescale_after_iterations steps