Skip to content

What happens when EMAHook and GradientCumulativeOptimizerHook is both used? #1509

Answered by Ezra-Yu
kwanyoungpark asked this question in Q&A
Discussion options

You must be logged in to vote

I am very sorry we have not done the relevant experiments, I think the biggest problem may not be on the EMA, generally speaking, EMA only has ~ 0.2 or so gain. And more frequent EMAs should not reduce the effectiveness of the model.

Maybe you can try --auto-scale and no GradientCumulativeOptimizerHook`. Our result was obtained by only using 16 GPUs, which is different from the 32 GPUs of the official paper.

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@kwanyoungpark
Comment options

Answer selected by kwanyoungpark
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants