iicd/Graph-DiT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update some relative path

Browse Source
This commit is contained in:
mhz 2024-07-29 00:20:55 +02:00
parent fd4f0452f9
commit d470261a74
13 changed files with 37 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graph_dit/naswot/naswot/datasets/get_dataset_with_transform.py
View File

@ -11,7 +11,7 @@ from PIL import Image
from .DownsampledImageNet import ImageNet16 from .DownsampledImageNet import ImageNet16
from .SearchDatasetWrap import SearchDataset from .SearchDatasetWrap import SearchDataset
from config_utils import load_config from naswot.config_utils import load_config
Dataset2Class = {'cifar10' : 10, Dataset2Class = {'cifar10' : 10,

8
graph_dit/naswot/naswot/pycls/core/benchmark.py
View File

@ -7,11 +7,11 @@
"""Benchmarking functions.""" """Benchmarking functions."""
import pycls.core.logging as logging import naswot.pycls.core.logging as logging
import pycls.datasets.loader as loader import naswot.pycls.datasets.loader as loader
import torch import torch
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
from pycls.core.timer import Timer from naswot.pycls.core.timer import Timer
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)

16
graph_dit/naswot/naswot/pycls/core/builders.py
View File

@ -8,14 +8,14 @@
"""Model and loss construction functions.""" """Model and loss construction functions."""
import torch import torch
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
from pycls.models.anynet import AnyNet from naswot.pycls.models.anynet import AnyNet
from pycls.models.effnet import EffNet from naswot.pycls.models.effnet import EffNet
from pycls.models.regnet import RegNet from naswot.pycls.models.regnet import RegNet
from pycls.models.resnet import ResNet from naswot.pycls.models.resnet import ResNet
from pycls.models.nas.nas import NAS from naswot.pycls.models.nas.nas import NAS
from pycls.models.nas.nas_search import NAS_Search from naswot.pycls.models.nas.nas_search import NAS_Search
from pycls.models.nas_bench.model_builder import NAS_Bench from naswot.pycls.models.nas_bench.model_builder import NAS_Bench
class LabelSmoothedCrossEntropyLoss(torch.nn.Module): class LabelSmoothedCrossEntropyLoss(torch.nn.Module):

4
graph_dit/naswot/naswot/pycls/core/checkpoint.py
View File

@ -9,9 +9,9 @@
import os import os
import pycls.core.distributed as dist import naswot.pycls.core.distributed as dist
import torch import torch
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
# Common prefix for checkpoint file names # Common prefix for checkpoint file names

2
graph_dit/naswot/naswot/pycls/core/config.py
View File

@ -11,7 +11,7 @@ import argparse
import os import os
import sys import sys
from pycls.core.io import cache_url from naswot.pycls.core.io import cache_url
from yacs.config import CfgNode as CfgNode from yacs.config import CfgNode as CfgNode

2
graph_dit/naswot/naswot/pycls/core/distributed.py
View File

@ -14,7 +14,7 @@ import threading
import traceback import traceback
import torch import torch
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
def is_master_proc(): def is_master_proc():

4
graph_dit/naswot/naswot/pycls/core/logging.py
View File

@ -13,9 +13,9 @@ import logging
import os import os
import sys import sys
import pycls.core.distributed as dist import naswot.pycls.core.distributed as dist
import simplejson import simplejson
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
# Show filename and line number in logs # Show filename and line number in logs

6
graph_dit/naswot/naswot/pycls/core/meters.py
View File

@ -10,10 +10,10 @@
from collections import deque from collections import deque
import numpy as np import numpy as np
import pycls.core.logging as logging import naswot.pycls.core.logging as logging
import torch import torch
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
from pycls.core.timer import Timer from naswot.pycls.core.timer import Timer
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)

2
graph_dit/naswot/naswot/pycls/core/net.py
View File

@ -12,7 +12,7 @@ import math
import torch import torch
import torch.nn as nn import torch.nn as nn
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
def init_weights(m): def init_weights(m):

2
graph_dit/naswot/naswot/pycls/core/optimizer.py
View File

@ -9,7 +9,7 @@
import numpy as np import numpy as np
import torch import torch
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
def construct_optimizer(model): def construct_optimizer(model):

4
graph_dit/naswot/naswot/pycls/models/anynet.py
View File

@ -7,9 +7,9 @@
"""AnyNet models.""" """AnyNet models."""
import pycls.core.net as net import naswot.pycls.core.net as net
import torch.nn as nn import torch.nn as nn
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
def get_stem_fun(stem_type): def get_stem_fun(stem_type):

2
graph_dit/naswot/naswot/pycls/models/common.py
View File

@ -8,7 +8,7 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
from pycls.core.config import cfg from naswot.pycls.core.config import cfg
def Preprocess(x): def Preprocess(x):

20
graph_dit/naswot/naswot/pycls/models/nas/nas.py
View File

@ -12,17 +12,17 @@ from torch.autograd import Variable
import torch import torch
import torch.nn as nn 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 naswot.pycls.core.config import cfg
from pycls.models.common import Preprocess from naswot.pycls.models.common import Preprocess
from pycls.models.common import Classifier from naswot.pycls.models.common import Classifier
from pycls.models.nas.genotypes import GENOTYPES from naswot.pycls.models.nas.genotypes import GENOTYPES
from pycls.models.nas.genotypes import Genotype from naswot.pycls.models.nas.genotypes import Genotype
from pycls.models.nas.operations import FactorizedReduce from naswot.pycls.models.nas.operations import FactorizedReduce
from pycls.models.nas.operations import OPS from naswot.pycls.models.nas.operations import OPS
from pycls.models.nas.operations import ReLUConvBN from naswot.pycls.models.nas.operations import ReLUConvBN
from pycls.models.nas.operations import Identity from naswot.pycls.models.nas.operations import Identity
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)