102->201 / NAS->autoDL / more configs of TAS / reorganize docs / fix bugs in NAS baselines
This commit is contained in:
@@ -2,3 +2,4 @@
|
||||
from .CifarNet import NetworkCIFAR as CifarNet
|
||||
from .ImageNet import NetworkImageNet as ImageNet
|
||||
from .genotypes import Networks
|
||||
from .genotypes import build_genotype_from_dict
|
||||
|
@@ -167,3 +167,6 @@ Networks = {'DARTS_V1': DARTS_V1,
|
||||
'PNASNet' : PNASNet,
|
||||
'SETN' : SETN,
|
||||
}
|
||||
|
||||
def build_genotype_from_dict(xdict):
|
||||
import pdb; pdb.set_trace()
|
||||
|
@@ -1,6 +1,11 @@
|
||||
##################################################
|
||||
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
|
||||
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2020 #
|
||||
##################################################
|
||||
# I write this package to make AutoDL-Projects to be compatible with the old GDAS projects.
|
||||
# Ideally, this package will be merged into lib/models/cell_infers in future.
|
||||
# Currently, this package is used to reproduce the results in GDAS (Searching for A Robust Neural Architecture in Four GPU Hours, CVPR 2019).
|
||||
##################################################
|
||||
|
||||
import torch
|
||||
|
||||
def obtain_nas_infer_model(config):
|
||||
|
Reference in New Issue
Block a user