LAVIS icon indicating copy to clipboard operation
LAVIS copied to clipboard

Finally, verify how to load the weights after blip2 training and how to configure them.

Open nuocheng opened this issue 1 year ago • 0 comments

Hello, first of all thank you for your work. I am using blip2 and have completed the first and second steps, both of which generated pth weight files. I would like to test the training results. How should I load these pth weight files? Help me

The final tested file

https://github.com/salesforce/LAVIS/blob/ac8fc98c93c02e2dfb727e24a361c4c309c8dbbc/projects/blip2/README.md?plain=1#L46

I found that the loading model is here, but I couldn't find the path to configure the 'pth' weights trained in the first two steps.

nuocheng avatar Jan 19 '24 03:01 nuocheng