When following the steps to upload the weights (trained locally), I’m getting the following error:
KeyError: 'model'
Here is what I’m trying to do:
project.version(6).deploy(MODELTYPE, RUN_PATH, "average_model.pth")
The file I’m trying to upload is average_model.pth, but it doesn’t have a ‘model’ entry. Which file are we supposed to upload? I don’t have a ‘best.pt’ file anywhere…
Thanks.
Ok. Seems that we need to create a yaml file and this was the correct deploy command:
version.deploy('yolonas', f"{CHECKPOINT_DIR}/{EXPERIMENT_NAME}/{RUN_NAME}/", "ckpt_best.pth")
It still fails on the roboflow side but at least I don’t have the KeyError: ‘model’ error anymore…
system
November 18, 2024, 12:04am
4
This topic was automatically closed 21 days after the last reply. New replies are no longer allowed.