File size: 275 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='unsup',
    filepath='unsup-cse-bert.pth',
    batch_size_train=8,
    batch_size_dev=64,
    temp=0.05, lr=1e-5,
)
finetune_bert(ARGUMENTS)