Mimi (Patcharin) Cheng

icon indicating an email [email protected]

icon location Toronto, Canada icon location 👩🏻‍💻 MLE | Data Scientist | MSc in Computer Science and Machine Learning | Mila - Quebec AI Institute | Université de Montréal

Results 2 comments of Mimi (Patcharin) Cheng

Throw error when retrieving an embedding for inferencing one image

Thanks for your prompt response @lucidrains! It's resnet-50 Here is the full script ``` resnet = models.resnet50(pretrained=True) model = BYOL( resnet, image_size = 256, hidden_layer = 'avgpool' ) imgs =...

Throw error when retrieving an embedding for inferencing one image

@lucidrains thanks for you explanation. It works with `imgs = torch.randn(2, 3, 256, 256)`. Just wondering is there a way to retrieve embedding for only one new coming image after...