Starting from the official repo, I\'m trying to generate intermediate outputs from a pre-trained model for a different downstream task. Since the network I\'m feeding it to is 2