text_plain_output_1.png
from transformers import DataCollatorWithPadding, BertForSequenceClassification, AutoModelForSequenceClassification, AutoTokenizer, Trainer, TrainingArguments

import torch

checkpoint = '../input/transformers/roberta-base'

device = 'cuda' if torch.cuda.is_available() else 'cpu'

model = AutoModelForSequenceClassification.from_pretrained(checkpoint, num_labels=2).to(device)