The whole code base is changed/cleaned/fixed/refactored. Lots of updates for standard training, differential privacy, and federated learning.

Complete overhaul of training process for federated. Much more cleaner than before. Same manner to be used for normal training in future.