Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions official/vision/classification/resnet/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,15 +9,14 @@
import argparse
import json

import model as resnet_model
import model as resnet_model # pylint-disable=import-error

import cv2
import numpy as np

import megengine
import megengine.data.transform as T
import megengine.functional as F
import megengine.jit as jit

logging = megengine.logger.get_logger()

Expand Down
14 changes: 5 additions & 9 deletions official/vision/classification/resnet/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down
16 changes: 7 additions & 9 deletions official/vision/classification/resnet/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down
3 changes: 1 addition & 2 deletions official/vision/classification/shufflenet/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,15 +9,14 @@
import argparse
import json

import model as snet_model
import model as snet_model # pylint-disable=import-error

import cv2
import numpy as np

import megengine
import megengine.data.transform as T
import megengine.functional as F
import megengine.jit as jit

logging = megengine.logger.get_logger()

Expand Down
6 changes: 3 additions & 3 deletions official/vision/classification/shufflenet/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
14 changes: 5 additions & 9 deletions official/vision/classification/shufflenet/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down
17 changes: 7 additions & 10 deletions official/vision/classification/shufflenet/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

Expand Down