File size: 463 Bytes
98c533d |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# T5.1.1 Efficient base nl36 model.
import seqio
include 't5x/examples/t5/t5_1_1/small.gin' # imports vocab, optimizer and model.
# ------------------- Model specification overrides --------------------------
VOCABULARY = @seqio.SentencePieceVocabulary()
seqio.SentencePieceVocabulary.sentencepiece_model_file = "spiece.model"
MODEL = @models.EncoderDecoderModel()
models.EncoderDecoderModel:
input_vocabulary = %VOCABULARY
output_vocabulary = %VOCABULARY |