Mentions légales du service

Skip to content

Feature/torch training routine

SZPYRKA Jean-Luc requested to merge feature/torch_training_routine into develop

implementation of OPACUS DP.

  • opacus initialisation is made in training_data() of the provided TorchTrainingPlan class
  • postprocess() routine is provided also as a method of TorchTrainingPlan class
  • and called in after_training_data() method of fedbiomed.common.torchnn.TorchTrainingPlan

Merge request reports