Torch training plan has been modified to enable/disable DP by simply providing a
'DP_args' dictionary in
training_args containing instructions on the type of DP training to be performed (local or central) and privacy parameters (sigma and clipping). The notebook
monai-2d-image-registration-LDP-CDP.ipynb shows an example of not-DP vs DP training on image registration task with monai.
Few points to discuss:
opacus should be integrated in the training plan.
opacus with MNIST should be modified according to the new training plan.