from constants import *
from types import SimpleNamespace
from finetuning import finetune_bert

ARGUMENTS = SimpleNamespace(
    mode='sup',
    filepath='sup-cse-bert.pth',
    batch_size_train=12,
    batch_size_dev=64,
    temp=0.05, lr=1e-5,
)
finetune_bert(ARGUMENTS)