| # |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| |
| |
| class CommonVars: |
| |
| # SAMPLER |
| TEST_SAMPLER = "sequence" |
| RANDOM_SAMPLER = "random" |
| RL_SAMPLER = "rl" |
| EA_SAMPLER = "ea" |
| BOHB_SAMPLER = "bohb" |
| |
| # EVALUATOR |
| ExpressFlow = "express_flow" |
| |
| GRAD_NORM = "grad_norm" |
| GRAD_PLAIN = "grad_plain" |
| |
| JACOB_CONV = "jacob_conv" |
| NAS_WOT = "nas_wot" |
| |
| NTK_CONDNUM = "ntk_cond_num" |
| NTK_TRACE = "ntk_trace" |
| NTK_TRACE_APPROX = "ntk_trace_approx" |
| |
| PRUNE_FISHER = "fisher" |
| PRUNE_GRASP = "grasp" |
| PRUNE_SNIP = "snip" |
| PRUNE_SYNFLOW = "synflow" |
| |
| WEIGHT_NORM = "weight_norm" |
| KNAS = "knas" |
| |
| JACFLOW = "jacflow" |
| |
| ALL_EVALUATOR = "all_matrix" |
| |
| # SEARCH SPACE |
| NASBENCH101 = "nas-bench-101" |
| NASBENCH201 = "nas-bench-201" |
| |
| # correlation coefficient metrics |
| KendallTau = "KendallTau" |
| Spearman = "Spearman" |
| Pearson = "Pearson" |
| AvgCorrelation = "average_correlation" |
| AllCorrelation = "all_correlation" |
| |
| |
| class Config: |
| |
| MLPSP = "mlp_sp" |
| NB101 = "nasbench101" |
| NB201 = "nasbench201" |
| DARTS = "darts" |
| NDS = "NDS" |
| |
| # vision dataset |
| c10_valid = "cifar10-valid" |
| c10 = "cifar10" |
| c100 = "cifar100" |
| imgNet = "ImageNet16-120" |
| imgNetFull = "ImageNet1k" |
| |
| # struct dataset |
| Frappe = "frappe" |
| Criteo = "criteo" |
| UCIDataset = "uci_diabetes" |
| |
| SUCCHALF = "SUCCHALF" |
| SUCCREJCT = "SUCCREJCT" |
| UNIFORM = "UNIFORM" |
| |
| |