when i load model, not found key variable

[JadeKim042386]

ML_Decoder/src_files/models/utils/factory.py

Lines 47 to 56 in 8a9e984

	if not load_head:
	if 'model' in state:
	key = 'model'
	else:
	key = 'state_dict'
	filtered_dict = {k: v for k, v in state[key].items() if
	(k in model.state_dict() and 'head.fc' not in k)}
	model.load_state_dict(filtered_dict, strict=False)
	else:
	model.load_state_dict(state[key], strict=True)

when load_head is True, it didn't defined key.

+ i know solution. so, i just notified this problem. plz fix it

[Esmaeel-Hilal]

Modify the code like this

if 'model' in state: 
    key = 'model' 
else: 
    key = 'state_dict'
if not load_head: 
     filtered_dict = {k: v for k, v in state[key].items() if 
                      (k in model.state_dict() and 'head.fc' not in k)} 
     model.load_state_dict(filtered_dict, strict=False) 
else: 
     model.load_state_dict(state[key], strict=True) 

I found the solution in another thread in the issues answered by https://github.com/yangyangtiaoguo