diff --git a/graph_dit/naswot/naswot/datasets/get_dataset_with_transform.py b/graph_dit/naswot/naswot/datasets/get_dataset_with_transform.py index 48513ff..4c7bb14 100644 --- a/graph_dit/naswot/naswot/datasets/get_dataset_with_transform.py +++ b/graph_dit/naswot/naswot/datasets/get_dataset_with_transform.py @@ -11,7 +11,7 @@ from PIL import Image from .DownsampledImageNet import ImageNet16 from .SearchDatasetWrap import SearchDataset -from config_utils import load_config +from naswot.config_utils import load_config Dataset2Class = {'cifar10' : 10, diff --git a/graph_dit/naswot/naswot/pycls/core/benchmark.py b/graph_dit/naswot/naswot/pycls/core/benchmark.py index b07c892..a2d4115 100644 --- a/graph_dit/naswot/naswot/pycls/core/benchmark.py +++ b/graph_dit/naswot/naswot/pycls/core/benchmark.py @@ -7,11 +7,11 @@ """Benchmarking functions.""" -import pycls.core.logging as logging -import pycls.datasets.loader as loader +import naswot.pycls.core.logging as logging +import naswot.pycls.datasets.loader as loader import torch -from pycls.core.config import cfg -from pycls.core.timer import Timer +from naswot.pycls.core.config import cfg +from naswot.pycls.core.timer import Timer logger = logging.get_logger(__name__) diff --git a/graph_dit/naswot/naswot/pycls/core/builders.py b/graph_dit/naswot/naswot/pycls/core/builders.py index bdf0778..4956658 100644 --- a/graph_dit/naswot/naswot/pycls/core/builders.py +++ b/graph_dit/naswot/naswot/pycls/core/builders.py @@ -8,14 +8,14 @@ """Model and loss construction functions.""" import torch -from pycls.core.config import cfg -from pycls.models.anynet import AnyNet -from pycls.models.effnet import EffNet -from pycls.models.regnet import RegNet -from pycls.models.resnet import ResNet -from pycls.models.nas.nas import NAS -from pycls.models.nas.nas_search import NAS_Search -from pycls.models.nas_bench.model_builder import NAS_Bench +from naswot.pycls.core.config import cfg +from naswot.pycls.models.anynet import AnyNet +from naswot.pycls.models.effnet import EffNet +from naswot.pycls.models.regnet import RegNet +from naswot.pycls.models.resnet import ResNet +from naswot.pycls.models.nas.nas import NAS +from naswot.pycls.models.nas.nas_search import NAS_Search +from naswot.pycls.models.nas_bench.model_builder import NAS_Bench class LabelSmoothedCrossEntropyLoss(torch.nn.Module): diff --git a/graph_dit/naswot/naswot/pycls/core/checkpoint.py b/graph_dit/naswot/naswot/pycls/core/checkpoint.py index 06bca15..62ee9aa 100644 --- a/graph_dit/naswot/naswot/pycls/core/checkpoint.py +++ b/graph_dit/naswot/naswot/pycls/core/checkpoint.py @@ -9,9 +9,9 @@ import os -import pycls.core.distributed as dist +import naswot.pycls.core.distributed as dist import torch -from pycls.core.config import cfg +from naswot.pycls.core.config import cfg # Common prefix for checkpoint file names diff --git a/graph_dit/naswot/naswot/pycls/core/config.py b/graph_dit/naswot/naswot/pycls/core/config.py index cb28145..7cad5c2 100644 --- a/graph_dit/naswot/naswot/pycls/core/config.py +++ b/graph_dit/naswot/naswot/pycls/core/config.py @@ -11,7 +11,7 @@ import argparse import os import sys -from pycls.core.io import cache_url +from naswot.pycls.core.io import cache_url from yacs.config import CfgNode as CfgNode diff --git a/graph_dit/naswot/naswot/pycls/core/distributed.py b/graph_dit/naswot/naswot/pycls/core/distributed.py index e2dc7db..67cb6dc 100644 --- a/graph_dit/naswot/naswot/pycls/core/distributed.py +++ b/graph_dit/naswot/naswot/pycls/core/distributed.py @@ -14,7 +14,7 @@ import threading import traceback import torch -from pycls.core.config import cfg +from naswot.pycls.core.config import cfg def is_master_proc(): diff --git a/graph_dit/naswot/naswot/pycls/core/logging.py b/graph_dit/naswot/naswot/pycls/core/logging.py index 31750b2..0827ea0 100644 --- a/graph_dit/naswot/naswot/pycls/core/logging.py +++ b/graph_dit/naswot/naswot/pycls/core/logging.py @@ -13,9 +13,9 @@ import logging import os import sys -import pycls.core.distributed as dist +import naswot.pycls.core.distributed as dist import simplejson -from pycls.core.config import cfg +from naswot.pycls.core.config import cfg # Show filename and line number in logs diff --git a/graph_dit/naswot/naswot/pycls/core/meters.py b/graph_dit/naswot/naswot/pycls/core/meters.py index a1aab7d..7aea1f2 100644 --- a/graph_dit/naswot/naswot/pycls/core/meters.py +++ b/graph_dit/naswot/naswot/pycls/core/meters.py @@ -10,10 +10,10 @@ from collections import deque import numpy as np -import pycls.core.logging as logging +import naswot.pycls.core.logging as logging import torch -from pycls.core.config import cfg -from pycls.core.timer import Timer +from naswot.pycls.core.config import cfg +from naswot.pycls.core.timer import Timer logger = logging.get_logger(__name__) diff --git a/graph_dit/naswot/naswot/pycls/core/net.py b/graph_dit/naswot/naswot/pycls/core/net.py index 6d18a7f..8102b31 100644 --- a/graph_dit/naswot/naswot/pycls/core/net.py +++ b/graph_dit/naswot/naswot/pycls/core/net.py @@ -12,7 +12,7 @@ import math import torch import torch.nn as nn -from pycls.core.config import cfg +from naswot.pycls.core.config import cfg def init_weights(m): diff --git a/graph_dit/naswot/naswot/pycls/core/optimizer.py b/graph_dit/naswot/naswot/pycls/core/optimizer.py index 766a984..c2b03f2 100644 --- a/graph_dit/naswot/naswot/pycls/core/optimizer.py +++ b/graph_dit/naswot/naswot/pycls/core/optimizer.py @@ -9,7 +9,7 @@ import numpy as np import torch -from pycls.core.config import cfg +from naswot.pycls.core.config import cfg def construct_optimizer(model): diff --git a/graph_dit/naswot/naswot/pycls/models/anynet.py b/graph_dit/naswot/naswot/pycls/models/anynet.py index 32a620c..e64c613 100644 --- a/graph_dit/naswot/naswot/pycls/models/anynet.py +++ b/graph_dit/naswot/naswot/pycls/models/anynet.py @@ -7,9 +7,9 @@ """AnyNet models.""" -import pycls.core.net as net +import naswot.pycls.core.net as net import torch.nn as nn -from pycls.core.config import cfg +from naswot.pycls.core.config import cfg def get_stem_fun(stem_type): diff --git a/graph_dit/naswot/naswot/pycls/models/common.py b/graph_dit/naswot/naswot/pycls/models/common.py index 0f78466..bdd4f9e 100644 --- a/graph_dit/naswot/naswot/pycls/models/common.py +++ b/graph_dit/naswot/naswot/pycls/models/common.py @@ -8,7 +8,7 @@ import torch import torch.nn as nn -from pycls.core.config import cfg +from naswot.pycls.core.config import cfg def Preprocess(x): diff --git a/graph_dit/naswot/naswot/pycls/models/nas/nas.py b/graph_dit/naswot/naswot/pycls/models/nas/nas.py index 13339e3..2044a98 100644 --- a/graph_dit/naswot/naswot/pycls/models/nas/nas.py +++ b/graph_dit/naswot/naswot/pycls/models/nas/nas.py @@ -12,17 +12,17 @@ from torch.autograd import Variable import torch import torch.nn as nn -import pycls.core.logging as logging +import naswot.pycls.core.logging as logging -from pycls.core.config import cfg -from pycls.models.common import Preprocess -from pycls.models.common import Classifier -from pycls.models.nas.genotypes import GENOTYPES -from pycls.models.nas.genotypes import Genotype -from pycls.models.nas.operations import FactorizedReduce -from pycls.models.nas.operations import OPS -from pycls.models.nas.operations import ReLUConvBN -from pycls.models.nas.operations import Identity +from naswot.pycls.core.config import cfg +from naswot.pycls.models.common import Preprocess +from naswot.pycls.models.common import Classifier +from naswot.pycls.models.nas.genotypes import GENOTYPES +from naswot.pycls.models.nas.genotypes import Genotype +from naswot.pycls.models.nas.operations import FactorizedReduce +from naswot.pycls.models.nas.operations import OPS +from naswot.pycls.models.nas.operations import ReLUConvBN +from naswot.pycls.models.nas.operations import Identity logger = logging.get_logger(__name__)