add momentum and eta schedule