voidful / lighting-hucode

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

lighting-hucode

Usage

modify config in train.py

model_config = "facebook/bart-base"
batch_size = 50
train_list = lista # list, sentences in line
eval_list = listb # list, sentences in line

# and
pl.Trainer() # training config

run

python train.py

About


Languages

Language:Python 100.0%