Comments (8)
Note that if the implementation is changed so that the weight decay is not multiplied by the learning rate you should subtract the weight decay from the weights instead of adding them (otherwise the weights will become larger, not smaller). Thus the update rule could be:
p.data.add_(-torch.mul(p.data, group['weight_decay']).addcdiv(-step_size, exp_avg, denom) )
or equivalently
p.data.add_(-group["weight_decay"], p.data).addcdiv_(-step_size, exp_avg, denom)
Another thing to note is that at least for the problem I tried I had to modify the weight decay significantly from the value used when using Adam optimizer. It is probably a good idea to do a sweep on just the weight decay value when switching from Adam to AdamW.
from adamw-pytorch.
Thank you for your advice. I would check it.
from adamw-pytorch.
@ousou Great comment! Sorry guys, I was too busy to dig more here though it will be really interesting. I should make it a rule like 10% time to do things that have no deadlines.
from adamw-pytorch.
Another thing is, I think the weight decay should not be divided by \sqrt(v_t)?
from adamw-pytorch.
Another thing is, I think the weight decay should not be divided by \sqrt(v_t)?
@WilliamLwj Yes, that's correct. Neither of the solutions above should do that, since the addcdiv-part is only using denom and exp_avg, not the weight decay. I noticed though that in my previous comment the suggested implementations actually seem to return different values, and I'm not really sure why. I think the second one is correct, but I can't be sure.
from adamw-pytorch.
I noticed though that in my previous comment the suggested implementations actually seem to return different values, and I'm not really sure why. I think the second one is correct, but I can't be sure.
It seems that the issue in the first version is with the -torch.mul part getting the minus wrong and thus getting the sign wrong in what's added to the weights. The correct version would be
p.data.add_((-torch.mul(p.data, group['weight_decay'])).addcdiv(-step_size, exp_avg, denom))
but it's easier to just use the second version.
from adamw-pytorch.
@ousou Should the second version of the code also multiply the learning rate, ie.
p.data.add_(-group['weight_decay'] * group['lr'], p.data)
p.data.addcdiv_(-step_size, exp_avg, denom)
from adamw-pytorch.
@ousou Should the second version of the code also multiply the learning rate, ie.
p.data.add_(-group['weight_decay'] * group['lr'], p.data)
p.data.addcdiv_(-step_size, exp_avg, denom)
@arunava555 No, the weight decay should not be multiplied by the learning rate - the AdamW paper specifies the update rule this way. In the current version of the implementation in this repository the weight decay is implicitly multiplied by the learning rate, and in this issue there is mentioned how to modify the update rule to not do this. In the AdamW paper it is mentioned that the weight decay should not be multiplied by learning date to decouple the two from each other. Decoupling them means that you can find the optimal learning rate and weight decay independently, i.e. changing the learning rate does not change the optimal weight decay (and vice versa).
In the original Adam the weight decay was also implicitly bound to the learning rate, so you had to find a new optimal weight decay when changing the learning rate. AdamW fixes this issue.
from adamw-pytorch.
Related Issues (1)
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from adamw-pytorch.