🤔 model registry - inference with pytorch model

[Fedege98]

🤔 model registry - inference with pytorch model

[KevinMusgrave]

If you have the name of the model in the model registry and the model version number:

from determined.experimental import client

model_name = ...
version_num = ...

# checkpoint_dir is the path that contains the downloaded checkpoint
# By default checkpoint_dir will be checkpoints/<checkpoint_uuid> in the current working directory.
# You can download to a custom path by passing in path=<path> into the download function.
checkpoint_dir = client.get_model(model_name).get_version(version_num).checkpoint.download()