code_structure.md

Project Directory Structure

• algorithm/
  • fedavg/
    • client.py
    • fedavg_trainer.py
    • optim.py
    • util.py
  • feddc/
    • client.py
    • feddc_trainer.py
    • util.py
  • feddyn/
    • client.py
    • comm_helpers.py
    • feddyn_trainer.py
    • optim.py
  • fedprox/
    • client.py
    • comm_helpers.py
    • fedprox_trainer.py
    • optim.py
  • local/
    • client.py
    • local_trainer.py
    • optim.py
    • util.py
• core/
  • base_dataset.py
  • base_model.py
  • base_network.py
  • logger.py
  • praser.py
  • util.py
• dataset/
• models/
  • fedst_ddpm_model/
    • base_model.py
    • ddpm_models/
      • config/
      • core/
        • base_dataset.py
        • base_model.py
        • base_network.py
        • logger.py
        • praser.py
        • util.py
      • ddpm_config.py
      • models/
        • guided_diffusion_modules/
          • nn.py
          • unet.py
        • loss.py
        • metric.py
        • model.py
        • network.py
        • sr3_modules/
          • unet.py
        • __init__.py
      • network.py
    • fedst_ddpm_model.py
    • networks.py
  • losses/
    • losses.py
    • __init__.py
  • unet_model/
    • base_model.py
    • networks.py
    • unet_model.py
  • __init__.py
• opt/
  • aaf_face_options.py
  • face_options.py
  • material_options.py
  • medical_material_options.py
  • medical_options.py
  • __init__.py
• pretrained_model/
  • DDPM_for_AAF_Face/
  • Face_Segment/
• scripts/
• test.py
• train.py
• util/
  • evaluation.py
  • image_pool.py
  • metrics.py
  • util.py