from constants import * | |
from types import SimpleNamespace | |
from classifier import classifier_run | |
ARGUMENTS = SimpleNamespace( | |
dataset='cfimdb', | |
batch_size=BATCH_SIZE_CFIMDB, | |
train=IDS_CFIMDB_TRAIN, | |
dev=IDS_CFIMDB_DEV, | |
test=IDS_CFIMDB_TEST, | |
lr=1e-3, | |
fine_tune_mode='last-linear-layer' | |
) | |
classifier_run(ARGUMENTS) |