How to integrate Opacus with Nvidia NeMo
yangjundy opened this issue · 1 comments
yangjundy commented
NeMo is a framework that is developed based on PyTorch and PyTorch Lightning. It has been noticed that Opacus is compatible with PyTorch Lightning, which opens up the possibility of leveraging Opacus within the NeMo framework.
It would be incredibly advantageous if an integration between Opacus and NeMo could be achieved without making significant modifications to the existing NeMo modules, such as ASR.
Any suggestions or guidance on how to accomplish this seamless integration would be highly appreciated.
yangjundy commented
I found that it works without much efforts.