How to save the model ? Since the training code has not been released yet , So I want to train the model myself. I am wondering how should I save the model correctly. Looking forward to your reply!

[sulizhi]

No description provided.

[sulizhi]

I used the 'hiera_base_16x224.pth' model to train the other datasets, but the results of r2 are always 0.05x. I think it maight be wrong when I load and save the model. Here are my code of loading and saving the model. Can you point out where I was wrong? Thanks very much!

[dbolya]

Hi, why are you setting model.head.bias.data[0] = 55.6?

Also it probably doesn't matter in this case, but you should probably be setting head.projection to be that linear, not head itself.