Skip to content

nlpaueb/nlp-optimizers

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Requirements

python version 3.10
make environment

pip install -r requirements.txt

Tuning

In order to do the tuning run:

python tuning.<dataset>.py -o <optim> -m <model> (-lr or -all)

dataset is one of the following (mnli, mrpc, sst2, stsb, cola)
optim is one of the following (adam, adamw, nadam, adamax, adabound, sgd, sgdm)
choose -lr if you want to tune only learning rate or -all if you want to tune all hyperparameters

Train

In order to train run:

python train.<data>.py -o <optim> -m <model> -s <seed>

dataset is one of the following (mnli, mrpc, sst2, stsb, cola)
optim is one of the following (adam, adamw, nadam, adamax, adabound, sgd, sgdm)
seed is the seed for splitting the dataset to train/validation/test

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages