3c874575bf40e8b1fa2280371131a8f29ebb3e98,src/gluonnlp/models/roberta.py,RobertaModel,hybrid_forward,#RobertaModel#Any#Any#Any#,227

Before Change

            pooled_out = self.apply_pooling(contextual_embeddings)
            outputs.append(pooled_out)

        return tuple(outputs) if len(outputs) > 1 else outputs[0]

    def get_initial_embedding(self, F, inputs):
        Get the initial token embeddings that considers the token type and positional embeddings

After Change

    def hybrid_forward(self, F, tokens, valid_length):
        embedding = self.get_initial_embedding(F, tokens)
        if self._layout != self._compute_layout:
            contextual_embeddings, additional_outputs = self.encoder(F.np.swapaxes(embedding, 0, 1),
                                                                     valid_length)
            contextual_embeddings = F.np.swapaxes(contextual_embeddings, 0, 1)
        else:
            contextual_embeddings, additional_outputs = self.encoder(embedding, valid_length)
        if self.use_pooler:

Instances