gpt-neo-fine-tuning-example
gpt-neo-fine-tuning-example copied to clipboard
Fine-Tune EleutherAI GPT-Neo And GPT-J-6B To Generate Netflix Movie Descriptions Using Hugginface And DeepSpeed
GPT-Neo-2.7B & GPT-J-6B Fine-Tuning Examples Using HuggingFace & DeepSpeed
Installation
cd venv/bin
./pip install -r ../../requirements.txt
./pip install deepspeed==0.5.9
GPT-Neo
Example with GPT-Neo-1.3B without DeepSpeed
Training and testing log with GPT-Neo-1.3B
Example with GPT-Neo-2.7B with DeepSpeed
DeepSpeed configuration with GPT-Neo-2.7B
Training and testing log with GPT-Neo-2.7B
GPU VRAM load during GPT-Neo-2.7B training
![](https://raw.githubusercontent.com/dredwardhyde/gpt-neo-fine-tuning-example/main/vram_gpt_neo_27.png)
RAM load during GPT-Neo-2.7B training
![](https://raw.githubusercontent.com/dredwardhyde/gpt-neo-fine-tuning-example/main/ram_gpt_neo_27.png)
Results
![](https://raw.githubusercontent.com/dredwardhyde/gpt-neo-fine-tuning-example/main/results.png)
GPT-J-6B
Example with GPT-J-6B with DeepSpeed
DeepSpeed configuration with GPT-J-6B
Training and testing log with GPT-J-6B
GPU VRAM load during GPT-J-6B training
![](https://raw.githubusercontent.com/dredwardhyde/gpt-neo-fine-tuning-example/main/vram_gpt_j.png)
RAM load during GPT-J-6B training
![](https://raw.githubusercontent.com/dredwardhyde/gpt-neo-fine-tuning-example/main/ram_gpt_j.png)