I would have intercut the two halves
And adjusted the script accordingly.
I love FMJ but it could have felt more organic and involving, and less linear, had the two location been intercut throughout.
And given a different ending to the training, probably.