I found there Is a difference about EMA between Sampler and Sampler_mol function in sampler.py
There isn't a row self.ema.copy_to(self.model.parameters()) in def Sampler_mol(object), but def Sampler(object) does.
Does this mean that there is no need for EMA while sampling molecules?
Thanks.
I found there Is a difference about EMA between Sampler and Sampler_mol function in sampler.py
There isn't a row
self.ema.copy_to(self.model.parameters())indef Sampler_mol(object), butdef Sampler(object)does.Does this mean that there is no need for EMA while sampling molecules?
Thanks.