kobart_summarization_model.bin
- kobart 문서 요약을 사용하기 위해 train한 ckpt파일을 huggingface binary로 추출한 파일입니다.
#올려진 학습코드에서 batch_size를 2로 줄여서 실행했습니다.
[use gpu]
python train.py --gradient_clip_val 1.0 --max_epochs 50 --default_root_dir logs --gpus 1 --batch_size 2 --num_workers 4
-
결과로 나온 epoch=49-val_loss=2.782.ckpt 파일을 추출했습니다.
-
학습 모델 파일이 필요하신 분은 아래 링크로 사용하시기 바랍니다.