import pathlib

train_roberta.py

@@ -2,6 +2,7 @@ from transformers import RobertaTokenizerFast, RobertaForSequenceClassification
 import torch
 from torch.utils.data import DataLoader
 from torch.utils.tensorboard import SummaryWriter
+from pathlib import Path
 
 
 from data import *