Unverified Commit c3835415 authored by Zaida Zhou's avatar Zaida Zhou Committed by GitHub
Browse files

[Refactor] Modify the importing path due to mmengine changes (#2224)

parent 5fdd246a
...@@ -2,7 +2,8 @@ ...@@ -2,7 +2,8 @@
import numpy as np import numpy as np
import pytest import pytest
import torch import torch
from mmengine.utils import TORCH_VERSION, digit_version from mmengine.utils import digit_version
from mmengine.utils.dl_utils import TORCH_VERSION
try: try:
# If PyTorch version >= 1.6.0 and fp16 is enabled, torch.cuda.amp.autocast # If PyTorch version >= 1.6.0 and fp16 is enabled, torch.cuda.amp.autocast
......
...@@ -4,7 +4,8 @@ import os ...@@ -4,7 +4,8 @@ import os
import numpy import numpy
import pytest import pytest
import torch import torch
from mmengine.utils import TORCH_VERSION, digit_version from mmengine.utils import digit_version
from mmengine.utils.dl_utils import TORCH_VERSION
try: try:
# If PyTorch version >= 1.6.0 and fp16 is enabled, torch.cuda.amp.autocast # If PyTorch version >= 1.6.0 and fp16 is enabled, torch.cuda.amp.autocast
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import pytest import pytest
import torch import torch
from mmengine.utils import TORCH_VERSION from mmengine.utils.dl_utils import TORCH_VERSION
import mmcv import mmcv
......
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