Ejemplos de HFBackboneConfig en Python

Ejemplo n.º 1

def test_model_has_correct_cfg():
    model = TranslationTransformer(
        HFBackboneConfig(
            pretrained_model_name_or_path="patrickvonplaten/t5-tiny-random"))
    assert model.hparams.downstream_model_type == "transformers.AutoModelForSeq2SeqLM"
    assert type(model.cfg) is TranslationConfig

Ejemplo n.º 2

def test_model_has_correct_cfg():
    model = MaskedLanguageModelingTransformer(
        HFBackboneConfig(pretrained_model_name_or_path='prajjwal1/bert-tiny'))
    assert model.hparams.downstream_model_type == 'transformers.AutoModelForMaskedLM'

Ejemplo n.º 3

Archivo: test_text_classification.py Proyecto: taeminlee/lightning-transformers

def test_model_has_correct_cfg():
    model = TextClassificationTransformer(
        HFBackboneConfig(pretrained_model_name_or_path='bert-base-cased'))
    assert model.hparams.downstream_model_type == 'transformers.AutoModelForSequenceClassification'

Ejemplo n.º 4

Archivo: test_question_answering.py Proyecto: taeminlee/lightning-transformers

def test_model_has_correct_cfg():
    model = QuestionAnsweringTransformer(
        HFBackboneConfig(pretrained_model_name_or_path='bert-base-cased'))
    assert model.hparams.downstream_model_type == 'transformers.AutoModelForQuestionAnswering'

Ejemplo n.º 5

Archivo: test_summarization.py Proyecto: taeminlee/lightning-transformers

def test_model_has_correct_cfg():
    model = SummarizationTransformer(
        HFBackboneConfig(pretrained_model_name_or_path='t5-base'))
    assert model.hparams.downstream_model_type == 'transformers.AutoModelForSeq2SeqLM'
    assert type(model.cfg) is SummarizationConfig

Ejemplo n.º 6

def test_model_has_correct_cfg():
    model = LanguageModelingTransformer(
        HFBackboneConfig(pretrained_model_name_or_path="sshleifer/tiny-gpt2"))
    assert model.hparams.downstream_model_type == "transformers.AutoModelForCausalLM"