diff --git a/official/vision/classification/resnet/inference.py b/official/vision/classification/resnet/inference.py index 042c85f5..3a8d6104 100644 --- a/official/vision/classification/resnet/inference.py +++ b/official/vision/classification/resnet/inference.py @@ -9,7 +9,7 @@ import argparse import json -import model as resnet_model +import model as resnet_model # pylint-disable=import-error import cv2 import numpy as np @@ -17,7 +17,6 @@ import megengine import megengine.data.transform as T import megengine.functional as F -import megengine.jit as jit logging = megengine.logger.get_logger() diff --git a/official/vision/classification/resnet/test.py b/official/vision/classification/resnet/test.py index 1e8a3292..d1878d06 100644 --- a/official/vision/classification/resnet/test.py +++ b/official/vision/classification/resnet/test.py @@ -7,20 +7,16 @@ # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. import argparse -import bisect import multiprocessing -import os -import threading import time -import model as resnet_model +import model as resnet_model # pylint-disable=import-error import megengine -from megengine import data as data -from megengine import distributed as dist -from megengine import functional as F -from megengine import jit as jit -from megengine.data import transform as T +import megengine.data as data +import megengine.data.transform as T +import megengine.distributed as dist +import megengine.functional as F logging = megengine.logger.get_logger() diff --git a/official/vision/classification/resnet/train.py b/official/vision/classification/resnet/train.py index fb39a44d..c8d7b7d2 100644 --- a/official/vision/classification/resnet/train.py +++ b/official/vision/classification/resnet/train.py @@ -10,19 +10,17 @@ import bisect import multiprocessing import os -import threading import time -import model as resnet_model +import model as resnet_model # pylint-disable=import-error import megengine -from megengine import autodiff as autodiff -from megengine import data as data -from megengine import distributed as dist -from megengine import functional as F -from megengine import jit as jit -from megengine import optimizer as optim -from megengine.data import transform as T +import megengine.autodiff as autodiff +import megengine.data as data +import megengine.data.transform as T +import megengine.distributed as dist +import megengine.functional as F +import megengine.optimizer as optim logging = megengine.logger.get_logger() diff --git a/official/vision/classification/shufflenet/inference.py b/official/vision/classification/shufflenet/inference.py index 44c3528c..7d598daa 100644 --- a/official/vision/classification/shufflenet/inference.py +++ b/official/vision/classification/shufflenet/inference.py @@ -9,7 +9,7 @@ import argparse import json -import model as snet_model +import model as snet_model # pylint-disable=import-error import cv2 import numpy as np @@ -17,7 +17,6 @@ import megengine import megengine.data.transform as T import megengine.functional as F -import megengine.jit as jit logging = megengine.logger.get_logger() diff --git a/official/vision/classification/shufflenet/model.py b/official/vision/classification/shufflenet/model.py index b923fc59..3d971eaf 100644 --- a/official/vision/classification/shufflenet/model.py +++ b/official/vision/classification/shufflenet/model.py @@ -33,9 +33,9 @@ # This file has been modified by Megvii ("Megvii Modifications"). # All Megvii Modifications are Copyright (C) 2014-2019 Megvii Inc. All rights reserved. # ------------------------------------------------------------------------------ -from megengine import functional as F -from megengine import hub as hub -from megengine import module as M +import megengine.functional as F +import megengine.hub as hub +import megengine.module as M class ShuffleV2Block(M.Module): diff --git a/official/vision/classification/shufflenet/test.py b/official/vision/classification/shufflenet/test.py index fc72ae41..8a2a428e 100644 --- a/official/vision/classification/shufflenet/test.py +++ b/official/vision/classification/shufflenet/test.py @@ -7,20 +7,16 @@ # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. import argparse -import bisect import multiprocessing -import os -import threading import time -import model as snet_model +import model as snet_model # pylint-disable=import-error import megengine -from megengine import data as data -from megengine import distributed as dist -from megengine import functional as F -from megengine import jit as jit -from megengine.data import transform as T +import megengine.data as data +import megengine.data.transform as T +import megengine.distributed as dist +import megengine.functional as F logging = megengine.logger.get_logger() diff --git a/official/vision/classification/shufflenet/train.py b/official/vision/classification/shufflenet/train.py index 42011428..cfe8fb98 100644 --- a/official/vision/classification/shufflenet/train.py +++ b/official/vision/classification/shufflenet/train.py @@ -7,22 +7,19 @@ # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. import argparse -import bisect import multiprocessing import os -import threading import time -import model as snet_model +import model as snet_model # pylint-disable=import-error import megengine -from megengine import autodiff as autodiff -from megengine import data as data -from megengine import distributed as dist -from megengine import functional as F -from megengine import jit as jit -from megengine import optimizer as optim -from megengine.data import transform as T +import megengine.autodiff as autodiff +import megengine.data as data +import megengine.data.transform as T +import megengine.distributed as dist +import megengine.functional as F +import megengine.optimizer as optim logging = megengine.logger.get_logger()