Question on code in language_fedma.py

[anushiya1]

Hi,

Thank you for your wonderful work and making the code public.

I have a question regarding the code in language_fedma.py. In line 302-303, why is there a 'pass' if layer_index ==2 for example (please refer to attached screenshot)? What happens to the case where we have more layers?

Can I use the code from lines 309-313 again for layer_index ==2 instead of a pass?

Thank you very much!