3.1.6. aijack.collaborative.fedkd package

3.1.6.1. Submodules

3.1.6.2. aijack.collaborative.fedkd.client module

class aijack.collaborative.fedkd.client.FedKDClient(student_model, teacher_model, task_lossfn, student_lr=0.1, teacher_lr=0.1, adaptive_distillation_losses=True, adaptive_hidden_losses=True, gradient_compression_ratio=1.0, user_id=0, send_gradient=True)

Bases: aijack.collaborative.fedavg.client.FedAVGClient

Implementation of FedKD (https://arxiv.org/abs/2108.13323)

loss(x, y)

3.1.6.3. Module contents

class aijack.collaborative.fedkd.FedKDClient(student_model, teacher_model, task_lossfn, student_lr=0.1, teacher_lr=0.1, adaptive_distillation_losses=True, adaptive_hidden_losses=True, gradient_compression_ratio=1.0, user_id=0, send_gradient=True)

Bases: aijack.collaborative.fedavg.client.FedAVGClient

Implementation of FedKD (https://arxiv.org/abs/2108.13323)

loss(x, y)