AttributeError: The following layers do not have gradients

[Camelliavin]

Thanks for providing the code. But when I'm running
python src/federated_main_s2.py --epochs=1 --num_users=1 --frac=1. --local_ep=100 --local_bs=50 --virtual_batch_size=50 --optimizer='sgd' --lr=0.001 --momentum=0.9 --dataset='dr' --dr_from_np=1 --withDP=1 --max_grad_norm=2. --noise_multiplier=1. --delta=1e-4
An error occurred:

Can someone tell me how to solve it? Thanks.

By the way, I would be very grateful if you could tell me in which article FL with Parallel DP (FPDP) was proposed.