Fed-LAMB: Layer-wise and Dimension-wise Locally Adaptive Federated Learning.