2409ca833031ab770e68a334cf2686d1d8d930d1,jiant/proj/main/modeling/taskmodels.py,,get_output_from_encoder,#Any#Any#Any#Any#,288

Before Change

        return EncoderOutput(pooled=output[1], unpooled=output[0],)
    elif len(output) > 2:
        // Extend later with attention, hidden_acts, etc
        return EncoderOutput(pooled=output[1], unpooled=output[0], other=output[2:])
    else:
        raise RuntimeError()

After Change

    
    model_arch = ModelArchitectures.from_encoder(encoder)
    if model_arch in [
        ModelArchitectures.BERT,
        ModelArchitectures.ROBERTA,
        ModelArchitectures.ALBERT,
        ModelArchitectures.XLM_ROBERTA,
    ]:
        pooled, unpooled, other = get_output_from_standard_transformer_models(
            encoder=encoder, input_ids=input_ids, segment_ids=segment_ids, input_mask=input_mask,
        )
    elif model_arch == ModelArchitectures.ELECTRA:
        pooled, unpooled, other = get_output_from_electra(
            encoder=encoder, input_ids=input_ids, segment_ids=segment_ids, input_mask=input_mask,
        )
    elif model_arch in [
        ModelArchitectures.BART,
        ModelArchitectures.MBART,
    ]:
        pooled, unpooled, other = get_output_from_bart_models(
            encoder=encoder, input_ids=input_ids, input_mask=input_mask,
        )
    else:
        raise KeyError(model_arch)

    // Extend later with attention, hidden_acts, etc
    if other:
        return EncoderOutput(pooled=pooled, unpooled=unpooled, other=other)
    else:
        return EncoderOutput(pooled=pooled, unpooled=unpooled)

Instances