lucidrains/ema-pytorch

Implementation error

sbl1996 opened this issue · 1 comments

sbl1996 commented

ema-pytorch/ema_pytorch/ema_pytorch.py

Line 112 in d65f4e3

for name, buffer in model.named_parameters():

I think it should be for name, buffer in model.named_buffers():.

A related hint for users: do not forget to call ema.eval() in your evaluation loop.

lucidrains commented

@sbl1996 ah yes, thank you 😞