Open rrjia opened 3 years ago
There is no train.py in https://github.com/dmlc/gluon-nlp/tree/master/scripts/question_answering or https://github.com/dmlc/gluon-nlp/tree/v0.x/scripts/question_answering
Please provide a link to the code you're concerned about. Thanks
code link is : https://github.com/dmlc/gluon-nlp/blob/qanet/scripts/question_answering/train.py
Description
class ExponentialMovingAverage in train.py the ema use two position:
if global_step == 1: for name, param in model.collect_params().items(): ema.add(name, param.data(CTX[0]))
for name, param in model.collect_params().items(): ema(name, param.data(CTX[0]))
but model‘s params not update ,final we save model params as:
save model after train one epoch
is the ema not work?