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

Before Change

    
    output = encoder(input_ids=input_ids, token_type_ids=segment_ids, attention_mask=input_mask)
    if len(output) == 2:
        return EncoderOutput(pooled=output[1], unpooled=output[0],)
    elif len(output) > 2:
        // Extend later with attention, hidden_acts, etc

After Change

            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)

Instances