What layers would you like to add RicardoGuadalupeGomezMartnez ?
It depends where you will be adding the layers. We cut off some post-processing in the heads output, so that we can decode this correctly on device. Even if you add base layers, this should be possible, but the "semantic meaning" of the outputs in modified model should still be the same.