pytorch/test/cpp
Sotiris Lamprinidis 41f2dbde31 Add AdamW to C++ frontend (#40009)
Summary:
Slightly modified Adam, following the python implementation, and the `ProducesPyTorchValues` tests pass. I had a problem with another test though (see commit c1a6241676ab84fc531c1c3a10f964aa5704092e), it seems that optimizing for two steps with the same optimizer vs optimizing for two steps using freshly initialized objects will produce the same output.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40009

Differential Revision: D22096053

Pulled By: glaringlee

fbshipit-source-id: a31a8f5488cb37c53752ddf15436efabdba67dc4
2020-06-18 15:28:12 -07:00
..
api Add AdamW to C++ frontend (#40009) 2020-06-18 15:28:12 -07:00
common Trim libshm deps, move tempfile.h to c10 (#17019) 2019-02-13 19:38:35 -08:00
dist_autograd Fix/relax CMake linter rules (#35574) 2020-03-27 16:52:33 -07:00
jit [JIT] IRParser: properly parse negative numbers. (#39981) 2020-06-15 12:28:41 -07:00
rpc [TensorPipe] Improve serialization (#39010) 2020-05-28 10:48:24 -07:00
tensorexpr [TensorExpr] Eliminate Cond statements when each branch is a different kind of empty (#39754) 2020-06-11 17:08:14 -07:00
__init__.py Add train() / eval() / is_training() to C++ ScriptModule API (#16044) 2019-02-01 13:07:38 -08:00