From d63ca69240400d67a77d9604b8cf24e90260efed Mon Sep 17 00:00:00 2001 From: Kai Chen Date: Mon, 28 Dec 2020 11:47:39 +0800 Subject: [PATCH] fix typo --- mmcv/utils/parrots_jit.py | 4 +--- mmcv/utils/parrots_wrapper.py | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/mmcv/utils/parrots_jit.py b/mmcv/utils/parrots_jit.py index 7c4eca9cdd..609c11acd0 100644 --- a/mmcv/utils/parrots_jit.py +++ b/mmcv/utils/parrots_jit.py @@ -1,6 +1,4 @@ -import torch - -TORCH_VERSION = torch.__version__ +from .parrots_wrapper import TORCH_VERSION if TORCH_VERSION == 'parrots': from parrots.jit import pat as jit diff --git a/mmcv/utils/parrots_wrapper.py b/mmcv/utils/parrots_wrapper.py index 675bc07471..25761be835 100644 --- a/mmcv/utils/parrots_wrapper.py +++ b/mmcv/utils/parrots_wrapper.py @@ -2,7 +2,7 @@ import torch -from .parrots_wrapper import TORCH_VERSION +TORCH_VERSION = torch.__version__ def _get_cuda_home():