import torch
from modeling_mytext_sequence_classification import MyTextConfig, MyTextSequenceClassification

config = MyTextConfig()
model = MyTextSequenceClassification(config)

input_ids = torch.randint(0, config.vocab_size, (2, 10))
attention_mask = torch.ones_like(input_ids)
labels = torch.tensor([0, 1])

outputs = model(input_ids=input_ids, attention_mask=attention_mask, labels=labels)
print(outputs.loss)
print(outputs.logits)

hf-june/HW-202337458-mytext-sequence-classification

HW-202337458-mytext-sequence-classification

Assignment

Student

Included Classes

Requirements

Files

How to Use

hf-june/
HW-202337458-mytext-sequence-classification