at epoch 60 switch from mom to IAG/S
rmsprop
rollback update