I can give an alternative view answer based on personal (well, near-personal) experiance.
Someone close to me is a teacher, and one of the kids they teach is a child who went nearly completely deaf suddenly over christmas last year. Child X (as we should call them) is extremely averse to loud noises at present and detests going out to play at break and lunch time, because X is not used to filtering the noise out. X would prefer to stay inside in the quiet, as X is used to doing when X is at home, where they often go without their hearing aids. ITs got to the point where he'll be "naughty" just so he doesn't have to go outside.
So in your story, soldiers have augmented senses that allow them to see in the dark, enhance what they want to hear, and filter what they don't.
Suddenly all that is taken away. Noises that the augment buddy would remove, suddenly don't go away. That noisy hubbub in the shopping centre becomes deafening, simultaneously whilst they are nearly deaf to noises they want to hear. The sunlight as they come out of a dark room on a summers day blinds them momentarily.
Functionally, anything other than allowing them to wean off would create massive PTSD and mental health issues for the soldiers, particularly given the paranoia of no longer easily seeing "friend" vs "Foe". Assuming that such augments are not just military tech, it would likely result in specialist doctors who adjust the sensitivity of the tech to ease them back into daily life. As in many treatments, not all soldiers may react well to this. Some may continue to need low light vision, others the filtering senses. This would be an important point - not everybody reacts in the same way to trauma or recovers in the same way.