CLIP学习 - 技术栈

CLIP学习

东皇太一在此2024-11-17 19:04

CLIP 学的是github官方给的教程

`Interacting_with_CLIP.ipynb`

注意：我运行程序遇到了很多问题

用低版本的numpy
pip install numpy==1.25.0

python 复制代码

with torch.no_grad():
    image_features = model.encode_image(image_input).float()
    text_features = model.encode_text(text_tokens).float()

是在这里崩溃的，我按照github的提示，显示更新了一下jupyter kernel，再把所有变量都放在cpu上，还修改了精度才运行成功的；

python 复制代码

# for CPU
image_input = image_input.float()
text_tokens = text_tokens.int()