Ye

IndexError and TypeError in Opacus when training a custom style model with multiple inputs [BUG]

2

**Environment** - Opacus Version: (1.5.4) - PyTorch Version: (2.4.1+cu121 ) - Hugging Face accelerate Version: (1.7.0) - Python Version: 3.9 - Hardware: 2 x NVIDIA GPUs (e.g., A100) - Training...