Train a transformer-based architecture to classify the news items in the AG News dataset into one of four categories: World, Sports, Business, Science.
pip install datasets
pip install torch
pip install transformers
An example command to train is given below
python train.py my_model.th electra --B=8 --lr=0.00001 --epochs=2 --seed=1