Skip to content

baseballChatbot7/KBO_MRC

Repository files navigation

Extration based Question Answering

PyTorch Lightning Config: Hydra Template

Description

  • koelectra 를 korquad v1 + custom kbo dataset 데이터셋에 fine-tuning하여 Question Answering
    • custom kbo dataset 구축은 여기를 참조해주세요.
    • question-answering 데이터셋 구축은 여기를 참조해주세요.
  • pretrained electra-small-v3 모델에 start token, end token을 예측하는 custom cnn layer를 추가해서 학습했습니다.
# default
python run.py

# train on CPU
python run.py trainer.gpus=0

# train on GPU
python run.py trainer.gpus=1

# train on sample configuration
python run.py experiment=example_simple.yaml

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages