If model_A shouldn’t be trained, this would be the right approach.
However, based on the initial description of the issue I understood that model_A should be further trained in eval() mode.
CC @superco to clarify the use case.