conallwang / MeGA

The official implementation of "MeGA: Hybrid Mesh-Gaussian Head Avatar for High-Fidelity Rendering and Head Editing".
151 stars 21 forks source link

missing keys during checkpoint loading #4

Closed LCFAW closed 3 days ago

LCFAW commented 3 days ago

Hello, I found the following output during the test phase which indicates the missing keys, I followed the readme and ran the metrics.sh. Although I get a result seems reasonable, I wonder whether it is normal to get this output during testing.

Thank you!

[2024-11-22 14:20:48,654 hairwrapper.py] [MODEL_RESTORE] missing keys in canonical_gs checkpoint: set() [2024-11-22 14:20:48,654 hairwrapper.py] [MODEL_RESTORE] missing keys in canonical_gs model: set() [2024-11-22 14:20:48,658 hairwrapper.py] [MODEL_RESTORE] missing keys in deform_mlp checkpoint: set() [2024-11-22 14:20:48,658 hairwrapper.py] [MODEL_RESTORE] missing keys in deform_mlp model: set() [2024-11-22 14:20:48,659 facewrapper.py] [MODEL_RESTORE] missing keys in neural_texture checkpoint: set() [2024-11-22 14:20:48,659 facewrapper.py] [MODEL_RESTORE] missing keys in neural_texture model: set() [2024-11-22 14:20:48,670 facewrapper.py] [MODEL_RESTORE] missing keys in dynamic_texture checkpoint: set() [2024-11-22 14:20:48,670 facewrapper.py] [MODEL_RESTORE] missing keys in dynamic_texture model: set() [2024-11-22 14:20:48,676 facewrapper.py] [MODEL_RESTORE] missing keys in view_texture checkpoint: set() [2024-11-22 14:20:48,676 facewrapper.py] [MODEL_RESTORE] missing keys in view_texture model: set() [2024-11-22 14:20:48,682 facewrapper.py] [MODEL_RESTORE] missing keys in disp_decoder checkpoint: set() [2024-11-22 14:20:48,682 facewrapper.py] [MODEL_RESTORE] missing keys in disp_decoder model: set() [2024-11-22 14:20:48,684 facewrapper.py] [MODEL_RESTORE] missing keys in pe checkpoint: set() [2024-11-22 14:20:48,684 facewrapper.py] [MODEL_RESTORE] missing keys in pe model: set() [2024-11-22 14:20:48,685 facewrapper.py] [MODEL_RESTORE] missing keys in head_mlp checkpoint: set() [2024-11-22 14:20:48,685 facewrapper.py] [MODEL_RESTORE] missing keys in head_mlp model: set()

conallwang commented 3 days ago

Yes, it's normal (i.e., the missing sets are all empty).

If you really get some missing keys, the missing sets would have some values, such as "missing keys in canonical_gs checkpoint: set('_xyz', '_rotation')".

LCFAW commented 3 days ago

Thank you for your detailed reply! Best wishes!

Yes, it's normal (i.e., the missing sets are all empty).

If you really get some missing keys, the missing sets would have some values, such as "missing keys in canonical_gs checkpoint: set('_xyz', '_rotation')".