Agreed, which is why you can probably get away with just 10 J-20A.There are different tiers to training — you have stages where you gain basic competency flying aircraft, stages where you practice basic maneuvers and strategies, and stages where you tie everything together and simulate “the real thing”. It is wasteful dedicating J-20A to anything less than highest fidelity training.
It is also important that the trainers you use before the J-20A are as similar as possible to the J-20A, such as the JL-XX with its twin canted tails and ventral stabilizers.
I’d also say that the J-20S is also great for training once new cadets join their brigades, they can combine regular operations with training in one sortie.
All in all, I’d say this can produce a lot of trained pilots quite efficiently.