kdhht2334 / AVCE_FER

[ECCV2022] The official repository of Emotion-aware Multi-view Contrastive Learning for Facial Emotion Recognition

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

missing keys when loading regressor

mdabbah opened this issue · comments

Hi,
First of all, Thank you so much for your work and sharing your code and pretrained weights!
I would appreciate if you could help me with a peroblem regarding loading the network

so i'm looking at the code an experimenting with it (more specifically the demo code)
there I see that the network is loaded in lines (320-322 demo-> run.py)

        encoder, regressor = nn_output()
        encoder.load_state_dict(torch.load('/home/daehakim/Desktop/FER_int_system/AVCE_demo/weights/Sparse_cont_enc.t7'), strict=False)
        regressor.load_state_dict(torch.load('/home/daehakim/Desktop/FER_int_system/AVCE_demo/weights/Sparse_cont_reg.t7'), strict=False)

but when loading the weights i get missing keys warning in the reggressor line

_IncompatibleKeys(missing_keys=['lin2.weight', 'lin2.bias', 'pw_conv.weight', 'pw_conv.bias'], unexpected_keys=[])

could it be that the weights file Sparse_cont_reg.t7 is corrupted / not up to date? could you please update it?

Thanks

Hi,

In my case, I cannot see any error when running run.py.

Please check whether you loaded a function Regressor_light_new or not in nn_output() of /my_util/fer_util.py.

Specifically, you can check model details in /my_util/fer_util.py.

If this still doesn't solve the problem, please provide a detailed error log.