"git@developer.sourcefind.cn:OpenDAS/ktransformers.git" did not exist on "bda9cf15e71514865280148bdb156f7f8d12ebe2"
Unverified Commit 16fd2fde authored by Tim Dettmers's avatar Tim Dettmers Committed by GitHub
Browse files

Merge pull request #601 from eltociear/patch-1

Fix typo in test_optim.py
parents d83c5fe4 87816e4e
...@@ -169,7 +169,7 @@ def test_optimizer32bit(dim1, dim2, gtype, optim_name): ...@@ -169,7 +169,7 @@ def test_optimizer32bit(dim1, dim2, gtype, optim_name):
if gtype != torch.float32: if gtype != torch.float32:
# the adam buffers should also be close because they are 32-bit # the adam buffers should also be close because they are 32-bit
# but the paramters can diverge because they are 16-bit # but the parameters can diverge because they are 16-bit
# the difference grow larger and larger with each update # the difference grow larger and larger with each update
# --> copy the state to keep weights close # --> copy the state to keep weights close
p1.data = p1.data.to(p2.dtype).float() p1.data = p1.data.to(p2.dtype).float()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment