# 使用示例
if __name__ == '__main__':
    # 导出 mobileclip_s0 模型
    export('mobileclip_blt', image_path="docs/fig_accuracy_latency.png")

模型已成功导出到: ./mobileclip_blt_export/
图像编码器输入尺寸: torch.Size([1, 3, 224, 224])
文本编码器输入尺寸: torch.Size([1, 77])
投影层输入尺寸: torch.Size([1, 1, 512])
正在开始验证ncnn模型...
图像编码器输出尺寸: torch.Size([1, 512])
图像编码器验证通过，MSE: 0.000000021671
文本编码器输出尺寸: torch.Size([1, 77, 512])
文本编码器验证通过，MSE: 0.000000000020
投影层输出尺寸: torch.Size([1, 1, 512])
投影层验证通过，MSE: 0.000000000000
ncnn模型验证完成。