minBERT / train_scripts /unsup_cfimdb_fm.py
GlowCheese's picture
Final touch
8eff58f
raw
history blame contribute delete
421 Bytes
from constants import *
from types import SimpleNamespace
from classifier import classifier_run
from .finetuned_bert import get_finetuned_bert
ARGUMENTS = SimpleNamespace(
dataset='cfimdb',
batch_size=BATCH_SIZE_CFIMDB,
train=IDS_CFIMDB_TRAIN,
dev=IDS_CFIMDB_DEV,
test=IDS_CFIMDB_TEST,
lr=1e-5,
fine_tune_mode='full-model'
)
bert = get_finetuned_bert('unsup')
classifier_run(ARGUMENTS, bert)