kurianbenoy/distilbert-base-uncased-finetuned-sst-2-english-finetuned-imdb · Clarification on Finetuning Dataset for distilbert-base-uncased-finetuned-sst-2-english-finetuned-imdb

Hi there,

I'm currently exploring the distilbert-base-uncased-finetuned-sst-2-english-finetuned-imdb available on Hugging Face and have encountered a bit of confusion regarding the finetuning dataset used. In the model's description, it is mentioned that the finetuning was performed on the IMDB dataset. However, upon inspecting the config file, I noticed that the "finetuning_task" is set to SST-2.

Could someone please clarify which dataset was actually used for finetuning this model? Knowing the specific dataset is crucial for my project, and I want to ensure I'm using the model correctly based on its training background.

Thank you for your assistance and for the great work on this model!