fadel / pytorch_ema

Tiny PyTorch library for maintaining a moving average of a collection of parameters.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Whether the model with ema needs bo be fixed in the validation phase —— “model(with ema).eval()”?

TIanCat opened this issue · comments

Whether the model with ema needs bo be fixed in the validation phase —— “model(with ema).eval()”?

Hi,

I'm not sure I understood your question, but I have updated the example in README.md to show how the store/restore methods (contributed by @Zehui-Lin) can be used for validation without affecting training.