The code for DP finetuning using Pytorch DDP for multiple GPUs can be found in fl_dp_ddp.py.
The example dataset and partition can be found in yelp_train.csv and yelp_iid_partition.csv.
For refinement, first run generate_embedding.py to get embeddings, then run refinement. 
