base model : skt/kogpt2-base-v2
blog : [PyTorch, KoGPT2] Fine-tuning하고 문장 생성하기(w/ full code)
python train.py --batch_size 32 --epochs 10 --lr 2e-5 --warmup_steps 200
python generate.py --num_per_label 10
Example for kogpt2 fine-tuning & generation
base model : skt/kogpt2-base-v2
blog : [PyTorch, KoGPT2] Fine-tuning하고 문장 생성하기(w/ full code)
python train.py --batch_size 32 --epochs 10 --lr 2e-5 --warmup_steps 200
python generate.py --num_per_label 10
Example for kogpt2 fine-tuning & generation
MIT License