MLBench Core
develop
MLBench
Benchmarks
Prerequisites
Installation
Component Overview
Tutorials
Components
Helm Chart
Dashboard
Benchmark Implementations
Core API
Additional Info
Developer Guide
Contributing
Changelog
MLBench Core
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
I
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
Z
_
__call__() (mlbench_core.evaluation.pytorch.metrics.TopKAccuracy method)
_agg() (mlbench_core.aggregation.pytorch.aggregation.Aggregation method)
_agg_gradients_by_layer() (mlbench_core.aggregation.pytorch.aggregation.Aggregation method)
_agg_gradients_by_model() (mlbench_core.aggregation.pytorch.aggregation.Aggregation method)
_agg_weights_by_layer() (mlbench_core.aggregation.pytorch.aggregation.Aggregation method)
_agg_weights_by_model() (mlbench_core.aggregation.pytorch.aggregation.Aggregation method)
A
add_symbol() (mlbench_core.dataset.nlp.pytorch.wmt17.Dictionary method)
agg_grad() (mlbench_core.aggregation.pytorch.aggregation.Aggregation method)
agg_model() (mlbench_core.aggregation.pytorch.aggregation.Aggregation method)
Aggregation (class in mlbench_core.aggregation.pytorch.aggregation)
AllReduceAggregation (class in mlbench_core.aggregation.pytorch.centralized)
AllReduceAggregationHVD (class in mlbench_core.aggregation.pytorch.centralized)
ApiClient (class in mlbench_core.api)
append_hidden() (mlbench_core.models.pytorch.gnmt.decoder.ResidualRecurrentDecoder method)
B
backward_loss() (mlbench_core.optim.pytorch.fp_optimizers.FP16Optimizer method)
batch_norm() (in module mlbench_core.models.tensorflow.resnet_model)
batch_step() (mlbench_core.lr_scheduler.pytorch.lr.ReduceLROnPlateauWithWarmup method)
BCELossRegularized (class in mlbench_core.evaluation.pytorch.criterion)
beam_search() (mlbench_core.utils.pytorch.inference.beam_search.SequenceGenerator method)
block_layer() (in module mlbench_core.models.tensorflow.resnet_model)
C
CentralizedAdam (class in mlbench_core.optim.pytorch.centralized)
CentralizedSGD (class in mlbench_core.optim.pytorch.centralized)
CentralizedSparsifiedSGD (class in mlbench_core.optim.pytorch.centralized)
Checkpointer (class in mlbench_core.utils.pytorch.checkpoint)
CheckpointsEvaluationControlFlow (class in mlbench_core.controlflow.pytorch)
Cifar10Model (class in mlbench_core.models.tensorflow.resnet_model)
CIFAR10V1 (class in mlbench_core.dataset.imagerecognition.pytorch.dataloader)
Classifier (class in mlbench_core.models.pytorch.gnmt.decoder)
collate_batch() (in module mlbench_core.dataset.nlp.pytorch.wmt17)
config_logging() (in module mlbench_core.utils.pytorch.helpers)
config_path() (in module mlbench_core.utils.pytorch.helpers)
config_pytorch() (in module mlbench_core.utils.pytorch.helpers)
conv2d_fixed_padding() (in module mlbench_core.models.tensorflow.resnet_model)
convert_dtype() (in module mlbench_core.controlflow.pytorch.helpers)
create_run() (mlbench_core.api.ApiClient method)
CustomCentralizedOptimizer (class in mlbench_core.optim.pytorch.centralized)
D
DatasetCifar (class in mlbench_core.dataset.imagerecognition.tensorflow)
DecentralizedAggregation (class in mlbench_core.aggregation.pytorch.decentralized)
DecentralizedSGD (class in mlbench_core.optim.pytorch.decentralized)
decode() (mlbench_core.models.pytorch.gnmt.GNMT method)
delete_run() (mlbench_core.api.ApiClient method)
detokenize() (mlbench_core.dataset.nlp.pytorch.wmt16.wmt16_tokenizer.WMT16Tokenizer method)
Dictionary (class in mlbench_core.dataset.nlp.pytorch.wmt17)
download_run_metrics() (mlbench_core.api.ApiClient method)
E
encode() (mlbench_core.models.pytorch.gnmt.GNMT method)
eos() (mlbench_core.dataset.nlp.pytorch.wmt17.Dictionary method)
evaluate_by_epochs() (mlbench_core.controlflow.pytorch.CheckpointsEvaluationControlFlow method)
ExponentialWarmupMultiStepLR (class in mlbench_core.lr_scheduler.pytorch.lr)
F
FCGraph (class in mlbench_core.utils.pytorch)
fixed_padding() (in module mlbench_core.models.tensorflow.resnet_model)
forward() (mlbench_core.models.pytorch.gnmt.decoder.Classifier method)
(mlbench_core.models.pytorch.gnmt.decoder.RecurrentAttention method)
(mlbench_core.models.pytorch.gnmt.decoder.ResidualRecurrentDecoder method)
(mlbench_core.models.pytorch.gnmt.encoder.ResidualRecurrentEncoder method)
(mlbench_core.models.pytorch.transformer.encoder.TransformerEncoder method)
(mlbench_core.models.pytorch.transformer.TransformerModel method)
fp16_to_fp32_flat_grad() (mlbench_core.optim.pytorch.fp_optimizers.FP16Optimizer static method)
FP16Optimizer (class in mlbench_core.optim.pytorch.fp_optimizers)
fp32_to_fp16_weights() (mlbench_core.optim.pytorch.fp_optimizers.FP16Optimizer static method)
G
generate() (mlbench_core.models.pytorch.gnmt.GNMT method)
(mlbench_core.models.pytorch.transformer.sequence_generator.SequenceGenerator method)
generate_batch_translations() (mlbench_core.models.pytorch.transformer.sequence_generator.SequenceGenerator method)
GenericCentralizedOptimizer (class in mlbench_core.optim.pytorch.centralized)
get_all_metrics() (mlbench_core.api.ApiClient method)
get_estimated_weights() (mlbench_core.optim.pytorch.optim.SparsifiedSGD method)
get_pod_metrics() (mlbench_core.api.ApiClient method)
get_run() (mlbench_core.api.ApiClient method)
get_run_metrics() (mlbench_core.api.ApiClient method)
get_runs() (mlbench_core.api.ApiClient method)
get_worker_pods() (mlbench_core.api.ApiClient method)
GNMT (class in mlbench_core.models.pytorch.gnmt)
greedy_search() (mlbench_core.utils.pytorch.inference.beam_search.SequenceGenerator method)
I
Imagenet (class in mlbench_core.dataset.imagerecognition.pytorch.dataloader)
index() (mlbench_core.dataset.nlp.pytorch.wmt17.Dictionary method)
init_hidden() (mlbench_core.models.pytorch.gnmt.decoder.ResidualRecurrentDecoder method)
initialize_backends() (in module mlbench_core.utils.pytorch)
initialize_flat_fp32_weight() (mlbench_core.optim.pytorch.fp_optimizers.FP16Optimizer method)
input_fn() (mlbench_core.dataset.imagerecognition.tensorflow.DatasetCifar method)
iterate_dataloader() (in module mlbench_core.controlflow.pytorch.helpers)
L
LabelSmoothing (class in mlbench_core.evaluation.pytorch.criterion)
LinearRegression (class in mlbench_core.models.pytorch.linear_models)
LMDBDataset (class in mlbench_core.dataset.linearmodels.pytorch.dataloader)
LMDBPTClass (class in mlbench_core.dataset.linearmodels.pytorch.dataloader)
load() (mlbench_core.dataset.nlp.pytorch.wmt17.Dictionary class method)
LogisticRegression (class in mlbench_core.models.pytorch.linear_models)
LRLinearWarmUp (class in mlbench_core.lr_scheduler.pytorch.lr)
M
manual_stepping() (in module mlbench_core.lr_scheduler.tensorflow)
maybe_download_and_extract() (mlbench_core.dataset.imagerecognition.tensorflow.DatasetCifar method)
maybe_range() (in module mlbench_core.controlflow.pytorch.helpers)
mlbench_core.aggregation
module
mlbench_core.aggregation.pytorch
module
mlbench_core.aggregation.pytorch.aggregation
module
mlbench_core.aggregation.pytorch.centralized
module
mlbench_core.aggregation.pytorch.decentralized
module
mlbench_core.api
module
mlbench_core.controlflow
module
mlbench_core.controlflow.pytorch
module
mlbench_core.controlflow.pytorch.helpers
module
mlbench_core.controlflow.tensorflow
module
mlbench_core.dataset
module
mlbench_core.dataset.imagerecognition
module
mlbench_core.dataset.imagerecognition.pytorch.dataloader
module
mlbench_core.dataset.imagerecognition.tensorflow
module
mlbench_core.dataset.linearmodels
module
mlbench_core.dataset.linearmodels.pytorch.dataloader
module
mlbench_core.dataset.nlp
module
mlbench_core.dataset.nlp.pytorch
module
mlbench_core.dataset.nlp.pytorch.wmt16.wmt16_tokenizer
module
mlbench_core.dataset.nlp.pytorch.wmt17
module
mlbench_core.evaluation
module
mlbench_core.evaluation.pytorch
module
mlbench_core.evaluation.pytorch.criterion
module
mlbench_core.evaluation.pytorch.metrics
module
mlbench_core.evaluation.tensorflow.criterion
module
mlbench_core.evaluation.tensorflow.metrics
module
mlbench_core.lr_scheduler
module
mlbench_core.lr_scheduler.pytorch.lr
module
mlbench_core.lr_scheduler.tensorflow
module
mlbench_core.models
module
mlbench_core.models.pytorch
module
mlbench_core.models.pytorch.gnmt
module
mlbench_core.models.pytorch.gnmt.decoder
module
mlbench_core.models.pytorch.gnmt.encoder
module
mlbench_core.models.pytorch.linear_models
module
mlbench_core.models.pytorch.resnet
module
mlbench_core.models.pytorch.transformer
module
mlbench_core.models.pytorch.transformer.decoder
module
mlbench_core.models.pytorch.transformer.encoder
module
mlbench_core.models.pytorch.transformer.modules
module
mlbench_core.models.pytorch.transformer.sequence_generator
module
mlbench_core.models.tensorflow
module
mlbench_core.models.tensorflow.resnet_model
module
mlbench_core.optim
module
mlbench_core.optim.pytorch
module
mlbench_core.optim.pytorch.centralized
module
mlbench_core.optim.pytorch.decentralized
module
mlbench_core.optim.pytorch.fp_optimizers
module
mlbench_core.optim.pytorch.optim
module
mlbench_core.utils
module
mlbench_core.utils.pytorch
module
mlbench_core.utils.pytorch.checkpoint
module
mlbench_core.utils.pytorch.helpers
module
mlbench_core.utils.pytorch.inference
module
mlbench_core.utils.pytorch.inference.beam_search
module
mlbench_core.utils.pytorch.utils
module
mlbench_core.utils.tensorflow
module
MLBENCH_IMAGES (in module mlbench_core.api)
Model (class in mlbench_core.models.tensorflow.resnet_model)
module
mlbench_core.aggregation
mlbench_core.aggregation.pytorch
mlbench_core.aggregation.pytorch.aggregation
mlbench_core.aggregation.pytorch.centralized
mlbench_core.aggregation.pytorch.decentralized
mlbench_core.api
mlbench_core.controlflow
mlbench_core.controlflow.pytorch
mlbench_core.controlflow.pytorch.helpers
mlbench_core.controlflow.tensorflow
mlbench_core.dataset
mlbench_core.dataset.imagerecognition
mlbench_core.dataset.imagerecognition.pytorch.dataloader
mlbench_core.dataset.imagerecognition.tensorflow
mlbench_core.dataset.linearmodels
mlbench_core.dataset.linearmodels.pytorch.dataloader
mlbench_core.dataset.nlp
mlbench_core.dataset.nlp.pytorch
mlbench_core.dataset.nlp.pytorch.wmt16.wmt16_tokenizer
mlbench_core.dataset.nlp.pytorch.wmt17
mlbench_core.evaluation
mlbench_core.evaluation.pytorch
mlbench_core.evaluation.pytorch.criterion
mlbench_core.evaluation.pytorch.metrics
mlbench_core.evaluation.tensorflow.criterion
mlbench_core.evaluation.tensorflow.metrics
mlbench_core.lr_scheduler
mlbench_core.lr_scheduler.pytorch.lr
mlbench_core.lr_scheduler.tensorflow
mlbench_core.models
mlbench_core.models.pytorch
mlbench_core.models.pytorch.gnmt
mlbench_core.models.pytorch.gnmt.decoder
mlbench_core.models.pytorch.gnmt.encoder
mlbench_core.models.pytorch.linear_models
mlbench_core.models.pytorch.resnet
mlbench_core.models.pytorch.transformer
mlbench_core.models.pytorch.transformer.decoder
mlbench_core.models.pytorch.transformer.encoder
mlbench_core.models.pytorch.transformer.modules
mlbench_core.models.pytorch.transformer.sequence_generator
mlbench_core.models.tensorflow
mlbench_core.models.tensorflow.resnet_model
mlbench_core.optim
mlbench_core.optim.pytorch
mlbench_core.optim.pytorch.centralized
mlbench_core.optim.pytorch.decentralized
mlbench_core.optim.pytorch.fp_optimizers
mlbench_core.optim.pytorch.optim
mlbench_core.utils
mlbench_core.utils.pytorch
mlbench_core.utils.pytorch.checkpoint
mlbench_core.utils.pytorch.helpers
mlbench_core.utils.pytorch.inference
mlbench_core.utils.pytorch.inference.beam_search
mlbench_core.utils.pytorch.utils
mlbench_core.utils.tensorflow
MSELossRegularized (class in mlbench_core.evaluation.pytorch.criterion)
MultiStepLRLinearWarmUp (class in mlbench_core.lr_scheduler.pytorch.lr)
N
name() (mlbench_core.evaluation.pytorch.metrics.TopKAccuracy property)
O
orthogonalize() (in module mlbench_core.utils.pytorch.utils)
P
pack_tensors() (in module mlbench_core.utils.pytorch.utils)
package_hidden() (mlbench_core.models.pytorch.gnmt.decoder.ResidualRecurrentDecoder method)
pad() (mlbench_core.dataset.nlp.pytorch.wmt17.Dictionary method)
parse_record() (mlbench_core.dataset.imagerecognition.tensorflow.DatasetCifar method)
post_metric() (mlbench_core.api.ApiClient method)
PowerAggregation (class in mlbench_core.aggregation.pytorch.centralized)
PowerSGD (class in mlbench_core.optim.pytorch.centralized)
prepare_batch() (in module mlbench_core.controlflow.pytorch.helpers)
preprocess_image() (mlbench_core.dataset.imagerecognition.tensorflow.DatasetCifar method)
R
record_dataset() (mlbench_core.dataset.imagerecognition.tensorflow.DatasetCifar method)
record_train_batch_stats() (in module mlbench_core.controlflow.pytorch)
record_validation_stats() (in module mlbench_core.controlflow.pytorch)
RecurrentAttention (class in mlbench_core.models.pytorch.gnmt.decoder)
ReduceLROnPlateauWithWarmup (class in mlbench_core.lr_scheduler.pytorch.lr)
ResidualRecurrentDecoder (class in mlbench_core.models.pytorch.gnmt.decoder)
ResidualRecurrentEncoder (class in mlbench_core.models.pytorch.gnmt.encoder)
ResNetCIFAR (class in mlbench_core.models.pytorch.resnet)
S
segment() (mlbench_core.dataset.nlp.pytorch.wmt16.wmt16_tokenizer.WMT16Tokenizer method)
SequenceGenerator (class in mlbench_core.models.pytorch.transformer.sequence_generator)
(class in mlbench_core.utils.pytorch.inference.beam_search)
SignSGD (class in mlbench_core.optim.pytorch.optim)
softmax_cross_entropy_with_logits_v2_l2_regularized() (in module mlbench_core.evaluation.tensorflow.criterion)
SparsifiedAggregation (class in mlbench_core.aggregation.pytorch.centralized)
SparsifiedSGD (class in mlbench_core.optim.pytorch.optim)
SparsifiedSGDLR (class in mlbench_core.lr_scheduler.pytorch.lr)
sparsify_gradients() (mlbench_core.optim.pytorch.optim.SparsifiedSGD method)
SQRTTimeDecayLR (class in mlbench_core.lr_scheduler.pytorch.lr)
SQRTTimeDecayLRWithWarmup (class in mlbench_core.lr_scheduler.pytorch.lr)
step() (mlbench_core.lr_scheduler.pytorch.lr.ReduceLROnPlateauWithWarmup method)
(mlbench_core.optim.pytorch.centralized.CentralizedSparsifiedSGD method)
(mlbench_core.optim.pytorch.centralized.CustomCentralizedOptimizer method)
(mlbench_core.optim.pytorch.centralized.GenericCentralizedOptimizer method)
(mlbench_core.optim.pytorch.centralized.PowerSGD method)
(mlbench_core.optim.pytorch.decentralized.DecentralizedSGD method)
(mlbench_core.optim.pytorch.fp_optimizers.FP16Optimizer method)
(mlbench_core.optim.pytorch.optim.SignSGD method)
(mlbench_core.optim.pytorch.optim.SparsifiedSGD method)
string() (mlbench_core.dataset.nlp.pytorch.wmt17.Dictionary method)
T
TimeDecayLR (class in mlbench_core.lr_scheduler.pytorch.lr)
topk_accuracy_with_logits() (in module mlbench_core.evaluation.tensorflow.metrics)
TopKAccuracy (class in mlbench_core.evaluation.pytorch.metrics)
train_and_eval() (mlbench_core.controlflow.tensorflow.TrainValidation method)
train_one_epoch() (mlbench_core.controlflow.tensorflow.TrainValidation method)
TrainValidation (class in mlbench_core.controlflow.tensorflow)
TransformerDecoder (class in mlbench_core.models.pytorch.transformer.decoder)
TransformerDecoderLayer (class in mlbench_core.models.pytorch.transformer.modules)
TransformerEncoder (class in mlbench_core.models.pytorch.transformer.encoder)
TransformerEncoderLayer (class in mlbench_core.models.pytorch.transformer.modules)
TransformerModel (class in mlbench_core.models.pytorch.transformer)
translate_batch() (mlbench_core.models.pytorch.transformer.sequence_generator.SequenceGenerator method)
U
unpack_tensors() (in module mlbench_core.utils.pytorch.utils)
update() (mlbench_core.dataset.nlp.pytorch.wmt17.Dictionary method)
update_estimated_weights() (mlbench_core.optim.pytorch.optim.SparsifiedSGD method)
V
valid_one_epoch() (mlbench_core.controlflow.tensorflow.TrainValidation method)
validation_round() (in module mlbench_core.controlflow.pytorch)
W
WMT16Dataset (class in mlbench_core.dataset.nlp.pytorch)
WMT16Tokenizer (class in mlbench_core.dataset.nlp.pytorch.wmt16.wmt16_tokenizer)
Z
zero_grad() (mlbench_core.optim.pytorch.fp_optimizers.FP16Optimizer method)
Read the Docs
v: develop
Versions
latest
stable
develop
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds