Unable to load pre-trained alexnet weights

[mika-long]

I have installed packages torch, luz, and torchvision.
I had the following code:

library(torch)
library(luz)
library(torchvision)

weights <- model_alexnet(pretrained=TRUE)

The above produces an error as follows:
Error in !self$..refer_to_state_dict.. : invalid argument type

Not sure what is wrong here.

[dfalbel]

You probably need to reinstall torch vision, using something like install.packages("torchvision, type="source").
Probably related to mlverse/torchvision#100 (comment)

[mika-long]

This worked! Thank you 🙏