To conduct experiments, please execute the following:
python main.py <dataset name> <model name>
usage: main.py [-h] [-e EXPERIMENT] [-m MODEL_NAME] [-d DATA_DIR] [-me MAX_EPOCH] [-bs BATCH_SIZE] [--lr LR] [--device DEVICE] [-nw NUM_WORKERS] [-us] [-o OPTIMIZER] [-wd WEIGHT_DECAY]
Reference : https://github.com/iiduka-researches/202009-coba/