File size: 413 Bytes
a0b398e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from constants import *
from types import SimpleNamespace
from classifier import classifier_run
from .finetuned_bert import get_finetuned_bert
ARGUMENTS = SimpleNamespace(
dataset='sst',
batch_size=BATCH_SIZE_SST,
train=IDS_SST_TRAIN,
dev=IDS_SST_DEV,
test=IDS_SST_TEST,
lr=1e-3,
fine_tune_mode='last-linear-layer'
)
bert = get_finetuned_bert('unsup')
classifier_run(ARGUMENTS, bert) |