KoBertSum
KoBertSum copied to clipboard
validation 실행 시 error
validate 모드일 떄,
train_extractive.py의 validate_ext()의 validate(), test_ext()가 순서대로 실행되는데
args.bert_data_path = 'PROJECT_PATH/ext/data/bert_data/valid_abs' 인 상태여서
validate() 함수에 쓰이는 데이터만 load되고,
test_ext() 함수에 쓰이는 데이터인 'PROJECT_PATH/ext/data/bert_data/test' 폴더의 데이터들은 load 되지 않습니다.
아마 .pt파일을 train, valid, test로 나눠서 저장하셔서 이런 문제가 생기지 않았나 합니다.
아 명하님 감사합니다. 이제야 확인했네요; 한 번 살펴보고 말씀드리겠습니다.