File size: 272 Bytes
a0b398e
 
 
 
 
 
8eff58f
a0b398e
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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)