I would make it constant and give the players the ability to choose if they want to replay content for better upgrades. This leaves the decision to the player. Recently finished Metroid dread and loved that I had to option to go back and upgrade.