search

QinbinLi / MOON

Model-Contrastive Federated Learning (CVPR 2021)
MIT License
263 stars 56 forks source link

Processing of Datasets #19

Closed junzhang-zj closed 1 year ago

junzhang-zj commented 1 year ago

Hey, I had a question about the dataset processing, whether the final version of MOON's transformation in the dataloader only applied 'transforms.ToTensor()' and nothing like 'transforms.Normalize()'. I really need your help so that I can follow my work better.

QinbinLi commented 1 year ago

Hi @junzhang-zj ,

For the used transformation during training, you can refer to the get_dataloader function in https://github.com/QinbinLi/MOON/blob/main/utils.py. 'transforms.Normalize()' is included.

junzhang-zj commented 1 year ago

Thank you! I confused partition_data with get_dataloader.