examples/research_projects/movement-pruning/emmental/modeling_bert_masked.py/BertIntermediate/forward
class BertIntermediate:    def forward(self, hidden_states, threshold):
        hidden_states = self.dense(hidden_states, threshold=threshold)
        hidden_states = self.intermediate_act_fn(hidden_states)
        return hidden_states