|
return x@torch.transpose(projection_matrix, 0, 1) |
|
feature = project( feature, torch.load('projection_matrix').cuda().T ).squeeze(0) |
The projection_matrix is clip_model.text_projection.weight.data, right? Is it enough to transpose projection matrix once?
