Hi, I’ve downloaded a dataset of about 13 million reviews from BoardGameGeek and tried collab learning on it. Having lots of fun with it and plan to make a blog out of it.
However when I want to load a saved model in a new session, I get an error.
size mismatch for u_weight.weight: copying a param with shape torch.Size([277772, 50]) from checkpoint, the shape in current model is torch.Size([277751, 50]).
size mismatch for u_bias.weight: copying a param with shape torch.Size([277772, 1]) from checkpoint, the shape in current model is torch.Size([277751, 1]).
The error is pretty, clear, but I don’t know why it is happening, since the dataset is the same. The only steps I take are:
- create Databunch
- create Learner
- load model
See my code below:
data = CollabDataBunch.from_df(reviews, user_name=‘user’,item_name=‘name’,rating_name=‘rating’,bs=100000)
learner = collab_learner(data, n_factors=50, y_range=(2.,10))
Any help would be greatly appreciated!