• Nicholas Lourie's avatar
    Update the doc string for T5WithLMHeadModel · 62f58046
    Nicholas Lourie authored
    T5WithLMHeadModel's doc string claims that indices of -1 are
    ignored while computing the cross-entropy loss in the forward
    pass; however, indices of -1 throw an error while indices of -100
    are ignored. This commit updates the doc string to be consistent
    with the class's behavior.
    62f58046
modeling_t5.py 42.5 KB