Don't forget a thing:
It must be only a single Drum Kit in all style parts. You cannot have multiple drums kits in different parts of style.Example: You cannot have Standard Kit on Main A and Acoustic Kit on Main B. The settings from Drum Setup will be reset on changing between Main A and Main B
PS: But as a workaround, if you want more than one kit, you can select, from beginning, a single Kit for all style parts, after that, change to style part you want a different kit, select for record the RHY1 or RHY2 channel (where you want another drum kit), press drum edit, and change the Kit from "Kit" button
Like this, the style will retain the settings and you will have different drums kits on different style parts
This approach will set a single key settings, so you have to do this setting for each key (/drum part - B2, C3, C#3, or what element you want from another drum kit).
I recommend after you make some changes to a drum kit, to do it also in YEM, and save it as a Custom Drum. Like this, you can use your new settings selecting your new Custom Drum Kit, and not making every time the settings...