skynbe / FedRCL

Official Implementation of FedRCL (CVPR 2024)
14 stars 2 forks source link

About the core code for the RCL loss. #2

Open polaris-973 opened 1 month ago

polaris-973 commented 1 month ago

First, I would like to thank the authors for their open source work. And can you provide the core code for the RCL loss? Or indicate exactly where it is in the repository, as I didn't find where the core code is after skimming.

skynbe commented 1 month ago

Thanks for your interest. We just refactored some things for easier understanding.

  1. RCLClient manages client-side operations for FedRCL traning.
  2. CLLoss is the implementation of contrastive learning loss. We implement RCL loss by a summation of SCL loss and divergence penalty loss as in here, each of which is a variant of CL loss.
  3. fedrcl.yaml shows the detailed configurations of RCL Loss. You can directly change the values here, or can create the new yaml files.

We will update the README file soon as well.