train.py questions - Githubissues

ronghuaiyang / arcface-pytorch

1.74k stars 392 forks source link

train.py questions #44

Open lknight8631 opened 5 years ago

lknight8631 commented 5 years ago

Hi, there are two learnable modules, CNN encoder and FC layer, in "train.py". I have traced the code. It seems that only the encoder is set to "model.train()", but "metric_fc" is not set to "metric_fc.train()". Is it possible to update the weights of the "metric_fc" module without this setting during the training stage?

ZhangKangLong commented 4 years ago

I guess that the defalut setting of model is “Train”，and the test phase do not need this module