Key error "loss" issue while training custom dataset

minimaxir / aitextgen

A robust Python tool for text-based AI training and generation using GPT-2.

https://docs.aitextgen.io

MIT License

1.84k stars 220 forks source link

Open vjarora1978 opened 1 year ago

vjarora1978 commented 1 year ago

I am getting this issue while training it