Как подтолкнуть модель НЛП, обученную с помощью pytorch, к обнимающемуся лицу?
#python #google-colaboratory #huggingface-transformers #pytorch-lightning #pytorch-dataloader Вопрос: Я обучил тонкую настройку модели T5 с помощью моего набора данных. Вот мой класс моделей class QA_model(pl.LightningModule): def __init__(self): super().__init__() self.model=T5ForConditionalGeneration.from_pretrained(model_t5,return_dict=True) def forward(self,input_ids, attention_mask,labels=None):…