how to load tta_wrapper model?

[JeongChanwoo]

I want to save the model with tta_wrapper and reuse it. However, when reloading the model, a repeat layer problem occurs. What is the solution?

I have loaded the reapeat layer from the tta_wrapper's layer, but the only response is that there is no n in init.

TypeError: init() missing 1 required positional argument: 'n'