value = pt_dict[tf2pt[str(var)]].numpy() KeyError: 'reduction_0.2.weight'

[ruyijidan]

value = pt_dict[tf2pt[str(var)]].numpy()

KeyError: 'reduction_0.2.weight'

[jhxing]

I got the same issue as you. I filtered the illegal key of the pt_dict by modifying the line 87 of train.py
if str(var) in tf_names:
to
if str(var) in tf_names and tf2pt[str(var)] in pt_dict.keys():

Still, got some other issues.