the model 'ArcMargin' seems not save parameter?

[shoutOutYangJie]

i notice your code only saves net.parameters() but excludes arcmargin.state_dict. or i ignore it?

[Xiaoccer]

Because in the test phase we use the 'net' to extract the features and do not need the weights of 'arcmargin'.