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) |