examples/research_projects/deebert/src/modeling_highway_bert.py/DeeBertEncoder/set_early_exit_entropy
class DeeBertEncoder:    def set_early_exit_entropy(self, x):
        if (type(x) is float) or (type(x) is int):
            for i in range(len(self.early_exit_entropy)):
                self.early_exit_entropy[i] = x
        else:
            self.early_exit_entropy = x