havakv / torchtuples

Training neural networks in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Use AdamW from pytorch

havakv opened this issue · comments

Should change torchtuples.optim.AdamW to instead use the version in pytorch