search

birkhoffkiki / GPFM

The official implementation of GPFM
39 stars 1 forks source link

Update README.md to correct an error to generate the image embeddings #6

Closed HelloWorldLTY closed 1 month ago

HelloWorldLTY commented 1 month ago

Hi, thanks for your work. I found that there is an issue when loading the image with current example code:

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor

I covert the image to cuda version and it resolved this problem. If you think it is helpful, please consider merging it. Thanks.

birkhoffkiki commented 1 month ago

Sure, thanks!