Any way to reduce number of TTA iterations

hi
TTA does iterations may be for all possible augmentations related to image orientation.
Is there a way to chose number ot TTA iterations that we want to reduce the inference time.

If test set size is big then it would take lot of time.

Regards
Jaideep…