Downloading model? cant find model file?

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?

This topic was automatically closed 60 days after the last reply. New replies are no longer allowed.