How to use locally downloaded models
#2
by
AL58763
- opened
I have downloaded the weights locally on my pc. I want to load the model with these weights. How can i do that?
Assuming you have installed the aim package as: pip install git+https://[email protected]/apple/ml-aim.git.
You can load the model with the downloaded checkpoints as follows:
from aim.utils import load_pretrained
model = load_pretrained("aim-600M-2B-imgs", backend="torch", backbone_ckpt_path="<BACKBONE_PATH>", head_ckpt_path="<HEAD_PATH>")
Please check the repo for more details https://github.com/apple/ml-aim/tree/main.
Thanks @alaaelnouby . I wanted to know since the performance of the aim is scalable, are the larger models being trained currently?