在使用以下代码处理图片数据时,遇到报错:
self and mat2 must have the same dtype, but got Float and Half
查阅很久都是让修改模型输入,但一直不行
visual_encoder = create_eva_vit_g(512).to(device)
output = visual_encoder(image)
经过高人指点后,亲测在模型初始化时后面加一个.half() 就能完美解决问题,如:
visual_encoder = create_eva_vit_g(512).to(device).half()
output = visual_encoder(image)