How to configure the number of GPU parameter of different nodes in fsdp mode

[wangleiofficial]

I have four nodes, How to configure the number of GPU parameter of different nodes in fsdp mode

g-1-0: 2*A100
g-1-1: 4*A40
g-1-2: 8*3090
g-1-3: 8*3090