Hi There
I have finished training my model and have been adding my model as an artifact to WandB.
I now want to use the model for inference, but when i go to models on WandB, it shows all the versions of my model but there are no saved files?
How do I download the latest model file to use?
Hello @markstent !
You can download the model using the following code:
import wandb
run = wandb.init()
artifact = run.use_artifact('<entity>/<project>/<model>:<version>', type='model')
artifact_dir = artifact.download()
However, like you mentioned, there is no files in the files tab of the model. Could you share a short code snippet of how you are logging these models to wandb?