Skip to content

GitLab

Commit 41bbf93c authored by lcskrishna's avatar lcskrishna
Browse files

missing import statement

parent 5bae299e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/L0/run_optimizers/test_adam.py
View file @ 41bbf93c
......@@ -5,6 +5,8 @@ import random
import torch
import apex
from apex.testing.common_utils import skipIfRocm
class TestFusedAdam(unittest.TestCase):
def setUp(self, max_abs_diff=1e-3, max_rel_diff=1, iters=7):
self.max_abs_diff = max_abs_diff
......
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