text_plain_output_1.png
import warnings

import os
import time
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.utils.data import Dataset
import pandas as pd
from sklearn import metrics
from sklearn.model_selection import train_test_split
import transformers
from transformers import AdamW, T5Tokenizer, T5ForConditionalGeneration
import torch_xla.core.xla_model as xm
import torch_xla.distributed.parallel_loader as pl
import torch_xla.distributed.xla_multiprocessing as xmp
import warnings
warnings.filterwarnings('ignore')