Skip to content

GitLab

Commit 27310f34 authored by lcskrishna's avatar lcskrishna
Browse files

missing import packages

parent c92b9751
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/L0/run_amp/test_multiple_models_optimizers_losses.py
View file @ 27310f34
...@@ -13,6 +13,8 @@ from torch.nn import Parameter ...@@ -13,6 +13,8 @@ from torch.nn import Parameter
from utils import common_init, HALF, FLOAT,\ from utils import common_init, HALF, FLOAT,\
ALWAYS_HALF, ALWAYS_FLOAT, MATCH_INPUT ALWAYS_HALF, ALWAYS_FLOAT, MATCH_INPUT
from apex.testing.common_utils import skipIfRocm
class MyModel(torch.nn.Module): class MyModel(torch.nn.Module):
def __init__(self, unique): def __init__(self, unique):
super(MyModel, self).__init__() super(MyModel, self).__init__()
......
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