未验证 提交 5412a334 编写于 作者: S Shaden Smith 提交者: GitHub

unit test rename (#442)

上级 6d176c45
......@@ -31,7 +31,7 @@ def check_equal(first, second, atol=1e-2, verbose=False):
(1024),
(1048576),
]) # yapf: disable
def test_adam_opt(model_size):
def test_cpu_adam_opt(model_size):
device = 'cpu'
rng_state = torch.get_rng_state()
param = torch.nn.Parameter(torch.randn(model_size, device=device))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册