Path Lines of Code nni/algorithms/__init__.py 1 nni/algorithms/compression/__init__.py 1 nni/algorithms/compression/pytorch/__init__.py 1 nni/algorithms/compression/pytorch/auto_compress/__init__.py 3 nni/algorithms/compression/pytorch/auto_compress/auto_compress_engine.py 139 nni/algorithms/compression/pytorch/auto_compress/constants.py 19 nni/algorithms/compression/pytorch/auto_compress/experiment.py 24 nni/algorithms/compression/pytorch/auto_compress/interface.py 38 nni/algorithms/compression/pytorch/auto_compress/trial_entry.py 12 nni/algorithms/compression/pytorch/auto_compress/utils.py 67 nni/algorithms/compression/pytorch/pruning/__init__.py 12 nni/algorithms/compression/pytorch/pruning/amc/__init__.py 1 nni/algorithms/compression/pytorch/pruning/amc/amc_pruner.py 158 nni/algorithms/compression/pytorch/pruning/amc/channel_pruning_env.py 338 nni/algorithms/compression/pytorch/pruning/amc/lib/__init__.py 1 nni/algorithms/compression/pytorch/pruning/amc/lib/agent.py 154 nni/algorithms/compression/pytorch/pruning/amc/lib/memory.py 151 nni/algorithms/compression/pytorch/pruning/amc/lib/net_measure.py 84 nni/algorithms/compression/pytorch/pruning/amc/lib/utils.py 73 nni/algorithms/compression/pytorch/pruning/apply_compression.py 10 nni/algorithms/compression/pytorch/pruning/auto_compress_pruner.py 115 nni/algorithms/compression/pytorch/pruning/constants.py 13 nni/algorithms/compression/pytorch/pruning/constants_pruner.py 7 nni/algorithms/compression/pytorch/pruning/dependency_aware_pruner.py 87 nni/algorithms/compression/pytorch/pruning/finegrained_pruning_masker.py 18 nni/algorithms/compression/pytorch/pruning/iterative_pruner.py 231 nni/algorithms/compression/pytorch/pruning/lottery_ticket.py 72 nni/algorithms/compression/pytorch/pruning/net_adapt_pruner.py 189 nni/algorithms/compression/pytorch/pruning/one_shot_pruner.py 42 nni/algorithms/compression/pytorch/pruning/sensitivity_pruner.py 207 nni/algorithms/compression/pytorch/pruning/simulated_annealing_pruner.py 207 nni/algorithms/compression/pytorch/pruning/structured_pruning_masker.py 452 nni/algorithms/compression/pytorch/pruning/transformer_pruner.py 174 nni/algorithms/compression/pytorch/pruning/transformer_pruning_head_masker.py 234 nni/algorithms/compression/pytorch/pruning/weight_masker.py 6 nni/algorithms/compression/pytorch/quantization/__init__.py 7 nni/algorithms/compression/pytorch/quantization/bnn_quantizer.py 65 nni/algorithms/compression/pytorch/quantization/dorefa_quantizer.py 58 nni/algorithms/compression/pytorch/quantization/lsq_quantizer.py 147 nni/algorithms/compression/pytorch/quantization/native_quantizer.py 28 nni/algorithms/compression/pytorch/quantization/observer_quantizer.py 165 nni/algorithms/compression/pytorch/quantization/qat_quantizer.py 285 nni/algorithms/compression/tensorflow/pruning/__init__.py 1 nni/algorithms/compression/tensorflow/pruning/one_shot_pruner.py 86 nni/algorithms/compression/v2/pytorch/__init__.py 1 nni/algorithms/compression/v2/pytorch/base/__init__.py 3 nni/algorithms/compression/v2/pytorch/base/compressor.py 143 nni/algorithms/compression/v2/pytorch/base/pruner.py 79 nni/algorithms/compression/v2/pytorch/base/scheduler.py 86 nni/algorithms/compression/v2/pytorch/pruning/__init__.py 6 nni/algorithms/compression/v2/pytorch/pruning/amc_pruner.py 123 nni/algorithms/compression/v2/pytorch/pruning/auto_compress_pruner.py 53 nni/algorithms/compression/v2/pytorch/pruning/basic_pruner.py 400 nni/algorithms/compression/v2/pytorch/pruning/basic_scheduler.py 89 nni/algorithms/compression/v2/pytorch/pruning/iterative_pruner.py 110 nni/algorithms/compression/v2/pytorch/pruning/movement_pruner.py 152 nni/algorithms/compression/v2/pytorch/pruning/tools/__init__.py 31 nni/algorithms/compression/v2/pytorch/pruning/tools/base.py 262 nni/algorithms/compression/v2/pytorch/pruning/tools/data_collector.py 29 nni/algorithms/compression/v2/pytorch/pruning/tools/metrics_calculator.py 83 nni/algorithms/compression/v2/pytorch/pruning/tools/rl_env/__init__.py 2 nni/algorithms/compression/v2/pytorch/pruning/tools/rl_env/agent.py 155 nni/algorithms/compression/v2/pytorch/pruning/tools/rl_env/amc_env.py 107 nni/algorithms/compression/v2/pytorch/pruning/tools/rl_env/memory.py 151 nni/algorithms/compression/v2/pytorch/pruning/tools/sparsity_allocator.py 129 nni/algorithms/compression/v2/pytorch/pruning/tools/task_generator.py 238 nni/algorithms/compression/v2/pytorch/utils/__init__.py 12 nni/algorithms/compression/v2/pytorch/utils/config_validation.py 40 nni/algorithms/compression/v2/pytorch/utils/constructor_helper.py 95 nni/algorithms/compression/v2/pytorch/utils/pruning.py 193 nni/algorithms/feature_engineering/__init__.py 1 nni/algorithms/feature_engineering/gbdt_selector/__init__.py 1 nni/algorithms/feature_engineering/gbdt_selector/gbdt_selector.py 45 nni/algorithms/feature_engineering/gradient_selector/__init__.py 1 nni/algorithms/feature_engineering/gradient_selector/constants.py 39 nni/algorithms/feature_engineering/gradient_selector/fginitialize.py 374 nni/algorithms/feature_engineering/gradient_selector/fgtrain.py 154 nni/algorithms/feature_engineering/gradient_selector/gradient_selector.py 346 nni/algorithms/feature_engineering/gradient_selector/learnability.py 308 nni/algorithms/feature_engineering/gradient_selector/syssettings.py 4 nni/algorithms/feature_engineering/gradient_selector/utils.py 41 nni/algorithms/hpo/__init__.py 1 nni/algorithms/hpo/batch_tuner.py 57 nni/algorithms/hpo/bohb_advisor/__init__.py 1 nni/algorithms/hpo/bohb_advisor/bohb_advisor.py 384 nni/algorithms/hpo/bohb_advisor/config_generator.py 171 nni/algorithms/hpo/curvefitting_assessor/__init__.py 1 nni/algorithms/hpo/curvefitting_assessor/curvefitting_assessor.py 72 nni/algorithms/hpo/curvefitting_assessor/curvefunctions.py 116 nni/algorithms/hpo/curvefitting_assessor/model_factory.py 145 nni/algorithms/hpo/dngo_tuner.py 107 nni/algorithms/hpo/evolution_tuner.py 129 nni/algorithms/hpo/gp_tuner/__init__.py 1 nni/algorithms/hpo/gp_tuner/gp_tuner.py 98 nni/algorithms/hpo/gp_tuner/target_space.py 112 nni/algorithms/hpo/gp_tuner/util.py 73 nni/algorithms/hpo/gridsearch_tuner.py 142 nni/algorithms/hpo/hyperband_advisor.py 247 nni/algorithms/hpo/hyperopt_tuner.py 337 nni/algorithms/hpo/medianstop_assessor.py 65 nni/algorithms/hpo/metis_tuner/Regression_GMM/CreateModel.py 29 nni/algorithms/hpo/metis_tuner/Regression_GMM/Selection.py 73 nni/algorithms/hpo/metis_tuner/Regression_GMM/__init__.py 1 nni/algorithms/hpo/metis_tuner/Regression_GP/CreateModel.py 26 nni/algorithms/hpo/metis_tuner/Regression_GP/OutlierDetection.py 54 nni/algorithms/hpo/metis_tuner/Regression_GP/Prediction.py 11 nni/algorithms/hpo/metis_tuner/Regression_GP/Selection.py 79 nni/algorithms/hpo/metis_tuner/Regression_GP/__init__.py 1 nni/algorithms/hpo/metis_tuner/__init__.py 1 nni/algorithms/hpo/metis_tuner/lib_acquisition_function.py 138 nni/algorithms/hpo/metis_tuner/lib_constraint_summation.py 72 nni/algorithms/hpo/metis_tuner/lib_data.py 35 nni/algorithms/hpo/metis_tuner/metis_tuner.py 424 nni/algorithms/hpo/networkmorphism_tuner/__init__.py 1 nni/algorithms/hpo/networkmorphism_tuner/bayesian.py 326 nni/algorithms/hpo/networkmorphism_tuner/graph.py 732 nni/algorithms/hpo/networkmorphism_tuner/graph_transformer.py 126 nni/algorithms/hpo/networkmorphism_tuner/layer_transformer.py 217 nni/algorithms/hpo/networkmorphism_tuner/layers.py 506 nni/algorithms/hpo/networkmorphism_tuner/networkmorphism_tuner.py 147 nni/algorithms/hpo/networkmorphism_tuner/nn.py 107 nni/algorithms/hpo/networkmorphism_tuner/utils.py 19 nni/algorithms/hpo/pbt_tuner.py 274 nni/algorithms/hpo/ppo_tuner/__init__.py 1 nni/algorithms/hpo/ppo_tuner/distri.py 131 nni/algorithms/hpo/ppo_tuner/model.py 72 nni/algorithms/hpo/ppo_tuner/policy.py 107 nni/algorithms/hpo/ppo_tuner/ppo_tuner.py 368 nni/algorithms/hpo/ppo_tuner/util.py 132 nni/algorithms/hpo/random_tuner.py 38 nni/algorithms/hpo/regularized_evolution_tuner.py 101 nni/algorithms/hpo/smac_tuner/__init__.py 1 nni/algorithms/hpo/smac_tuner/convert_ss_to_scenario.py 67 nni/algorithms/hpo/smac_tuner/smac_tuner.py 199 nni/algorithms/hpo/tpe_tuner.py 204 nni/algorithms/nas/__init__.py 1 nni/algorithms/nas/pytorch/__init__.py 1 nni/algorithms/nas/pytorch/cdarts/__init__.py 2 nni/algorithms/nas/pytorch/cdarts/mutator.py 53 nni/algorithms/nas/pytorch/cdarts/trainer.py 178 nni/algorithms/nas/pytorch/cdarts/utils.py 53 nni/algorithms/nas/pytorch/classic_nas/__init__.py 1 nni/algorithms/nas/pytorch/classic_nas/mutator.py 110 nni/algorithms/nas/pytorch/cream/__init__.py 1 nni/algorithms/nas/pytorch/cream/trainer.py 256 nni/algorithms/nas/pytorch/cream/utils.py 22 nni/algorithms/nas/pytorch/darts/__init__.py 2 nni/algorithms/nas/pytorch/darts/mutator.py 52 nni/algorithms/nas/pytorch/darts/trainer.py 124 nni/algorithms/nas/pytorch/enas/__init__.py 2 nni/algorithms/nas/pytorch/enas/mutator.py 141 nni/algorithms/nas/pytorch/enas/trainer.py 133 nni/algorithms/nas/pytorch/fbnet/__init__.py 10 nni/algorithms/nas/pytorch/fbnet/mutator.py 92 nni/algorithms/nas/pytorch/fbnet/trainer.py 265 nni/algorithms/nas/pytorch/fbnet/utils.py 220 nni/algorithms/nas/pytorch/pdarts/__init__.py 1 nni/algorithms/nas/pytorch/pdarts/mutator.py 58 nni/algorithms/nas/pytorch/pdarts/trainer.py 57 nni/algorithms/nas/pytorch/proxylessnas/__init__.py 2 nni/algorithms/nas/pytorch/proxylessnas/mutator.py 238 nni/algorithms/nas/pytorch/proxylessnas/trainer.py 300 nni/algorithms/nas/pytorch/proxylessnas/utils.py 28 nni/algorithms/nas/pytorch/random/__init__.py 1 nni/algorithms/nas/pytorch/random/mutator.py 21 nni/algorithms/nas/pytorch/spos/__init__.py 3 nni/algorithms/nas/pytorch/spos/evolution.py 150 nni/algorithms/nas/pytorch/spos/mutator.py 25 nni/algorithms/nas/pytorch/spos/trainer.py 49 nni/algorithms/nas/tensorflow/__init__.py 1 nni/algorithms/nas/tensorflow/classic_nas/__init__.py 1 nni/algorithms/nas/tensorflow/classic_nas/mutator.py 109 nni/algorithms/nas/tensorflow/enas/__init__.py 2 nni/algorithms/nas/tensorflow/enas/mutator.py 132 nni/algorithms/nas/tensorflow/enas/trainer.py 169