simplified momentum updates for pretraining and finetuning SdA models, and moved the learning rate to be a shared variable to make decay simpler.