examples/research_projects/deebert/src/modeling_highway_bert.py/DeeBertEncoder/init_highway_pooler
class DeeBertEncoder:    def init_highway_pooler(self, pooler):
        loaded_model = pooler.state_dict()
        for highway in self.highway:
            for name, param in highway.pooler.state_dict().items():
                param.copy_(loaded_model[name])