Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
deepspeed
Commits
5412a334
"apps/vscode:/vscode.git/clone" did not exist on "1022d5d64dae7b29cb3fda451ae4b4095632c190"
Unverified
Commit
5412a334
authored
Sep 25, 2020
by
Shaden Smith
Committed by
GitHub
Sep 25, 2020
Browse files
unit test rename (#442)
parent
6d176c45
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tests/unit/test_cpu_adam.py
tests/unit/test_cpu_adam.py
+1
-1
No files found.
tests/unit/test_adam
_acuracy
.py
→
tests/unit/test_
cpu_
adam.py
View file @
5412a334
...
...
@@ -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
))
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment