Skip to content

Commit

Permalink
minor changes
Browse files Browse the repository at this point in the history
  • Loading branch information
rohitgr7 committed Sep 16, 2020
1 parent 262edc8 commit ba0c80d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/models/test_amp.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
import tests.base.develop_pipelines as tpipes
import tests.base.develop_utils as tutils
from pytorch_lightning import Trainer
from pytorch_lightning.loggers import WandbLogger
from pytorch_lightning.trainer.states import TrainerState
from pytorch_lightning.utilities.exceptions import MisconfigurationException
from tests.base import EvalModelTemplate
Expand Down Expand Up @@ -149,7 +150,7 @@ def test_cpu_model_with_amp(tmpdir):


def test_amp_without_apex(tmpdir):
"""Check that even with apex amp type without requesting percussion=16 the amp backend is void."""
"""Check that even with apex amp type without requesting precision=16 the amp backend is void."""
os.environ['PL_DEV_DEBUG'] = '1'
model = EvalModelTemplate()

Expand Down

0 comments on commit ba0c80d

Please sign in to comment.