LAVIS
LAVIS copied to clipboard
Finally, verify how to load the weights after blip2 training and how to configure them.
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.