Ready to test implementation of momentum, weight decay in SdA models.

L.