Closed whuhit closed 5 years ago
in train.py line 134 and 135 self.net.module.ema.mu *= momentum self.net.module.ema.mu += mu * (1 - momentum)
self.net.module.ema.mu *= momentum self.net.module.ema.mu += mu * (1 - momentum)
it maybe like this self.net.module.emau.mu *= momentum self.net.module.emau.mu += mu * (1 - momentum)
self.net.module.emau.mu *= momentum self.net.module.emau.mu += mu * (1 - momentum)
Yes... Thank you very much!
XiaLiPKU
commented
5 years ago XiaLiPKU
commented
5 years ago
in train.py line 134 and 135
self.net.module.ema.mu *= momentum self.net.module.ema.mu += mu * (1 - momentum)it maybe like this
self.net.module.emau.mu *= momentum self.net.module.emau.mu += mu * (1 - momentum)