from constants import * | |
from types import SimpleNamespace | |
from finetuning import finetune_bert | |
ARGUMENTS = SimpleNamespace( | |
mode='unsup', | |
filepath='unsup-cse-bert.pth', | |
batch_size_train=8, | |
batch_size_dev=64, | |
temp=0.05, lr=1e-5, | |
) | |
finetune_bert(ARGUMENTS) |