the model 'ArcMargin' seems not save parameter?

Xiaoccer / MobileFaceNet_Pytorch

MobileFaceNets: Efficient CNNs for Accurate Real-Time Face Verification on Mobile Devices

348 stars 89 forks source link

Open shoutOutYangJie opened 5 years ago

shoutOutYangJie commented 5 years ago

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

Xiaoccer commented 5 years ago

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