Thank you for the interest you have shown.
Actually, this example is a simple code that I wrote when I was learning FL for the first time. The aggregation method is valid for only one architecture.
You can find the improved version and attack simulations here: https://github.com/eceisik/ARFED.