Closed LeeJeongHwi closed 1 week ago
설정해주어야 하는 부분은 다음과 같습니다.
base_config.json에서 QA model과 Retrieval model의 config 설정 값과 데이터셋 버전의 경로를 주어야합니다.
retrieval/sparse/config
Reader 모델을 학습시켜서 reader/extractive/models/outputs/에 모델을 저장합니다.
reader/extractive/models/outputs/
train_mrc.sh
bash reader/train_mrc.sh --type=ext --do_eval
inference.sh를 실행
inference.sh
bash inference.sh --qa ext --retrieval sparse --do_valid
--qa
--retrieval
--do_valid
-
Overview
12 에 따라 Extraction QA 모델과 Sparse 모델 결합한 코드입니다.
How to
설정해주어야 하는 부분은 다음과 같습니다.
1. config Path 설정
base_config.json에서 QA model과 Retrieval model의 config 설정 값과 데이터셋 버전의 경로를 주어야합니다.
retrieval/sparse/config
에 있는 config 이름만 넣어도 됩니다.2. Reader 학습
Reader 모델을 학습시켜서
reader/extractive/models/outputs/
에 모델을 저장합니다.train_mrc.sh
를bash reader/train_mrc.sh --type=ext --do_eval
명령어로 실행 (reader 폴더 밖에서 실행해야함)train_mrc.sh
파일에서 config path 설정Inference 실행
inference.sh
를 실행bash inference.sh --qa ext --retrieval sparse --do_valid
--qa
: QA model의 타입 설정--retrieval
: Retrieval 타입 설정--do_valid
: validation의 predictions을 추출할 것인지 선택inference.sh
에서 경로 설정 가능To Reviewer
-
Issue Tags