iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refine lib -> xautodl

Browse Source
This commit is contained in:
D-X-Y
2021-05-19 13:00:33 +08:00
parent 94a149b33f
commit 95fb5a54b1
34 changed files with 118 additions and 1154 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
xautodl/models/__init__.py
View File

@@ -19,9 +19,9 @@ __all__ = [
]
# useful modules
from config_utils import dict2config
from models.SharedUtils import change_key
from models.cell_searchs import CellStructure, CellArchitectures
from xautodl.config_utils import dict2config
from .SharedUtils import change_key
from .cell_searchs import CellStructure, CellArchitectures
# Cell-based NAS Models

5
xautodl/models/cell_searchs/search_model_gdas_frc_nasnet.py
View File

@@ -4,8 +4,9 @@
import torch
import torch.nn as nn
from copy import deepcopy
from models.cell_searchs.search_cells import NASNetSearchCell as SearchCell
from models.cell_operations import RAW_OP_CLASSES
from .search_cells import NASNetSearchCell as SearchCell
from ..cell_operations import RAW_OP_CLASSES
# The macro structure is based on NASNet