Where is the fine-tuned model output?
I’m new to using AutoTrain on Hugging Face and I encountered an issue during my first attempt at fine-tuning a model. I have a free account, because I want to see whether I can get something to work before I start paying for training. Here’s a summary of what I did and the problems I’m facing:Training Configuration:I trained using Mistral-7B-Instruct-v0.2Dataset: I uploaded a tiny JSONL file (24 records) with a single “text” field for training.Training Parameters: I set the training to run for one epoch.Training Process:The training ran for a couple of seconds.I received a message that the space was paused, which I assumed meant the training had completed.Issue:After the training supposedly completed, I can’t find any output files or trained models.I checked all available tabs and sections in the AutoTrain interface but didn’t see anything labeled “Models,” “Artifacts,” “Results,” or similar.I reviewed the logs but didn’t find any clear indications of where the output is stored.I checked my Hugging Face profile under the “Models” heading, but it says “None yet.”Questions:Where should I look in the AutoTrain interface to find the trained model and output files?Are there any additional steps I need to take to ensure the trained model is saved and accessible?With a free account, I don’t have any GPUs assigned. But is that a problem with only 24 short training samples and one epoch?Any guidance or tips would be greatly appreciated!
2024-06-09T19:27:39Z
I assume AutoTrain pushes your model to a repository on the hub, under your profile.cc@abhishek
2024-06-09T22:19:05Z
Thank you for your reply. I am not seeing any model that I have fine-tuned, and I have tried training several times. Perhaps I need to pay for GPUs to do fine-tuning, even with a tiny training file.
2024-09-13T03:28:21Z
Did you every figure out the issue? I’m having the same problem and I paid for the smallest Nvidia GPU to see if that was the problem but no luck yet.
2024-09-13T06:37:03Z
cc@abhishek
2024-09-13T06:48:54Z
if you used Hugging face spaces for training models, the model is saved in your hf account with the name that you specified as project name upon successful training.if you train locally model is in output directory which is again project name specified by you and will be pushed to hf hub if you chose to do so.
2024-09-19T04:02:30Z
Thank you alot!!