I think you could call .train() on all submodules via e.g. model.apply() or alternatively try to use:
model.eval()
model.training = True

This would also set all submodules to the eval mode and would reset the parent model to the training mode.