Archai
Contents:
Installing Archai
Archai Features
Archai - A 30 Minute Tutorial
Petridish - Code Walkthrough
Frequently Asked Questions (FAQs)
Roadmap
Directory Structure
APIs
Archai
Docs
»
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Z
A
Accumulator (class in archai.common.metrics)
accuracy() (in module archai.common.ml_utils)
activations() (archai.algos.divnas.divop.DivOp property)
add() (archai.common.metrics.Accumulator method)
add_dict() (archai.common.metrics.Accumulator method)
add_epoch() (archai.common.metrics.RunMetrics method)
add_named_augs() (in module archai.datasets.augmentation)
add_scalar() (archai.common.common.SummaryWriterDummy method)
add_timing() (in module archai.common.timing)
affine_grid_generator() (in module archai.nas.operations)
all_empty() (archai.nas.model_desc.CellDesc method)
(archai.nas.model_desc.ModelDesc method)
all_full() (archai.nas.model_desc.CellDesc method)
(archai.nas.model_desc.ModelDesc method)
all_owned() (archai.nas.arch_module.ArchModule method)
ApexUtils (class in archai.common.apex_utils)
append() (archai.common.multi_optim.MultiOptim method)
append_csv_file() (in module archai.common.utils)
append_to_filename() (in module archai.common.utils)
apply_augment() (in module archai.datasets.augmentation)
arch_params() (archai.nas.arch_module.ArchModule method)
archai
module
archai.algos
module
archai.algos.darts
module
archai.algos.darts.bilevel_arch_trainer
module
archai.algos.darts.bilevel_optimizer
module
archai.algos.darts.bilevel_optimizer_slow
module
archai.algos.darts.darts_exp_runner
module
archai.algos.darts.darts_model_desc_builder
module
archai.algos.darts.mixed_op
module
archai.algos.divnas
module
archai.algos.divnas.analyse_activations
module
archai.algos.divnas.divnas_cell
module
archai.algos.divnas.divnas_exp_runner
module
archai.algos.divnas.divnas_finalizers
module
archai.algos.divnas.divnas_model_desc_builder
module
archai.algos.divnas.divnas_rank_finalizer
module
archai.algos.divnas.divop
module
archai.algos.divnas.seqopt
module
archai.algos.divnas.wmr
module
archai.algos.gumbelsoftmax
module
archai.algos.gumbelsoftmax.gs_arch_trainer
module
archai.algos.gumbelsoftmax.gs_exp_runner
module
archai.algos.gumbelsoftmax.gs_finalizers
module
archai.algos.gumbelsoftmax.gs_model_desc_builder
module
archai.algos.gumbelsoftmax.gs_op
module
archai.algos.manual
module
archai.algos.manual.manual_evaluater
module
archai.algos.manual.manual_exp_runner
module
archai.algos.manual.manual_searcher
module
archai.algos.petridish
module
archai.algos.petridish.evaluater_petridish
module
archai.algos.petridish.petridish_exp_runner
module
archai.algos.petridish.petridish_model_desc_builder
module
archai.algos.petridish.petridish_op
module
archai.algos.petridish.petridish_utils
module
archai.algos.petridish.searcher_petridish
module
archai.algos.random
module
archai.algos.random.random_exp_runner
module
archai.algos.random.random_model_desc_builder
module
archai.algos.xnas
module
archai.algos.xnas.xnas_arch_trainer
module
archai.algos.xnas.xnas_exp_runner
module
archai.algos.xnas.xnas_model_desc_builder
module
archai.algos.xnas.xnas_op
module
archai.common
module
archai.common.apex_utils
module
archai.common.atomic_file_handler
module
archai.common.checkpoint
module
archai.common.cocob
module
archai.common.common
module
archai.common.config
module
archai.common.metrics
module
archai.common.ml_losses
module
archai.common.ml_utils
module
archai.common.model_summary
module
archai.common.multi_optim
module
archai.common.ordereddict_logger
module
archai.common.stopwatch
module
archai.common.tester
module
archai.common.timing
module
archai.common.trainer
module
archai.common.utils
module
archai.common.warmup_scheduler
module
archai.common.yaml_utils
module
archai.datasets
module
archai.datasets.aug_policies
module
archai.datasets.augmentation
module
archai.datasets.data
module
archai.datasets.dataset_provider
module
archai.datasets.distributed_stratified_sampler
module
archai.datasets.limit_dataset
module
archai.datasets.providers
module
archai.datasets.providers.cifar100_provider
module
archai.datasets.providers.cifar10_provider
module
archai.datasets.providers.fashion_mnist_provider
module
archai.datasets.providers.flower102_provider
module
archai.datasets.providers.food101_provider
module
archai.datasets.providers.imagenet_folder
module
archai.datasets.providers.imagenet_provider
module
archai.datasets.providers.mit67_provider
module
archai.datasets.providers.mnist_provider
module
archai.datasets.providers.sport8_provider
module
archai.datasets.providers.svhn_provider
module
archai.datasets.transforms
module
archai.datasets.transforms.brightness
module
archai.datasets.transforms.cutout_custom
module
archai.datasets.transforms.lighting
module
archai.nas
module
archai.nas.arch_module
module
archai.nas.arch_params
module
archai.nas.arch_trainer
module
archai.nas.cell
module
archai.nas.dag_edge
module
archai.nas.evaluater
module
archai.nas.exp_runner
module
archai.nas.finalizers
module
archai.nas.model
module
archai.nas.model_desc
module
archai.nas.model_desc_builder
module
archai.nas.nas_utils
module
archai.nas.operations
module
archai.nas.random_finalizers
module
archai.nas.search_combinations
module
archai.nas.searcher
module
archai.nas.vis_model_desc
module
archai.networks
module
archai.networks.pyramidnet
module
archai.networks.resnet
module
archai.networks.shakedrop
module
archai.networks.shakeshake
module
archai.networks.shakeshake.shake_resnet
module
archai.networks.shakeshake.shake_resnext
module
archai.networks.shakeshake.shakeshake
module
archai.networks.wideresnet
module
ArchModule (class in archai.nas.arch_module)
ArchParams (class in archai.nas.arch_params)
ArchTrainer (class in archai.nas.arch_trainer)
arsaug_policy() (in module archai.datasets.augmentation)
AtomicFileHandler (class in archai.common.atomic_file_handler)
augment_list() (in module archai.datasets.augmentation)
Augmentation (class in archai.datasets.augmentation)
autoaug2arsaug() (in module archai.datasets.augmentation)
autoaug_paper_cifar10() (in module archai.datasets.augmentation)
autoaug_policy() (in module archai.datasets.augmentation)
AutoContrast() (in module archai.datasets.augmentation)
AuxTower (class in archai.nas.model)
AuxTowerDesc (class in archai.nas.model_desc)
AverageMeter (class in archai.common.utils)
AvgPool2d7x7 (class in archai.nas.operations)
B
backward() (archai.common.apex_utils.ApexUtils method)
(archai.networks.shakedrop.ShakeDropFunction static method)
(archai.networks.shakeshake.shakeshake.ShakeShake static method)
barrier() (archai.common.apex_utils.ApexUtils method)
BasicBlock (class in archai.networks.pyramidnet)
(class in archai.networks.resnet)
best_epoch() (archai.common.metrics.RunMetrics method)
best_train_top1() (archai.common.metrics.Metrics method)
best_val_top1() (archai.common.metrics.Metrics method)
BilevelArchTrainer (class in archai.algos.darts.bilevel_arch_trainer)
BilevelOptimizer (class in archai.algos.darts.bilevel_optimizer)
(class in archai.algos.darts.bilevel_optimizer_slow)
Bottleneck (class in archai.networks.pyramidnet)
(class in archai.networks.resnet)
Brightness (class in archai.datasets.transforms.brightness)
Brightness() (in module archai.datasets.augmentation)
build() (archai.nas.model_desc_builder.ModelDescBuilder method)
build_aux_tower() (archai.nas.model_desc_builder.ModelDescBuilder method)
build_cell() (archai.nas.model_desc_builder.ModelDescBuilder method)
build_cell_post_op() (archai.nas.model_desc_builder.ModelDescBuilder method)
build_cell_stems() (archai.nas.model_desc_builder.ModelDescBuilder method)
build_cells() (archai.algos.random.random_model_desc_builder.RandomModelDescBuilder method)
(archai.nas.model_desc_builder.ModelDescBuilder method)
build_logits_op() (archai.nas.model_desc_builder.ModelDescBuilder method)
build_model_desc() (archai.algos.petridish.searcher_petridish.SearcherPetridish method)
(archai.nas.searcher.Searcher method)
build_model_pool() (archai.nas.model_desc_builder.ModelDescBuilder method)
build_model_stems() (archai.nas.model_desc_builder.ModelDescBuilder method)
build_nodes() (archai.algos.darts.darts_model_desc_builder.DartsModelDescBuilder method)
(archai.algos.divnas.divnas_model_desc_builder.DivnasModelDescBuilder method)
(archai.algos.gumbelsoftmax.gs_model_desc_builder.GsModelDescBuilder method)
(archai.algos.random.random_model_desc_builder.RandomModelDescBuilder method)
(archai.algos.xnas.xnas_model_desc_builder.XnasModelDescBuilder method)
(archai.nas.model_desc_builder.ModelDescBuilder method)
build_nodes_from_template() (archai.nas.model_desc_builder.ModelDescBuilder method)
C
can_drop_path() (archai.algos.darts.mixed_op.MixedOp method)
(archai.algos.divnas.divop.DivOp method)
(archai.algos.gumbelsoftmax.gs_op.GsOp method)
(archai.algos.xnas.xnas_op.XnasOp method)
(archai.nas.operations.AvgPool2d7x7 method)
(archai.nas.operations.Identity method)
(archai.nas.operations.LinearOp method)
(archai.nas.operations.MergeOp method)
(archai.nas.operations.Op method)
(archai.nas.operations.PoolAdaptiveAvg2D method)
(archai.nas.operations.SkipConnect method)
(archai.nas.operations.StemConv3x3 method)
(archai.nas.operations.StemConv3x3S4 method)
(archai.nas.operations.StemConv3x3S4S2 method)
Cell (class in archai.nas.cell)
cell_descs() (archai.nas.model_desc.ModelDesc method)
cell_type_count() (archai.nas.model_desc.ModelDesc method)
CellDesc (class in archai.nas.model_desc)
CellType (class in archai.nas.model_desc)
change_filepath_ext() (in module archai.common.utils)
change_filepath_name() (in module archai.common.utils)
channel_norm() (in module archai.common.ml_utils)
CheckPoint (class in archai.common.checkpoint)
checkpoint_empty() (in module archai.nas.nas_utils)
Cifar100Provider (class in archai.datasets.providers.cifar100_provider)
Cifar10Provider (class in archai.datasets.providers.cifar10_provider)
clean_ensure_expdir() (in module archai.common.common)
clean_log_result() (archai.nas.searcher.Searcher method)
clear() (archai.common.stopwatch.StopWatch method)
clear_collect_activations() (archai.algos.divnas.divnas_cell.Divnas_Cell method)
clear_timings() (in module archai.common.timing)
clear_trainables() (archai.nas.model_desc.CellDesc method)
(archai.nas.model_desc.EdgeDesc method)
(archai.nas.model_desc.ModelDesc method)
(archai.nas.model_desc.NodeDesc method)
(archai.nas.model_desc.OpDesc method)
clip_grad() (archai.common.apex_utils.ApexUtils method)
clone() (archai.nas.model_desc.CellDesc method)
(archai.nas.model_desc.ConvMacroParams method)
(archai.nas.model_desc.EdgeDesc method)
(archai.nas.model_desc.ModelDesc method)
(archai.nas.model_desc.NodeDesc method)
(archai.nas.model_desc.OpDesc method)
close() (archai.common.atomic_file_handler.AtomicFileHandler method)
(archai.common.ordereddict_logger.OrderedDictLogger method)
CocobBackprop (class in archai.common.cocob)
CocobOns (class in archai.common.cocob)
collect_activations() (archai.algos.divnas.divnas_cell.Divnas_Cell method)
(archai.algos.divnas.divop.DivOp property)
collect_features() (in module archai.algos.divnas.analyse_activations)
Color() (in module archai.datasets.augmentation)
commit() (archai.common.checkpoint.CheckPoint method)
common_init() (in module archai.common.common)
CommonState (class in archai.common.common)
compute_brute_force_sol() (in module archai.algos.divnas.analyse_activations)
compute_correlation() (in module archai.algos.divnas.analyse_activations)
compute_covariance_offline() (in module archai.algos.divnas.analyse_activations)
compute_euclidean_dist_quantiles() (in module archai.algos.divnas.analyse_activations)
compute_loss() (archai.common.trainer.Trainer method)
(archai.nas.arch_trainer.ArchTrainer method)
compute_marginal_gain() (in module archai.algos.divnas.analyse_activations)
compute_rbf_kernel_covariance() (in module archai.algos.divnas.analyse_activations)
ConcateChannelsOp (class in archai.nas.operations)
Config (class in archai.common.config)
Contrast() (in module archai.datasets.augmentation)
conv3x3() (in module archai.networks.pyramidnet)
(in module archai.networks.resnet)
(in module archai.networks.wideresnet)
conv_init() (in module archai.networks.wideresnet)
ConvBNReLU (class in archai.nas.operations)
ConvexHullPoint (class in archai.algos.petridish.petridish_utils)
ConvMacroParams (class in archai.nas.model_desc)
copy_dir() (in module archai.common.utils)
copy_file() (in module archai.common.utils)
copy_file_basic() (in module archai.common.utils)
copy_search_to_eval() (archai.algos.manual.manual_exp_runner.ManualExperimentRunner method)
(archai.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method)
(archai.nas.exp_runner.ExperimentRunner method)
create() (archai.nas.operations.Op static method)
create_arch_params() (archai.nas.arch_module.ArchModule method)
create_cell_templates() (archai.nas.model_desc_builder.ModelDescBuilder method)
create_checkpoint() (in module archai.nas.nas_utils)
create_conf() (in module archai.common.common)
create_dataset_provider() (in module archai.datasets.data)
create_dirs() (in module archai.common.common)
create_logger() (in module archai.common.common)
(in module archai.common.utils)
create_lr_scheduler() (in module archai.common.ml_utils)
create_model() (archai.algos.manual.manual_evaluater.ManualEvaluater method)
(archai.nas.evaluater.Evaluater method)
create_multi_optim() (archai.common.trainer.Trainer method)
create_optimizer() (archai.algos.gumbelsoftmax.gs_arch_trainer.GsArchTrainer method)
(archai.algos.xnas.xnas_arch_trainer.XnasArchTrainer method)
(archai.common.trainer.Trainer method)
(in module archai.common.ml_utils)
create_scheduler() (archai.common.trainer.Trainer method)
create_submod_f() (in module archai.algos.divnas.analyse_activations)
create_tb_writer() (in module archai.common.common)
cuda_device_names() (in module archai.common.utils)
cur_elapsed() (archai.common.timing.MeasureBlockTime method)
cur_epoch() (archai.common.metrics.Metrics method)
(archai.common.metrics.RunMetrics method)
Cutout() (in module archai.datasets.augmentation)
CutoutAbs() (in module archai.datasets.augmentation)
CutoutCustom (class in archai.datasets.transforms.cutout_custom)
D
DagEdge (class in archai.nas.dag_edge)
DartsExperimentRunner (class in archai.algos.darts.darts_exp_runner)
DartsModelDescBuilder (class in archai.algos.darts.darts_model_desc_builder)
DatasetProvider (class in archai.datasets.dataset_provider)
debug() (archai.common.ordereddict_logger.OrderedDictLogger method)
deep_comp() (in module archai.common.utils)
deep_update() (in module archai.common.config)
(in module archai.common.utils)
device_type() (archai.nas.model.Model method)
DilConv (class in archai.nas.operations)
dir_downloads() (in module archai.common.utils)
DistributedStratifiedSampler (class in archai.datasets.distributed_stratified_sampler)
Divnas_Cell (class in archai.algos.divnas.divnas_cell)
DivnasExperimentRunner (class in archai.algos.divnas.divnas_exp_runner)
DivnasFinalizers (class in archai.algos.divnas.divnas_finalizers)
DivnasModelDescBuilder (class in archai.algos.divnas.divnas_model_desc_builder)
DivnasRankFinalizers (class in archai.algos.divnas.divnas_rank_finalizer)
DivOp (class in archai.algos.divnas.divop)
download() (archai.datasets.providers.imagenet_folder.ImageNetFolder method)
download_and_extract_tar() (in module archai.common.utils)
draw_cell_desc() (in module archai.nas.vis_model_desc)
draw_model_desc() (in module archai.nas.vis_model_desc)
drop_path_() (in module archai.common.ml_utils)
drop_path_prob() (archai.nas.model.Model method)
DropPath_ (class in archai.nas.operations)
duration() (archai.common.metrics.EpochMetrics method)
(archai.common.metrics.RunMetrics method)
E
EdgeDesc (class in archai.nas.model_desc)
emit() (archai.common.atomic_file_handler.AtomicFileHandler method)
empty() (archai.nas.arch_params.ArchParams static method)
ensure_pytorch_ver() (in module archai.common.ml_utils)
epoch() (archai.common.multi_optim.MultiOptim method)
(archai.common.trainer.Trainer method)
epoch_time_avg() (archai.common.metrics.RunMetrics method)
EpochMetrics (class in archai.common.metrics)
epochs() (archai.common.metrics.Metrics method)
Equalize() (in module archai.datasets.augmentation)
EVAL (archai.algos.petridish.petridish_utils.JobStage attribute)
EVAL_TRAINED (archai.algos.petridish.petridish_utils.JobStage attribute)
EvalResult (class in archai.nas.evaluater)
evaluate() (archai.algos.petridish.evaluater_petridish.EvaluaterPetridish method)
(archai.nas.evaluater.Evaluater method)
Evaluater (class in archai.nas.evaluater)
evaluater() (archai.algos.manual.manual_exp_runner.ManualExperimentRunner method)
(archai.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method)
(archai.nas.exp_runner.ExperimentRunner method)
EvaluaterPetridish (class in archai.algos.petridish.evaluater_petridish)
exec_shell_command() (in module archai.common.utils)
expansion (archai.networks.resnet.BasicBlock attribute)
(archai.networks.resnet.Bottleneck attribute)
expdir_abspath() (in module archai.common.common)
ExperimentRunner (class in archai.nas.exp_runner)
extra_repr() (archai.datasets.providers.imagenet_folder.ImageNetFolder method)
(archai.nas.operations.DropPath_ method)
extract_tar() (in module archai.common.utils)
F
fa_reduced_cifar10() (in module archai.datasets.aug_policies)
fa_reduced_svhn() (in module archai.datasets.aug_policies)
fa_resnet50_rimagenet() (in module archai.datasets.aug_policies)
FacConv (class in archai.nas.operations)
FactorizedReduce (class in archai.nas.operations)
FashionMnistProvider (class in archai.datasets.providers.fashion_mnist_provider)
filepath_ext() (in module archai.common.utils)
filepath_name_ext() (in module archai.common.utils)
filepath_name_only() (in module archai.common.utils)
filepath_without_ext() (in module archai.common.utils)
finalize() (archai.algos.darts.mixed_op.MixedOp method)
(archai.algos.divnas.divop.DivOp method)
(archai.algos.gumbelsoftmax.gs_op.GsOp method)
(archai.algos.petridish.petridish_op.PetridishOp method)
(archai.algos.xnas.xnas_op.XnasOp method)
(archai.nas.operations.Op method)
finalize_cell() (archai.algos.divnas.divnas_finalizers.DivnasFinalizers method)
(archai.algos.divnas.divnas_rank_finalizer.DivnasRankFinalizers method)
(archai.nas.finalizers.Finalizers method)
finalize_cells() (archai.nas.finalizers.Finalizers method)
finalize_edge() (archai.nas.finalizers.Finalizers method)
finalize_model() (archai.algos.divnas.divnas_finalizers.DivnasFinalizers method)
(archai.algos.divnas.divnas_rank_finalizer.DivnasRankFinalizers method)
(archai.nas.finalizers.Finalizers method)
(archai.nas.searcher.Searcher method)
finalize_node() (archai.algos.divnas.divnas_finalizers.DivnasFinalizers method)
(archai.algos.divnas.divnas_rank_finalizer.DivnasRankFinalizers method)
(archai.algos.gumbelsoftmax.gs_finalizers.GsFinalizers method)
(archai.nas.finalizers.Finalizers method)
(archai.nas.random_finalizers.RandomFinalizers method)
Finalizers (class in archai.nas.finalizers)
finalizers() (archai.algos.divnas.divnas_exp_runner.DivnasExperimentRunner method)
(archai.algos.gumbelsoftmax.gs_exp_runner.GsExperimentRunner method)
(archai.nas.exp_runner.ExperimentRunner method)
first_or_default() (in module archai.common.utils)
fit() (archai.common.trainer.Trainer method)
Flip() (in module archai.datasets.augmentation)
float_parameter() (in module archai.datasets.augmentation)
Flower102Provider (class in archai.datasets.providers.flower102_provider)
flush() (archai.common.atomic_file_handler.AtomicFileHandler method)
(archai.common.common.SummaryWriterDummy method)
fmt() (in module archai.common.utils)
Food101Provider (class in archai.datasets.providers.food101_provider)
forward() (archai.algos.darts.mixed_op.MixedOp method)
(archai.algos.divnas.divop.DivOp method)
(archai.algos.gumbelsoftmax.gs_op.GsOp method)
(archai.algos.petridish.petridish_op.PetridishOp method)
(archai.algos.petridish.petridish_op.StopForward method)
(archai.algos.petridish.petridish_op.StopForwardReductionOp method)
(archai.algos.petridish.petridish_op.StopGradient method)
(archai.algos.petridish.petridish_op.StopGradientReduction method)
(archai.algos.petridish.petridish_op.TempIdentityOp method)
(archai.algos.xnas.xnas_op.XnasOp method)
(archai.common.ml_losses.LabelSmoothing method)
(archai.common.ml_losses.NLLMultiLabelSmooth method)
(archai.common.ml_losses.SmoothCrossEntropyLoss method)
(archai.nas.cell.Cell method)
(archai.nas.dag_edge.DagEdge method)
(archai.nas.model.AuxTower method)
(archai.nas.model.Model method)
(archai.nas.operations.AvgPool2d7x7 method)
(archai.nas.operations.ConcateChannelsOp method)
(archai.nas.operations.ConvBNReLU method)
(archai.nas.operations.DilConv method)
(archai.nas.operations.DropPath_ method)
(archai.nas.operations.FacConv method)
(archai.nas.operations.FactorizedReduce method)
(archai.nas.operations.Identity method)
(archai.nas.operations.LinearOp method)
(archai.nas.operations.MergeOp method)
(archai.nas.operations.MultiOp method)
(archai.nas.operations.PoolAdaptiveAvg2D method)
(archai.nas.operations.PoolBN method)
(archai.nas.operations.ProjectChannelsOp method)
(archai.nas.operations.ReLUConvBN method)
(archai.nas.operations.SepConv method)
(archai.nas.operations.SkipConnect method)
(archai.nas.operations.StemConv3x3 method)
(archai.nas.operations.StemConv3x3S4 method)
(archai.nas.operations.StemConv3x3S4S2 method)
(archai.nas.operations.Zero method)
(archai.networks.pyramidnet.BasicBlock method)
(archai.networks.pyramidnet.Bottleneck method)
(archai.networks.pyramidnet.PyramidNet method)
(archai.networks.resnet.BasicBlock method)
(archai.networks.resnet.Bottleneck method)
(archai.networks.resnet.ResNet method)
(archai.networks.shakedrop.ShakeDrop method)
(archai.networks.shakedrop.ShakeDropFunction static method)
(archai.networks.shakeshake.shake_resnet.ShakeBlock method)
(archai.networks.shakeshake.shake_resnet.ShakeResNet method)
(archai.networks.shakeshake.shake_resnext.ShakeBottleNeck method)
(archai.networks.shakeshake.shake_resnext.ShakeResNeXt method)
(archai.networks.shakeshake.shakeshake.ShakeShake static method)
(archai.networks.shakeshake.shakeshake.Shortcut method)
(archai.networks.wideresnet.WideBasic method)
(archai.networks.wideresnet.WideResNet method)
from_module() (archai.nas.arch_params.ArchParams static method)
full_path() (in module archai.common.utils)
G
get() (archai.common.stopwatch.StopWatch static method)
get_all_tensor_stats() (in module archai.common.model_summary)
get_all_timings() (in module archai.common.timing)
get_augment() (in module archai.datasets.augmentation)
get_batch() (in module archai.algos.divnas.analyse_activations)
get_cell_template() (archai.nas.model_desc_builder.ModelDescBuilder method)
get_cell_type() (archai.nas.model_desc_builder.ModelDescBuilder method)
get_combinations() (archai.nas.search_combinations.SearchCombinations method)
get_conf() (archai.nas.exp_runner.ExperimentRunner method)
(in module archai.common.common)
get_conf_cell() (archai.nas.model_desc_builder.ModelDescBuilder method)
get_conf_common() (in module archai.common.common)
get_conf_dataset() (archai.nas.model_desc_builder.ModelDescBuilder method)
(in module archai.common.common)
get_conf_model_stems() (archai.nas.model_desc_builder.ModelDescBuilder method)
get_data() (archai.nas.evaluater.Evaluater method)
(archai.nas.searcher.Searcher method)
(in module archai.datasets.data)
get_datadir() (in module archai.common.common)
get_dataloaders() (in module archai.datasets.data)
get_datasets() (archai.datasets.dataset_provider.DatasetProvider method)
(archai.datasets.providers.cifar100_provider.Cifar100Provider method)
(archai.datasets.providers.cifar10_provider.Cifar10Provider method)
(archai.datasets.providers.fashion_mnist_provider.FashionMnistProvider method)
(archai.datasets.providers.flower102_provider.Flower102Provider method)
(archai.datasets.providers.food101_provider.Food101Provider method)
(archai.datasets.providers.imagenet_provider.ImagenetProvider method)
(archai.datasets.providers.mit67_provider.Mit67Provider method)
(archai.datasets.providers.mnist_provider.MnistProvider method)
(archai.datasets.providers.sport8_provider.Sport8Provider method)
(archai.datasets.providers.svhn_provider.SvhnProvider method)
get_device() (archai.common.tester.Tester method)
(archai.common.trainer.Trainer method)
get_dict() (archai.common.metrics.Accumulator method)
get_edge_ranks() (archai.nas.finalizers.Finalizers method)
get_elapsed() (archai.common.stopwatch.StopWatch method)
get_expdir() (in module archai.common.common)
get_experiment_name() (in module archai.common.common)
get_expname() (archai.nas.exp_runner.ExperimentRunner method)
get_inst() (archai.common.config.Config static method)
get_last() (in module archai.common.timing)
get_logger() (in module archai.common.common)
get_lossfn() (in module archai.common.ml_utils)
get_lr() (archai.common.multi_optim.MultiOptim method)
(archai.common.warmup_scheduler.GradualWarmupScheduler method)
get_metrics() (archai.common.tester.Tester method)
(archai.common.trainer.Trainer method)
get_model() (in module archai.networks)
get_model_stats() (in module archai.nas.nas_utils)
get_node_channels() (archai.nas.model_desc_builder.ModelDescBuilder method)
get_node_count() (archai.nas.model_desc_builder.ModelDescBuilder method)
get_optim_lr() (in module archai.common.ml_utils)
get_optimizer() (archai.common.trainer.Trainer method)
get_provider_type() (in module archai.datasets.dataset_provider)
get_reduction_indices() (archai.nas.model_desc_builder.ModelDescBuilder method)
get_scheduler() (archai.common.trainer.Trainer method)
get_shape() (in module archai.common.model_summary)
get_state() (in module archai.common.common)
get_tb_writer() (in module archai.common.common)
get_tensor_stat() (in module archai.common.model_summary)
get_timing() (in module archai.common.timing)
get_trainables() (archai.nas.operations.Op method)
get_trainables_from() (archai.nas.model_desc_builder.ModelDescBuilder method)
get_transforms() (archai.datasets.dataset_provider.DatasetProvider method)
(archai.datasets.providers.cifar100_provider.Cifar100Provider method)
(archai.datasets.providers.cifar10_provider.Cifar10Provider method)
(archai.datasets.providers.fashion_mnist_provider.FashionMnistProvider method)
(archai.datasets.providers.flower102_provider.Flower102Provider method)
(archai.datasets.providers.food101_provider.Food101Provider method)
(archai.datasets.providers.imagenet_provider.ImagenetProvider method)
(archai.datasets.providers.mit67_provider.Mit67Provider method)
(archai.datasets.providers.mnist_provider.MnistProvider method)
(archai.datasets.providers.sport8_provider.Sport8Provider method)
(archai.datasets.providers.svhn_provider.SvhnProvider method)
get_val() (archai.common.config.Config method)
GradualWarmupScheduler (class in archai.common.warmup_scheduler)
greedy_op_selection() (in module archai.algos.divnas.analyse_activations)
GsArchTrainer (class in archai.algos.gumbelsoftmax.gs_arch_trainer)
GsExperimentRunner (class in archai.algos.gumbelsoftmax.gs_exp_runner)
GsFinalizers (class in archai.algos.gumbelsoftmax.gs_finalizers)
GsModelDescBuilder (class in archai.algos.gumbelsoftmax.gs_model_desc_builder)
GsOp (class in archai.algos.gumbelsoftmax.gs_op)
H
has_aux_tower() (archai.nas.model_desc.ModelDesc method)
has_kind() (archai.nas.arch_params.ArchParams method)
has_method() (in module archai.common.utils)
hull_points2tsv() (in module archai.algos.petridish.petridish_utils)
I
Identity (class in archai.nas.operations)
ImageNetFolder (class in archai.datasets.providers.imagenet_folder)
ImagenetProvider (class in archai.datasets.providers.imagenet_provider)
info() (archai.common.ordereddict_logger.OrderedDictLogger method)
init_from() (in module archai.common.common)
int_parameter() (in module archai.datasets.augmentation)
Invert() (in module archai.datasets.augmentation)
is_better_metrics() (archai.nas.search_combinations.SearchCombinations method)
is_debugging() (in module archai.common.utils)
is_dist() (archai.common.apex_utils.ApexUtils method)
(archai.common.metrics.Metrics method)
is_empty() (archai.common.checkpoint.CheckPoint method)
is_main_process() (in module archai.common.utils)
is_master() (archai.common.apex_utils.ApexUtils method)
is_mixed() (archai.common.apex_utils.ApexUtils method)
is_proper_path() (in module archai.common.yaml_utils)
is_ray() (archai.common.apex_utils.ApexUtils method)
is_scaler() (in module archai.common.model_summary)
is_trained_stage() (archai.algos.petridish.petridish_utils.ConvexHullPoint method)
items() (archai.common.metrics.Accumulator method)
J
JobStage (class in archai.algos.petridish.petridish_utils)
join_chunks() (in module archai.common.ml_utils)
K
keys() (archai.common.stopwatch.StopWatch method)
L
LabelSmoothing (class in archai.common.ml_losses)
Lighting (class in archai.datasets.transforms.lighting)
LimitDataset (class in archai.datasets.limit_dataset)
LinearOp (class in archai.nas.operations)
load() (archai.common.ordereddict_logger.OrderedDictLogger method)
(archai.nas.model_desc.ModelDesc static method)
load_existing() (archai.common.checkpoint.CheckPoint method)
load_model() (in module archai.common.ml_utils)
load_state_dict() (archai.algos.darts.bilevel_optimizer.BilevelOptimizer method)
(archai.algos.darts.bilevel_optimizer_slow.BilevelOptimizer method)
(archai.common.apex_utils.ApexUtils method)
(archai.common.metrics.Metrics method)
(archai.common.multi_optim.MultiOptim method)
(archai.common.tester.Tester method)
(archai.nas.model_desc.CellDesc method)
(archai.nas.model_desc.EdgeDesc method)
(archai.nas.model_desc.ModelDesc method)
(archai.nas.model_desc.NodeDesc method)
(archai.nas.model_desc.OpDesc method)
(in module archai.common.utils)
M
main() (in module archai.algos.divnas.analyse_activations)
main_process_pid() (in module archai.common.utils)
ManualEvaluater (class in archai.algos.manual.manual_evaluater)
ManualExperimentRunner (class in archai.algos.manual.manual_exp_runner)
ManualSearcher (class in archai.algos.manual.manual_searcher)
MeasureBlockTime (class in archai.common.timing)
MeasureTime() (in module archai.common.timing)
MergeOp (class in archai.nas.operations)
meta_file() (archai.datasets.providers.imagenet_folder.ImageNetFolder property)
Metrics (class in archai.common.metrics)
Mit67Provider (class in archai.datasets.providers.mit67_provider)
MixedOp (class in archai.algos.darts.mixed_op)
MnistProvider (class in archai.datasets.providers.mnist_provider)
Model (class in archai.nas.model)
model_desc_builder() (archai.algos.darts.darts_exp_runner.DartsExperimentRunner method)
(archai.algos.divnas.divnas_exp_runner.DivnasExperimentRunner method)
(archai.algos.gumbelsoftmax.gs_exp_runner.GsExperimentRunner method)
(archai.algos.manual.manual_exp_runner.ManualExperimentRunner method)
(archai.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method)
(archai.algos.random.random_exp_runner.RandomExperimentRunner method)
(archai.algos.xnas.xnas_exp_runner.XnasExperimentRunner method)
(archai.nas.exp_runner.ExperimentRunner method)
model_descs_on_front() (in module archai.algos.petridish.petridish_utils)
model_from_desc() (archai.nas.evaluater.Evaluater method)
ModelDesc (class in archai.nas.model_desc)
ModelDescBuilder (class in archai.nas.model_desc_builder)
ModelMetrics (class in archai.nas.searcher)
module
archai
archai.algos
archai.algos.darts
archai.algos.darts.bilevel_arch_trainer
archai.algos.darts.bilevel_optimizer
archai.algos.darts.bilevel_optimizer_slow
archai.algos.darts.darts_exp_runner
archai.algos.darts.darts_model_desc_builder
archai.algos.darts.mixed_op
archai.algos.divnas
archai.algos.divnas.analyse_activations
archai.algos.divnas.divnas_cell
archai.algos.divnas.divnas_exp_runner
archai.algos.divnas.divnas_finalizers
archai.algos.divnas.divnas_model_desc_builder
archai.algos.divnas.divnas_rank_finalizer
archai.algos.divnas.divop
archai.algos.divnas.seqopt
archai.algos.divnas.wmr
archai.algos.gumbelsoftmax
archai.algos.gumbelsoftmax.gs_arch_trainer
archai.algos.gumbelsoftmax.gs_exp_runner
archai.algos.gumbelsoftmax.gs_finalizers
archai.algos.gumbelsoftmax.gs_model_desc_builder
archai.algos.gumbelsoftmax.gs_op
archai.algos.manual
archai.algos.manual.manual_evaluater
archai.algos.manual.manual_exp_runner
archai.algos.manual.manual_searcher
archai.algos.petridish
archai.algos.petridish.evaluater_petridish
archai.algos.petridish.petridish_exp_runner
archai.algos.petridish.petridish_model_desc_builder
archai.algos.petridish.petridish_op
archai.algos.petridish.petridish_utils
archai.algos.petridish.searcher_petridish
archai.algos.random
archai.algos.random.random_exp_runner
archai.algos.random.random_model_desc_builder
archai.algos.xnas
archai.algos.xnas.xnas_arch_trainer
archai.algos.xnas.xnas_exp_runner
archai.algos.xnas.xnas_model_desc_builder
archai.algos.xnas.xnas_op
archai.common
archai.common.apex_utils
archai.common.atomic_file_handler
archai.common.checkpoint
archai.common.cocob
archai.common.common
archai.common.config
archai.common.metrics
archai.common.ml_losses
archai.common.ml_utils
archai.common.model_summary
archai.common.multi_optim
archai.common.ordereddict_logger
archai.common.stopwatch
archai.common.tester
archai.common.timing
archai.common.trainer
archai.common.utils
archai.common.warmup_scheduler
archai.common.yaml_utils
archai.datasets
archai.datasets.aug_policies
archai.datasets.augmentation
archai.datasets.data
archai.datasets.dataset_provider
archai.datasets.distributed_stratified_sampler
archai.datasets.limit_dataset
archai.datasets.providers
archai.datasets.providers.cifar100_provider
archai.datasets.providers.cifar10_provider
archai.datasets.providers.fashion_mnist_provider
archai.datasets.providers.flower102_provider
archai.datasets.providers.food101_provider
archai.datasets.providers.imagenet_folder
archai.datasets.providers.imagenet_provider
archai.datasets.providers.mit67_provider
archai.datasets.providers.mnist_provider
archai.datasets.providers.sport8_provider
archai.datasets.providers.svhn_provider
archai.datasets.transforms
archai.datasets.transforms.brightness
archai.datasets.transforms.cutout_custom
archai.datasets.transforms.lighting
archai.nas
archai.nas.arch_module
archai.nas.arch_params
archai.nas.arch_trainer
archai.nas.cell
archai.nas.dag_edge
archai.nas.evaluater
archai.nas.exp_runner
archai.nas.finalizers
archai.nas.model
archai.nas.model_desc
archai.nas.model_desc_builder
archai.nas.nas_utils
archai.nas.operations
archai.nas.random_finalizers
archai.nas.search_combinations
archai.nas.searcher
archai.nas.vis_model_desc
archai.networks
archai.networks.pyramidnet
archai.networks.resnet
archai.networks.shakedrop
archai.networks.shakeshake
archai.networks.shakeshake.shake_resnet
archai.networks.shakeshake.shake_resnext
archai.networks.shakeshake.shakeshake
archai.networks.wideresnet
MultiOp (class in archai.nas.operations)
MultiOptim (class in archai.common.multi_optim)
N
new() (archai.common.checkpoint.CheckPoint method)
next_stage() (archai.algos.petridish.petridish_utils.ConvexHullPoint method)
NLLMultiLabelSmooth (class in archai.common.ml_losses)
no_duplicates() (in module archai.datasets.augmentation)
NodeDesc (class in archai.nas.model_desc)
nodes() (archai.nas.model_desc.CellDesc method)
nonarch_from_module() (archai.nas.arch_params.ArchParams static method)
nonarch_params() (archai.nas.arch_module.ArchModule method)
num_class() (in module archai.networks)
num_primitive_ops() (archai.algos.divnas.divop.DivOp property)
O
on_app_exit() (in module archai.common.common)
Op (class in archai.nas.operations)
op() (archai.nas.dag_edge.DagEdge method)
OpDesc (class in archai.nas.model_desc)
ops() (archai.algos.darts.mixed_op.MixedOp method)
(archai.algos.divnas.divop.DivOp method)
(archai.algos.gumbelsoftmax.gs_op.GsOp method)
(archai.algos.petridish.petridish_op.PetridishOp method)
(archai.nas.cell.Cell method)
(archai.nas.model.Model method)
(archai.nas.operations.Op method)
OptimSched (class in archai.common.multi_optim)
OrderedDictLogger (class in archai.common.ordereddict_logger)
outchannel_ratio (archai.networks.pyramidnet.BasicBlock attribute)
(archai.networks.pyramidnet.Bottleneck attribute)
P
param_by_kind() (archai.nas.arch_params.ArchParams method)
param_size() (in module archai.common.ml_utils)
paramdict_by_kind() (archai.nas.arch_params.ArchParams method)
paramlist_by_kind() (archai.nas.arch_params.ArchParams method)
path() (archai.common.ordereddict_logger.OrderedDictLogger method)
pause() (archai.common.stopwatch.StopWatch method)
PetridishExperimentRunner (class in archai.algos.petridish.petridish_exp_runner)
PetridishModelBuilder (class in archai.algos.petridish.petridish_model_desc_builder)
PetridishOp (class in archai.algos.petridish.petridish_op)
plot_all_covs() (in module archai.algos.divnas.analyse_activations)
plot_frontier() (in module archai.algos.petridish.petridish_utils)
plot_pool() (in module archai.algos.petridish.petridish_utils)
plot_seed_model_stats() (in module archai.algos.petridish.petridish_utils)
policy_decoder() (in module archai.datasets.augmentation)
PoolAdaptiveAvg2D (class in archai.nas.operations)
PoolBN (class in archai.nas.operations)
popd() (archai.common.ordereddict_logger.OrderedDictLogger method)
post_epoch() (archai.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method)
(archai.algos.xnas.xnas_arch_trainer.XnasArchTrainer method)
(archai.common.metrics.EpochMetrics method)
(archai.common.metrics.Metrics method)
(archai.common.trainer.Trainer method)
(archai.nas.arch_trainer.ArchTrainer method)
post_fit() (archai.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method)
(archai.algos.xnas.xnas_arch_trainer.XnasArchTrainer method)
(archai.common.trainer.Trainer method)
post_run() (archai.common.metrics.Metrics method)
(archai.common.metrics.RunMetrics method)
post_step() (archai.common.metrics.EpochMetrics method)
(archai.common.metrics.Metrics method)
(archai.common.trainer.Trainer method)
Posterize() (in module archai.datasets.augmentation)
Posterize2() (in module archai.datasets.augmentation)
pre_build() (archai.algos.darts.darts_model_desc_builder.DartsModelDescBuilder method)
(archai.algos.divnas.divnas_model_desc_builder.DivnasModelDescBuilder method)
(archai.algos.gumbelsoftmax.gs_model_desc_builder.GsModelDescBuilder method)
(archai.algos.petridish.petridish_model_desc_builder.PetridishModelBuilder method)
(archai.algos.xnas.xnas_model_desc_builder.XnasModelDescBuilder method)
(archai.nas.model_desc_builder.ModelDescBuilder method)
pre_epoch() (archai.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method)
(archai.algos.xnas.xnas_arch_trainer.XnasArchTrainer method)
(archai.common.metrics.EpochMetrics method)
(archai.common.metrics.Metrics method)
(archai.common.trainer.Trainer method)
pre_fit() (archai.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method)
(archai.algos.xnas.xnas_arch_trainer.XnasArchTrainer method)
(archai.common.trainer.Trainer method)
pre_run() (archai.common.metrics.Metrics method)
(archai.common.metrics.RunMetrics method)
pre_step() (archai.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method)
(archai.algos.gumbelsoftmax.gs_arch_trainer.GsArchTrainer method)
(archai.algos.xnas.xnas_arch_trainer.XnasArchTrainer method)
(archai.common.metrics.EpochMetrics method)
(archai.common.metrics.Metrics method)
(archai.common.trainer.Trainer method)
PRIMITIVES (archai.algos.darts.mixed_op.MixedOp attribute)
(archai.algos.divnas.divop.DivOp attribute)
(archai.algos.gumbelsoftmax.gs_op.GsOp attribute)
(archai.algos.petridish.petridish_op.PetridishOp attribute)
(archai.algos.random.random_model_desc_builder.RandOps attribute)
(archai.algos.xnas.xnas_op.XnasOp attribute)
print_all_timings() (in module archai.common.timing)
print_timing() (in module archai.common.timing)
process_name() (in module archai.common.utils)
ProjectChannelsOp (class in archai.nas.operations)
pushd() (archai.common.ordereddict_logger.OrderedDictLogger method)
pyramidal_make_layer() (archai.networks.pyramidnet.PyramidNet method)
PyramidNet (class in archai.networks.pyramidnet)
R
RandomBrightness (class in archai.datasets.transforms.brightness)
RandomExperimentRunner (class in archai.algos.random.random_exp_runner)
RandomFinalizers (class in archai.nas.random_finalizers)
RandomModelDescBuilder (class in archai.algos.random.random_model_desc_builder)
RandOps (class in archai.algos.random.random_model_desc_builder)
rbf() (in module archai.algos.divnas.analyse_activations)
read_string() (in module archai.common.utils)
record_checkpoint() (archai.nas.search_combinations.SearchCombinations method)
reduce() (archai.common.apex_utils.ApexUtils method)
reduce_max() (archai.common.metrics.Metrics method)
reduce_mean() (archai.common.metrics.Metrics method)
reduce_min() (archai.common.metrics.Metrics method)
reduce_sum() (archai.common.metrics.Metrics method)
Reduction (archai.nas.model_desc.CellType attribute)
register_dataset_provider() (in module archai.datasets.dataset_provider)
register_op() (archai.nas.operations.Op static method)
Regular (archai.nas.model_desc.CellType attribute)
ReLUConvBN (class in archai.nas.operations)
remove_deplicates() (in module archai.datasets.augmentation)
reset() (archai.common.ordereddict_logger.OrderedDictLogger method)
(archai.common.utils.AverageMeter method)
reset_cells() (archai.nas.model_desc.ModelDesc method)
reset_nodes() (archai.nas.model_desc.CellDesc method)
ResNet (class in archai.networks.resnet)
resolve_all() (in module archai.common.yaml_utils)
restore_checkpoint() (archai.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method)
(archai.common.trainer.Trainer method)
(archai.nas.search_combinations.SearchCombinations method)
Rotate() (in module archai.datasets.augmentation)
run() (archai.nas.exp_runner.ExperimentRunner method)
run_eval() (archai.nas.exp_runner.ExperimentRunner method)
run_search() (archai.nas.exp_runner.ExperimentRunner method)
RunMetrics (class in archai.common.metrics)
S
sample() (archai.algos.divnas.wmr.Wmr method)
sample_from_hull() (in module archai.algos.petridish.petridish_utils)
sample_sequence() (archai.algos.divnas.seqopt.SeqOpt method)
SamplePairing() (in module archai.datasets.augmentation)
save() (archai.common.metrics.Metrics method)
(archai.common.ordereddict_logger.OrderedDictLogger method)
(archai.nas.model_desc.ModelDesc method)
save_hull() (in module archai.algos.petridish.petridish_utils)
save_hull_frontier() (in module archai.algos.petridish.petridish_utils)
save_model() (in module archai.common.ml_utils)
save_trained() (archai.nas.search_combinations.SearchCombinations method)
SEARCH (archai.algos.petridish.petridish_utils.JobStage attribute)
search() (archai.algos.manual.manual_searcher.ManualSearcher method)
(archai.algos.petridish.searcher_petridish.SearcherPetridish method)
(archai.nas.search_combinations.SearchCombinations method)
(archai.nas.searcher.Searcher method)
search_model_desc() (archai.nas.searcher.Searcher method)
search_model_desc_dist (archai.algos.petridish.searcher_petridish.SearcherPetridish attribute)
SEARCH_TRAINED (archai.algos.petridish.petridish_utils.JobStage attribute)
SearchCombinations (class in archai.nas.search_combinations)
Searcher (class in archai.nas.searcher)
searcher() (archai.algos.manual.manual_exp_runner.ManualExperimentRunner method)
(archai.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method)
(archai.nas.exp_runner.ExperimentRunner method)
SearcherPetridish (class in archai.algos.petridish.searcher_petridish)
SearchResult (class in archai.nas.searcher)
SEED (archai.algos.petridish.petridish_utils.JobStage attribute)
seed_cell() (archai.nas.model_desc_builder.ModelDescBuilder method)
SEED_TRAINED (archai.algos.petridish.petridish_utils.JobStage attribute)
select_edges() (archai.nas.finalizers.Finalizers method)
SepConv (class in archai.nas.operations)
SeqOpt (class in archai.algos.divnas.seqopt)
set() (archai.common.stopwatch.StopWatch static method)
set_arch_params() (archai.nas.arch_module.ArchModule method)
set_epoch() (archai.datasets.distributed_stratified_sampler.DistributedStratifiedSampler method)
set_inst() (archai.common.config.Config static method)
set_op_sampled_weights() (archai.algos.gumbelsoftmax.gs_op.GsOp method)
set_optim_lr() (in module archai.common.ml_utils)
set_trainables() (archai.nas.operations.Op method)
setup_cuda() (in module archai.common.utils)
ShakeBlock (class in archai.networks.shakeshake.shake_resnet)
ShakeBottleNeck (class in archai.networks.shakeshake.shake_resnext)
ShakeDrop (class in archai.networks.shakedrop)
ShakeDropFunction (class in archai.networks.shakedrop)
ShakeResNet (class in archai.networks.shakeshake.shake_resnet)
ShakeResNeXt (class in archai.networks.shakeshake.shake_resnext)
ShakeShake (class in archai.networks.shakeshake.shakeshake)
Sharpness() (in module archai.datasets.augmentation)
ShearX() (in module archai.datasets.augmentation)
ShearY() (in module archai.datasets.augmentation)
Shortcut (class in archai.networks.shakeshake.shakeshake)
SkipConnect (class in archai.nas.operations)
SmoothCrossEntropyLoss (class in archai.common.ml_losses)
Solarize() (in module archai.datasets.augmentation)
split_folder() (archai.datasets.providers.imagenet_folder.ImageNetFolder property)
Sport8Provider (class in archai.datasets.providers.sport8_provider)
start() (archai.common.stopwatch.StopWatch method)
state_dict() (archai.algos.darts.bilevel_optimizer.BilevelOptimizer method)
(archai.algos.darts.bilevel_optimizer_slow.BilevelOptimizer method)
(archai.common.apex_utils.ApexUtils method)
(archai.common.metrics.Metrics method)
(archai.common.multi_optim.MultiOptim method)
(archai.common.tester.Tester method)
(archai.nas.model_desc.CellDesc method)
(archai.nas.model_desc.EdgeDesc method)
(archai.nas.model_desc.ModelDesc method)
(archai.nas.model_desc.NodeDesc method)
(archai.nas.model_desc.OpDesc method)
(in module archai.common.utils)
StemBase (class in archai.nas.operations)
StemConv3x3 (class in archai.nas.operations)
StemConv3x3S4 (class in archai.nas.operations)
StemConv3x3S4S2 (class in archai.nas.operations)
step() (archai.algos.darts.bilevel_optimizer.BilevelOptimizer method)
(archai.algos.darts.bilevel_optimizer_slow.BilevelOptimizer method)
(archai.common.cocob.CocobBackprop method)
(archai.common.cocob.CocobOns method)
(archai.common.multi_optim.MultiOptim method)
(archai.common.warmup_scheduler.GradualWarmupScheduler method)
step_ReduceLROnPlateau() (archai.common.warmup_scheduler.GradualWarmupScheduler method)
step_time_avg() (archai.common.metrics.RunMetrics method)
StopForward (class in archai.algos.petridish.petridish_op)
StopForwardReductionOp (class in archai.algos.petridish.petridish_op)
StopGradient (class in archai.algos.petridish.petridish_op)
StopGradientReduction (class in archai.algos.petridish.petridish_op)
stopwatch (archai.common.stopwatch.StopWatch attribute)
StopWatch (class in archai.common.stopwatch)
subscribe() (archai.common.checkpoint.CheckPoint method)
SubsetSampler (class in archai.datasets.data)
summary() (archai.nas.model.Model method)
(in module archai.common.model_summary)
summary_string() (in module archai.common.model_summary)
SummaryWriterDummy (class in archai.common.common)
SvhnProvider (class in archai.datasets.providers.svhn_provider)
sync_devices() (archai.common.apex_utils.ApexUtils method)
T
TempIdentityOp (class in archai.algos.petridish.petridish_op)
tensor() (in module archai.algos.divnas.divnas_cell)
(in module archai.nas.cell)
terminator (archai.common.atomic_file_handler.AtomicFileHandler attribute)
test() (archai.common.tester.Tester method)
Tester (class in archai.common.tester)
to_amp() (archai.common.apex_utils.ApexUtils method)
to_dict() (archai.common.config.Config method)
train_model() (archai.nas.evaluater.Evaluater method)
train_model_desc() (archai.nas.searcher.Searcher method)
train_model_desc_dist (archai.algos.petridish.searcher_petridish.SearcherPetridish attribute)
Trainer (class in archai.common.trainer)
trainer_class() (archai.algos.darts.darts_exp_runner.DartsExperimentRunner method)
(archai.algos.divnas.divnas_exp_runner.DivnasExperimentRunner method)
(archai.algos.gumbelsoftmax.gs_exp_runner.GsExperimentRunner method)
(archai.algos.manual.manual_exp_runner.ManualExperimentRunner method)
(archai.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method)
(archai.algos.random.random_exp_runner.RandomExperimentRunner method)
(archai.algos.xnas.xnas_exp_runner.XnasExperimentRunner method)
(archai.nas.exp_runner.ExperimentRunner method)
TranslateX() (in module archai.datasets.augmentation)
TranslateXAbs() (in module archai.datasets.augmentation)
TranslateY() (in module archai.datasets.augmentation)
TranslateYAbs() (in module archai.datasets.augmentation)
U
update() (archai.algos.divnas.seqopt.SeqOpt method)
(archai.algos.divnas.wmr.Wmr method)
(archai.common.utils.AverageMeter method)
update_alphas() (archai.algos.xnas.xnas_op.XnasOp method)
update_checkpoint() (archai.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method)
(archai.algos.xnas.xnas_arch_trainer.XnasArchTrainer method)
(archai.common.trainer.Trainer method)
update_covs() (archai.algos.divnas.divnas_cell.Divnas_Cell method)
update_envvars() (in module archai.common.common)
V
valid_splits() (archai.datasets.providers.imagenet_folder.ImageNetFolder property)
W
warn() (archai.common.ordereddict_logger.OrderedDictLogger method)
weights() (archai.algos.divnas.wmr.Wmr property)
WideBasic (class in archai.networks.wideresnet)
WideResNet (class in archai.networks.wideresnet)
Wmr (class in archai.algos.divnas.wmr)
write_string() (in module archai.common.utils)
X
XnasArchTrainer (class in archai.algos.xnas.xnas_arch_trainer)
XnasExperimentRunner (class in archai.algos.xnas.xnas_exp_runner)
XnasModelDescBuilder (class in archai.algos.xnas.xnas_model_desc_builder)
XnasOp (class in archai.algos.xnas.xnas_op)
Z
Zero (class in archai.nas.operations)
zero_file() (in module archai.common.utils)
zero_grad() (archai.common.multi_optim.MultiOptim method)
zip_eq() (in module archai.common.utils)