minBERT / train_scripts /base_sst_fm.py
GlowCheese's picture
Final touch
8eff58f
raw
history blame contribute delete
318 Bytes
from constants import *
from types import SimpleNamespace
from classifier import classifier_run
ARGUMENTS = SimpleNamespace(
dataset='sst',
batch_size=BATCH_SIZE_SST,
train=IDS_SST_TRAIN,
dev=IDS_SST_DEV,
test=IDS_SST_TEST,
lr=1e-5,
fine_tune_mode='full-model'
)
classifier_run(ARGUMENTS)