I have two models, one primary and one auxiliary. I want to fix the weights of the auxiliary network. However, I am unsure if I should put the auxiliary network in train mode or eval mode.
Signal_2_pred = PrimaryModel(Signal_1)
loss = MSE(Signal_2_pred, Signal_2_true)
loss += MSE(AuxiliaryModel(Signal_2_pred), Signal_2_aux)
I do not want to change the weights of the AuxiliaryModel, but I do want the gradients to be propagated. Does this require train mode or eval mode for the auxiliary?