AIToolbox
latest
Components:
torchtrain
Train Loop
TTModel
Callbacks
Schedulers
Multi-Loss and Multi-Optimizer
Multi-GPU Training
Automatic Mixed Precision Training
Advanced Topics
Message Passing Service
Model Prediction Store
Model Wrap and Batch Feed Definition
experiment
Result Package
Result Metric
Experiment Saving
Training History
cloud
Saving to Cloud
Loading Models from Cloud
Data Access
AWS Simple Email Service
nlp
Examples:
Examples
API:
aitoolbox
torchtrain
callbacks
abstract
basic
ddp
gradient
model_load
model_save
performance_eval
tensorboard
train_schedule
wandb
data
batch_model_feed_defs
dataset
schedulers
basic
warmup
train_loop
components
callback_handler
ddp_handler
message_passing
model_prediction_store
pred_collate_fns
train_loop
train_loop_tracking
model
model_predict
multi_loss_optim
parallel
experiment
core_metrics
abstract_metric
classification
regression
local_load
local_model_load
local_save
folder_create
local_model_save
local_results_save
result_package
abstract_result_packages
basic_packages
hf_evaluate_packages
torch_metrics_packages
result_reporting
hyperparam_reporter
report_generator
experiment_saver
local_experiment_saver
training_history
cloud
AWS
data_access
model_load
model_save
results_save
simple_email_service
GoogleCloud
data_access
model_load
model_save
results_save
nlp
core
core
vocabulary
experiment_evaluation
NLP_metrics
NLP_result_package
attention_heatmap
torch_collate_fns
utils
dict_util
file_system
util
AIToolbox
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Z
_
__add__() (aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric method)
(aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
__call__() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
__contains__() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
__iadd__() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
__len__() (aitoolbox.experiment.result_package.abstract_result_packages.MultipleResultPackageWrapper method)
__radd__() (aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric method)
(aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
_backward_pass() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
_calculate_batch_loss() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
_create_other_object_pkg() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage static method)
_get_data() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
_get_metric_self_other_val() (aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric method)
_has_data() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
_insert_data() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
_merge_dicts() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
_optimizer_step() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
_optimizer_zero_grad() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
_print_save_loss() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
_spawn_fit() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
_train() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
_train_ddp() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
_train_dp() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
A
AbstractBaseMetric (class in aitoolbox.experiment.core_metrics.abstract_metric)
AbstractCallback (class in aitoolbox.torchtrain.callbacks.abstract)
AbstractDatasetFetcher (class in aitoolbox.cloud.AWS.data_access)
AbstractExperimentCallback (class in aitoolbox.torchtrain.callbacks.abstract)
AbstractExperimentSaver (class in aitoolbox.experiment.experiment_saver)
AbstractLocalModelLoader (class in aitoolbox.experiment.local_load.local_model_load)
AbstractLocalModelSaver (class in aitoolbox.experiment.local_save.local_model_save)
AbstractLocalResultsSaver (class in aitoolbox.experiment.local_save.local_results_save)
AbstractModelFeedDefinition (class in aitoolbox.torchtrain.data.batch_model_feed_defs)
AbstractModelSaver (class in aitoolbox.cloud.AWS.model_save)
AbstractResultPackage (class in aitoolbox.experiment.result_package.abstract_result_packages)
AbstractResultsSaver (class in aitoolbox.cloud.AWS.results_save)
AbstractScheduler (class in aitoolbox.torchtrain.schedulers.basic)
AccuracyMetric (class in aitoolbox.experiment.core_metrics.classification)
add_distributed_samplers() (aitoolbox.torchtrain.train_loop.components.ddp_handler.DDPHandler method)
add_history_dict() (aitoolbox.experiment.training_history.TrainingHistory method)
add_merge_dicts() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
add_merge_multi_pkg_wrap() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.experiment.result_package.abstract_result_packages.MultipleResultPackageWrapper method)
add_sentence() (aitoolbox.nlp.core.vocabulary.Vocabulary method)
add_word() (aitoolbox.nlp.core.vocabulary.Vocabulary method)
aitoolbox
module
,
[1]
aitoolbox.cloud
module
aitoolbox.cloud.AWS
module
aitoolbox.cloud.AWS.data_access
module
aitoolbox.cloud.AWS.model_load
module
aitoolbox.cloud.AWS.model_save
module
aitoolbox.cloud.AWS.results_save
module
aitoolbox.cloud.AWS.simple_email_service
module
aitoolbox.cloud.GoogleCloud
module
aitoolbox.cloud.GoogleCloud.data_access
module
aitoolbox.cloud.GoogleCloud.model_load
module
aitoolbox.cloud.GoogleCloud.model_save
module
aitoolbox.cloud.GoogleCloud.results_save
module
aitoolbox.experiment
module
aitoolbox.experiment.core_metrics
module
aitoolbox.experiment.core_metrics.abstract_metric
module
aitoolbox.experiment.core_metrics.classification
module
aitoolbox.experiment.core_metrics.regression
module
aitoolbox.experiment.experiment_saver
module
aitoolbox.experiment.local_experiment_saver
module
aitoolbox.experiment.local_load
module
aitoolbox.experiment.local_load.local_model_load
module
aitoolbox.experiment.local_save
module
aitoolbox.experiment.local_save.folder_create
module
aitoolbox.experiment.local_save.local_model_save
module
aitoolbox.experiment.local_save.local_results_save
module
aitoolbox.experiment.result_package
module
aitoolbox.experiment.result_package.abstract_result_packages
module
aitoolbox.experiment.result_package.basic_packages
module
aitoolbox.experiment.result_package.hf_evaluate_packages
module
aitoolbox.experiment.result_package.torch_metrics_packages
module
aitoolbox.experiment.result_reporting
module
aitoolbox.experiment.result_reporting.hyperparam_reporter
module
aitoolbox.experiment.result_reporting.report_generator
module
aitoolbox.experiment.training_history
module
aitoolbox.nlp
module
aitoolbox.nlp.core
module
aitoolbox.nlp.core.core
module
aitoolbox.nlp.core.vocabulary
module
aitoolbox.nlp.experiment_evaluation
module
aitoolbox.nlp.experiment_evaluation.attention_heatmap
module
aitoolbox.nlp.experiment_evaluation.NLP_metrics
module
aitoolbox.nlp.experiment_evaluation.NLP_result_package
module
aitoolbox.nlp.torch_collate_fns
module
aitoolbox.torchtrain
module
aitoolbox.torchtrain.callbacks
module
aitoolbox.torchtrain.callbacks.abstract
module
aitoolbox.torchtrain.callbacks.basic
module
aitoolbox.torchtrain.callbacks.ddp
module
aitoolbox.torchtrain.callbacks.gradient
module
aitoolbox.torchtrain.callbacks.model_load
module
aitoolbox.torchtrain.callbacks.model_save
module
aitoolbox.torchtrain.callbacks.performance_eval
module
aitoolbox.torchtrain.callbacks.tensorboard
module
aitoolbox.torchtrain.callbacks.train_schedule
module
aitoolbox.torchtrain.callbacks.wandb
module
aitoolbox.torchtrain.data
module
aitoolbox.torchtrain.data.batch_model_feed_defs
module
aitoolbox.torchtrain.data.dataset
module
aitoolbox.torchtrain.model
module
aitoolbox.torchtrain.model_predict
module
aitoolbox.torchtrain.multi_loss_optim
module
aitoolbox.torchtrain.parallel
module
aitoolbox.torchtrain.schedulers
module
aitoolbox.torchtrain.schedulers.basic
module
aitoolbox.torchtrain.schedulers.warmup
module
aitoolbox.torchtrain.train_loop
module
aitoolbox.torchtrain.train_loop.components
module
aitoolbox.torchtrain.train_loop.components.callback_handler
module
aitoolbox.torchtrain.train_loop.components.ddp_handler
module
aitoolbox.torchtrain.train_loop.components.message_passing
module
aitoolbox.torchtrain.train_loop.components.model_prediction_store
module
aitoolbox.torchtrain.train_loop.components.pred_collate_fns
module
aitoolbox.torchtrain.train_loop.train_loop
module
aitoolbox.torchtrain.train_loop.train_loop_tracking
module
aitoolbox.utils
module
aitoolbox.utils.dict_util
module
aitoolbox.utils.file_system
module
aitoolbox.utils.util
module
AlertConfig (class in aitoolbox.torchtrain.callbacks.wandb)
AllPredictionsSame (class in aitoolbox.torchtrain.callbacks.basic)
append_concat_predictions() (in module aitoolbox.torchtrain.train_loop.components.pred_collate_fns)
append_predictions() (in module aitoolbox.torchtrain.train_loop.components.pred_collate_fns)
AttentionHeatMap (class in aitoolbox.nlp.experiment_evaluation.attention_heatmap)
auto_execute_end_of_epoch() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
auto_execute_end_of_training() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
auto_purge() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
auto_y_input_array_convert() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage static method)
B
backward() (aitoolbox.torchtrain.multi_loss_optim.MultiLoss method)
BaseDataLoader (class in aitoolbox.cloud.AWS.data_access)
BaseDataSaver (class in aitoolbox.cloud.AWS.data_access)
BaseFullExperimentGoogleStorageSaver (class in aitoolbox.experiment.experiment_saver)
BaseFullExperimentLocalSaver (class in aitoolbox.experiment.local_experiment_saver)
BaseFullExperimentS3Saver (class in aitoolbox.experiment.experiment_saver)
BaseFullExperimentSaver (class in aitoolbox.experiment.experiment_saver)
BaseGoogleStorageDataLoader (class in aitoolbox.cloud.GoogleCloud.data_access)
BaseGoogleStorageDataSaver (class in aitoolbox.cloud.GoogleCloud.data_access)
BaseLocalModelSaver (class in aitoolbox.experiment.local_save.local_model_save)
BaseLocalResultsSaver (class in aitoolbox.experiment.local_save.local_results_save)
BaseModelGoogleStorageLoader (class in aitoolbox.cloud.GoogleCloud.model_load)
BaseModelGoogleStorageSaver (class in aitoolbox.cloud.GoogleCloud.model_save)
BaseModelLoader (class in aitoolbox.cloud.AWS.model_load)
BaseModelSaver (class in aitoolbox.cloud.AWS.model_save)
BaseResultsGoogleStorageSaver (class in aitoolbox.cloud.GoogleCloud.results_save)
BaseResultsSaver (class in aitoolbox.cloud.AWS.results_save)
BasicDataset (class in aitoolbox.torchtrain.data.dataset)
BinaryClassificationResultPackage (class in aitoolbox.experiment.result_package.basic_packages)
BLEUCorpusScoreMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
BLEUScoreStrTorchNLPMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
BLEUSentenceScoreMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
build_loader_sampler() (aitoolbox.torchtrain.train_loop.components.ddp_handler.DDPHandler static method)
build_test_dataloader() (aitoolbox.torchtrain.callbacks.ddp.InMultiProcessDataLoad method)
build_train_dataloader() (aitoolbox.torchtrain.callbacks.ddp.InMultiProcessDataLoad method)
build_val_dataloader() (aitoolbox.torchtrain.callbacks.ddp.InMultiProcessDataLoad method)
C
calculate_metric() (aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric method)
(aitoolbox.experiment.core_metrics.classification.AccuracyMetric method)
(aitoolbox.experiment.core_metrics.classification.F1ScoreMetric method)
(aitoolbox.experiment.core_metrics.classification.PrecisionMetric method)
(aitoolbox.experiment.core_metrics.classification.PrecisionRecallCurveAUCMetric method)
(aitoolbox.experiment.core_metrics.classification.RecallMetric method)
(aitoolbox.experiment.core_metrics.classification.ROCAUCMetric method)
(aitoolbox.experiment.core_metrics.regression.MeanAbsoluteErrorMetric method)
(aitoolbox.experiment.core_metrics.regression.MeanSquaredErrorMetric method)
(aitoolbox.nlp.experiment_evaluation.attention_heatmap.AttentionHeatMap method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.BLEUCorpusScoreMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.BLEUScoreStrTorchNLPMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.BLEUSentenceScoreMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.ExactMatchTextMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.F1TextMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.GLUEMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.PerplexityMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.ROUGEMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.ROUGEPerlMetric method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.XNLIMetric method)
CallbacksHandler (class in aitoolbox.torchtrain.train_loop.components.callback_handler)
check_alerts() (aitoolbox.torchtrain.callbacks.wandb.WandBTracking method)
check_if_history_updated() (aitoolbox.torchtrain.callbacks.performance_eval.TrainHistoryFormatter method)
check_if_model_loaded() (aitoolbox.experiment.local_load.local_model_load.PyTorchLocalModelLoader method)
check_if_result_packages_possible() (aitoolbox.torchtrain.train_loop.train_loop_tracking.TrainLoopEndSave method)
check_model_dict_contents() (aitoolbox.experiment.local_save.local_model_save.PyTorchLocalModelSaver static method)
check_result_packages() (aitoolbox.torchtrain.callbacks.model_save.ModelTrainEndSave method)
check_transl_sent_num_match() (aitoolbox.nlp.experiment_evaluation.NLP_metrics.BLEUSentenceScoreMetric static method)
ClassificationResultPackage (class in aitoolbox.experiment.result_package.basic_packages)
combine_dict_elements() (in module aitoolbox.utils.dict_util)
combine_prediction_metadata_batches() (in module aitoolbox.utils.dict_util)
compute_f1() (aitoolbox.nlp.experiment_evaluation.NLP_metrics.F1TextMetric static method)
concat_metric() (aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric method)
ConstantWithWarmupScheduler (class in aitoolbox.torchtrain.schedulers.warmup)
convert_idx_sent2sent() (aitoolbox.nlp.core.vocabulary.Vocabulary method)
convert_loss_to_float_dict_format() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop static method)
convert_sent2idx_sent() (aitoolbox.nlp.core.vocabulary.Vocabulary method)
copy_function() (in module aitoolbox.utils.util)
copy_to_cloud_storage() (aitoolbox.experiment.result_reporting.hyperparam_reporter.HyperParamSourceReporter method)
CosineWithWarmupScheduler (class in aitoolbox.torchtrain.schedulers.warmup)
cpu() (aitoolbox.torchtrain.multi_loss_optim.MultiLoss method)
create_base_folder() (aitoolbox.experiment.local_save.folder_create.ExperimentFolder static method)
create_experiment_cloud_storage_folder_structure() (aitoolbox.cloud.AWS.model_save.BaseModelSaver method)
(aitoolbox.cloud.AWS.results_save.BaseResultsSaver method)
create_experiment_local_folder_structure() (aitoolbox.experiment.local_save.local_results_save.BaseLocalResultsSaver method)
create_experiment_local_models_folder() (aitoolbox.experiment.local_save.local_model_save.BaseLocalModelSaver method)
create_experiment_local_results_folder() (aitoolbox.experiment.local_save.local_results_save.BaseLocalResultsSaver static method)
create_folder_hierarchy() (in module aitoolbox.utils.file_system)
create_log_dir() (aitoolbox.torchtrain.callbacks.tensorboard.TensorboardReporterBaseCB method)
create_plot_dirs() (aitoolbox.torchtrain.callbacks.gradient.GradDistributionPlot method)
cuda() (aitoolbox.torchtrain.multi_loss_optim.MultiLoss method)
D
DataSubsetTestRun (class in aitoolbox.torchtrain.callbacks.basic)
DDPHandler (class in aitoolbox.torchtrain.train_loop.components.ddp_handler)
decide_if_remove_suboptimal_model() (aitoolbox.experiment.local_save.local_model_save.LocalSubOptimalModelRemover method)
detach() (aitoolbox.torchtrain.multi_loss_optim.MultiLoss method)
device (aitoolbox.torchtrain.multi_loss_optim.MultiLoss property)
DistributedSamplerSetEpoch (class in aitoolbox.torchtrain.callbacks.ddp)
dump_answer_text_to_disk() (aitoolbox.nlp.experiment_evaluation.NLP_metrics.ROUGEMetric static method)
(aitoolbox.nlp.experiment_evaluation.NLP_metrics.ROUGEPerlMetric static method)
dump_translation_text_to_disk() (aitoolbox.nlp.experiment_evaluation.NLP_metrics.BLEUSentenceScoreMetric static method)
E
EarlyStopping (class in aitoolbox.torchtrain.callbacks.basic)
EmailNotification (class in aitoolbox.torchtrain.callbacks.basic)
end_of_epoch_trigger() (aitoolbox.torchtrain.train_loop.components.message_passing.MessageService method)
enforce_callbacks_quality() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
evaluate_loss_on_test_set() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
evaluate_loss_on_train_set() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
evaluate_loss_on_validation_set() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
evaluate_metric() (aitoolbox.torchtrain.model_predict.PyTorchModelPredictor method)
evaluate_metric_list() (aitoolbox.torchtrain.model_predict.PyTorchModelPredictor method)
evaluate_model() (aitoolbox.torchtrain.model_predict.PyTorchModelPredictor method)
evaluate_model_loss() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
evaluate_model_performance() (aitoolbox.torchtrain.callbacks.performance_eval.ModelPerformanceEvaluation method)
evaluate_result_package() (aitoolbox.torchtrain.model_predict.PyTorchModelPredictor method)
ExactMatchTextMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
execute_after_batch_prediction() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_batch_begin() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_batch_end() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_batch_end_callbacks() (aitoolbox.torchtrain.model_predict.PyTorchModelPredictor method)
execute_callback() (aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
execute_epoch_begin() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_epoch_end() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_epoch_end_callbacks() (aitoolbox.torchtrain.model_predict.PyTorchModelPredictor method)
execute_gradient_update() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_multiprocess_start() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_optimizer_step() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_train_begin() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
execute_train_end() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
exists_local_data_folder() (aitoolbox.cloud.AWS.data_access.BaseDataLoader method)
ExperimentFolder (class in aitoolbox.experiment.local_save.folder_create)
F
F1ScoreMetric (class in aitoolbox.experiment.core_metrics.classification)
F1TextMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
fetch_dataset() (aitoolbox.cloud.AWS.data_access.AbstractDatasetFetcher method)
find_sub_list() (in module aitoolbox.nlp.core.core)
fit() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
flatten_combine_dict() (in module aitoolbox.utils.dict_util)
flatten_dict() (in module aitoolbox.utils.dict_util)
flatten_list_of_lists() (in module aitoolbox.utils.util)
format_history() (aitoolbox.torchtrain.callbacks.performance_eval.TrainHistoryFormatter method)
forward() (aitoolbox.torchtrain.model.MultiGPUModelWrap method)
FullKerasExperimentGoogleStorageSaver (class in aitoolbox.experiment.experiment_saver)
FullKerasExperimentLocalSaver (class in aitoolbox.experiment.local_experiment_saver)
FullKerasExperimentS3Saver (class in aitoolbox.experiment.experiment_saver)
FullPyTorchExperimentGoogleStorageSaver (class in aitoolbox.experiment.experiment_saver)
FullPyTorchExperimentLocalSaver (class in aitoolbox.experiment.local_experiment_saver)
FullPyTorchExperimentS3Saver (class in aitoolbox.experiment.experiment_saver)
function_exists() (in module aitoolbox.utils.util)
FunctionOnTrainLoop (class in aitoolbox.torchtrain.callbacks.basic)
G
GeneralLRSchedulerCallback (class in aitoolbox.torchtrain.schedulers.basic)
GeneralResultPackage (class in aitoolbox.experiment.result_package.basic_packages)
generate_dist_plots() (aitoolbox.experiment.result_reporting.report_generator.GradientPlotter static method)
generate_plots() (aitoolbox.experiment.result_reporting.report_generator.TrainingHistoryPlotter static method)
generate_report() (aitoolbox.experiment.result_reporting.report_generator.GradientPlotter method)
(aitoolbox.experiment.result_reporting.report_generator.TrainingHistoryPlotter method)
(aitoolbox.experiment.result_reporting.report_generator.TrainingHistoryWriter method)
get_additional_results_dump_paths() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.experiment.result_package.abstract_result_packages.MultipleResultPackageWrapper method)
get_base_folder_paths() (aitoolbox.experiment.local_save.folder_create.ExperimentFolder static method)
get_experiment_local_results_folder_paths() (aitoolbox.experiment.local_save.local_results_save.BaseLocalResultsSaver static method)
get_hyperparameters() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
get_hyperparams_html() (aitoolbox.torchtrain.callbacks.basic.EmailNotification method)
get_loss() (aitoolbox.torchtrain.data.batch_model_feed_defs.AbstractModelFeedDefinition method)
(aitoolbox.torchtrain.model.TTBasicModel method)
(aitoolbox.torchtrain.model.TTBasicMultiGPUModel method)
(aitoolbox.torchtrain.model.TTModel method)
(aitoolbox.torchtrain.parallel.TTParallelBase method)
get_loss_dict() (aitoolbox.torchtrain.multi_loss_optim.MultiLoss method)
get_loss_eval() (aitoolbox.torchtrain.data.batch_model_feed_defs.AbstractModelFeedDefinition method)
(aitoolbox.torchtrain.model.TTModel method)
(aitoolbox.torchtrain.parallel.TTParallelBase method)
get_metric() (aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric method)
get_metric_dict() (aitoolbox.experiment.core_metrics.abstract_metric.AbstractBaseMetric method)
get_metric_list_html() (aitoolbox.torchtrain.callbacks.basic.EmailNotification method)
get_num_training_steps() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
get_predictions() (aitoolbox.torchtrain.data.batch_model_feed_defs.AbstractModelFeedDefinition method)
(aitoolbox.torchtrain.model.TTBasicModel method)
(aitoolbox.torchtrain.model.TTModel method)
(aitoolbox.torchtrain.parallel.TTParallelBase method)
get_results() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
get_schedulers() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
get_test_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
get_test_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
get_tokens() (aitoolbox.nlp.experiment_evaluation.NLP_metrics.F1TextMetric static method)
get_train_history() (aitoolbox.experiment.training_history.TrainingHistory method)
get_train_history_dict() (aitoolbox.experiment.training_history.TrainingHistory method)
get_train_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
get_train_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
get_val_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
get_val_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
GLUEMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
GLUEResultPackage (class in aitoolbox.nlp.experiment_evaluation.NLP_result_package)
GoogleStorageResultsSaver (class in aitoolbox.cloud.GoogleCloud.results_save)
GradDistributionPlot (class in aitoolbox.torchtrain.callbacks.gradient)
gradient_plot() (aitoolbox.torchtrain.callbacks.gradient.GradDistributionPlot method)
GradientCallbackBase (class in aitoolbox.torchtrain.callbacks.gradient)
GradientPlotter (class in aitoolbox.experiment.result_reporting.report_generator)
gradients_report() (aitoolbox.torchtrain.callbacks.gradient.GradientStatsPrint method)
GradientStatsPrint (class in aitoolbox.torchtrain.callbacks.gradient)
GradNormClip (class in aitoolbox.torchtrain.callbacks.gradient)
GradValueClip (class in aitoolbox.torchtrain.callbacks.gradient)
H
HardRestartsCosineWithWarmupScheduler (class in aitoolbox.torchtrain.schedulers.warmup)
has_test_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
has_test_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
has_train_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
has_train_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
has_val_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
has_val_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
HFEvaluateResultPackage (class in aitoolbox.experiment.result_package.hf_evaluate_packages)
HyperParamSourceReporter (class in aitoolbox.experiment.result_reporting.hyperparam_reporter)
I
init_amp() (aitoolbox.cloud.AWS.model_load.PyTorchS3ModelLoader method)
(aitoolbox.experiment.local_load.local_model_load.PyTorchLocalModelLoader method)
init_model() (aitoolbox.cloud.AWS.model_load.PyTorchS3ModelLoader method)
(aitoolbox.experiment.local_load.local_model_load.PyTorchLocalModelLoader method)
init_model_loader() (aitoolbox.torchtrain.callbacks.model_load.ModelLoadContinueTraining method)
init_optimizer() (aitoolbox.cloud.AWS.model_load.PyTorchS3ModelLoader method)
(aitoolbox.experiment.local_load.local_model_load.PyTorchLocalModelLoader method)
init_scheduler() (aitoolbox.cloud.AWS.model_load.PyTorchS3ModelLoader method)
(aitoolbox.experiment.local_load.local_model_load.PyTorchLocalModelLoader method)
InMultiProcessDataLoad (class in aitoolbox.torchtrain.callbacks.ddp)
insert_metric_result_into_history() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
insert_single_result_into_history() (aitoolbox.experiment.training_history.TrainingHistory method)
insert_test_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
insert_test_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
insert_train_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
insert_train_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
insert_val_loss() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
insert_val_predictions() (aitoolbox.torchtrain.train_loop.components.model_prediction_store.ModelPredictionStore method)
is_empty_function() (in module aitoolbox.utils.util)
is_main_process() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
item() (aitoolbox.torchtrain.multi_loss_optim.MultiLoss method)
items() (aitoolbox.experiment.training_history.TrainingHistory method)
K
KEEP_FOREVER (aitoolbox.torchtrain.train_loop.components.message_passing.MessageHandling attribute)
keep_list_transf() (in module aitoolbox.torchtrain.train_loop.components.pred_collate_fns)
KerasGoogleStorageModelSaver (class in aitoolbox.cloud.GoogleCloud.model_save)
KerasLocalModelSaver (class in aitoolbox.experiment.local_save.local_model_save)
KerasS3ModelSaver (class in aitoolbox.cloud.AWS.model_save)
keys() (aitoolbox.experiment.training_history.TrainingHistory method)
L
LambdaLRScheduler (class in aitoolbox.torchtrain.schedulers.basic)
LinearWithWarmupScheduler (class in aitoolbox.torchtrain.schedulers.warmup)
list_additional_results_dump_paths() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.MachineTranslationResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.QuestionAnswerResultPackage method)
ListDataset (class in aitoolbox.torchtrain.data.dataset)
ListRegisteredCallbacks (class in aitoolbox.torchtrain.callbacks.basic)
load_file() (aitoolbox.cloud.AWS.data_access.BaseDataLoader method)
(aitoolbox.cloud.GoogleCloud.data_access.BaseGoogleStorageDataLoader method)
load_model() (aitoolbox.cloud.AWS.model_load.BaseModelLoader method)
(aitoolbox.experiment.local_load.local_model_load.AbstractLocalModelLoader method)
(aitoolbox.experiment.local_load.local_model_load.PyTorchLocalModelLoader method)
load_model_from_path() (aitoolbox.experiment.local_load.local_model_load.PyTorchLocalModelLoader method)
load_state_dict() (aitoolbox.torchtrain.multi_loss_optim.MultiOptimizer method)
(aitoolbox.torchtrain.schedulers.basic.AbstractScheduler method)
LocalResultsSaver (class in aitoolbox.experiment.local_save.local_results_save)
LocalSubOptimalModelRemover (class in aitoolbox.experiment.local_save.local_model_save)
log_mid_train_loss() (aitoolbox.torchtrain.callbacks.tensorboard.TensorboardReporterBaseCB method)
(aitoolbox.torchtrain.callbacks.wandb.WandBTracking method)
log_train_history_metrics() (aitoolbox.torchtrain.callbacks.tensorboard.TensorboardReporterBaseCB method)
(aitoolbox.torchtrain.callbacks.wandb.WandBTracking method)
LogUpload (class in aitoolbox.torchtrain.callbacks.basic)
M
MachineTranslationResultPackage (class in aitoolbox.nlp.experiment_evaluation.NLP_result_package)
MeanAbsoluteErrorMetric (class in aitoolbox.experiment.core_metrics.regression)
MeanSquaredErrorMetric (class in aitoolbox.experiment.core_metrics.regression)
Message (class in aitoolbox.torchtrain.train_loop.components.message_passing)
MessageHandling (class in aitoolbox.torchtrain.train_loop.components.message_passing)
MessageService (class in aitoolbox.torchtrain.train_loop.components.message_passing)
metric_compute() (aitoolbox.experiment.result_package.torch_metrics_packages.TorchMetricsPackage method)
metric_name (aitoolbox.torchtrain.callbacks.wandb.AlertConfig attribute)
metric_reset() (aitoolbox.experiment.result_package.torch_metrics_packages.TorchMetricsPackage method)
MetricHistoryRename (class in aitoolbox.torchtrain.callbacks.performance_eval)
model_get_loss() (aitoolbox.torchtrain.model_predict.PyTorchModelPredictor method)
model_predict() (aitoolbox.torchtrain.model_predict.PyTorchModelPredictor method)
ModelCheckpoint (class in aitoolbox.torchtrain.callbacks.model_save)
ModelIterationCheckpoint (class in aitoolbox.torchtrain.callbacks.model_save)
ModelLoadContinueTraining (class in aitoolbox.torchtrain.callbacks.model_load)
ModelPerformanceEvaluation (class in aitoolbox.torchtrain.callbacks.performance_eval)
ModelPerformancePrintReport (class in aitoolbox.torchtrain.callbacks.performance_eval)
ModelPredictionStore (class in aitoolbox.torchtrain.train_loop.components.model_prediction_store)
ModelTrainEndSave (class in aitoolbox.torchtrain.callbacks.model_save)
ModelTrainHistoryBaseCB (class in aitoolbox.torchtrain.callbacks.performance_eval)
ModelTrainHistoryFileWriter (class in aitoolbox.torchtrain.callbacks.performance_eval)
ModelTrainHistoryPlot (class in aitoolbox.torchtrain.callbacks.performance_eval)
ModelWrap (class in aitoolbox.torchtrain.model)
module
aitoolbox
,
[1]
aitoolbox.cloud
aitoolbox.cloud.AWS
aitoolbox.cloud.AWS.data_access
aitoolbox.cloud.AWS.model_load
aitoolbox.cloud.AWS.model_save
aitoolbox.cloud.AWS.results_save
aitoolbox.cloud.AWS.simple_email_service
aitoolbox.cloud.GoogleCloud
aitoolbox.cloud.GoogleCloud.data_access
aitoolbox.cloud.GoogleCloud.model_load
aitoolbox.cloud.GoogleCloud.model_save
aitoolbox.cloud.GoogleCloud.results_save
aitoolbox.experiment
aitoolbox.experiment.core_metrics
aitoolbox.experiment.core_metrics.abstract_metric
aitoolbox.experiment.core_metrics.classification
aitoolbox.experiment.core_metrics.regression
aitoolbox.experiment.experiment_saver
aitoolbox.experiment.local_experiment_saver
aitoolbox.experiment.local_load
aitoolbox.experiment.local_load.local_model_load
aitoolbox.experiment.local_save
aitoolbox.experiment.local_save.folder_create
aitoolbox.experiment.local_save.local_model_save
aitoolbox.experiment.local_save.local_results_save
aitoolbox.experiment.result_package
aitoolbox.experiment.result_package.abstract_result_packages
aitoolbox.experiment.result_package.basic_packages
aitoolbox.experiment.result_package.hf_evaluate_packages
aitoolbox.experiment.result_package.torch_metrics_packages
aitoolbox.experiment.result_reporting
aitoolbox.experiment.result_reporting.hyperparam_reporter
aitoolbox.experiment.result_reporting.report_generator
aitoolbox.experiment.training_history
aitoolbox.nlp
aitoolbox.nlp.core
aitoolbox.nlp.core.core
aitoolbox.nlp.core.vocabulary
aitoolbox.nlp.experiment_evaluation
aitoolbox.nlp.experiment_evaluation.attention_heatmap
aitoolbox.nlp.experiment_evaluation.NLP_metrics
aitoolbox.nlp.experiment_evaluation.NLP_result_package
aitoolbox.nlp.torch_collate_fns
aitoolbox.torchtrain
aitoolbox.torchtrain.callbacks
aitoolbox.torchtrain.callbacks.abstract
aitoolbox.torchtrain.callbacks.basic
aitoolbox.torchtrain.callbacks.ddp
aitoolbox.torchtrain.callbacks.gradient
aitoolbox.torchtrain.callbacks.model_load
aitoolbox.torchtrain.callbacks.model_save
aitoolbox.torchtrain.callbacks.performance_eval
aitoolbox.torchtrain.callbacks.tensorboard
aitoolbox.torchtrain.callbacks.train_schedule
aitoolbox.torchtrain.callbacks.wandb
aitoolbox.torchtrain.data
aitoolbox.torchtrain.data.batch_model_feed_defs
aitoolbox.torchtrain.data.dataset
aitoolbox.torchtrain.model
aitoolbox.torchtrain.model_predict
aitoolbox.torchtrain.multi_loss_optim
aitoolbox.torchtrain.parallel
aitoolbox.torchtrain.schedulers
aitoolbox.torchtrain.schedulers.basic
aitoolbox.torchtrain.schedulers.warmup
aitoolbox.torchtrain.train_loop
aitoolbox.torchtrain.train_loop.components
aitoolbox.torchtrain.train_loop.components.callback_handler
aitoolbox.torchtrain.train_loop.components.ddp_handler
aitoolbox.torchtrain.train_loop.components.message_passing
aitoolbox.torchtrain.train_loop.components.model_prediction_store
aitoolbox.torchtrain.train_loop.components.pred_collate_fns
aitoolbox.torchtrain.train_loop.train_loop
aitoolbox.torchtrain.train_loop.train_loop_tracking
aitoolbox.utils
aitoolbox.utils.dict_util
aitoolbox.utils.file_system
aitoolbox.utils.util
mp_filter_callbacks() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
mp_sync() (aitoolbox.torchtrain.train_loop.components.ddp_handler.DDPHandler method)
mp_sync_dict() (aitoolbox.torchtrain.train_loop.components.ddp_handler.DDPHandler method)
MultiGPUModelWrap (class in aitoolbox.torchtrain.model)
MultiLoss (class in aitoolbox.torchtrain.multi_loss_optim)
MultiOptimizer (class in aitoolbox.torchtrain.multi_loss_optim)
MultipleResultPackageWrapper (class in aitoolbox.experiment.result_package.abstract_result_packages)
MultiStepLRScheduler (class in aitoolbox.torchtrain.schedulers.basic)
N
normalize_answer() (aitoolbox.nlp.experiment_evaluation.NLP_metrics.ExactMatchTextMetric static method)
normalize_string() (in module aitoolbox.nlp.core.core)
numpy() (aitoolbox.torchtrain.multi_loss_optim.MultiLoss method)
O
objective (aitoolbox.torchtrain.callbacks.wandb.AlertConfig attribute)
on_after_batch_prediction() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
on_after_gradient_update() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
(aitoolbox.torchtrain.callbacks.gradient.GradientStatsPrint method)
(aitoolbox.torchtrain.callbacks.gradient.GradNormClip method)
(aitoolbox.torchtrain.callbacks.gradient.GradValueClip method)
on_after_optimizer_step() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
on_batch_begin() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
on_batch_end() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
(aitoolbox.torchtrain.callbacks.model_save.ModelIterationCheckpoint method)
(aitoolbox.torchtrain.callbacks.tensorboard.TensorboardFullTracking method)
(aitoolbox.torchtrain.callbacks.tensorboard.TensorboardTrainBatchLoss method)
(aitoolbox.torchtrain.callbacks.wandb.WandBTracking method)
(aitoolbox.torchtrain.schedulers.basic.LambdaLRScheduler method)
on_epoch_begin() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
(aitoolbox.torchtrain.callbacks.ddp.DistributedSamplerSetEpoch method)
on_epoch_end() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.AllPredictionsSame method)
(aitoolbox.torchtrain.callbacks.basic.EarlyStopping method)
(aitoolbox.torchtrain.callbacks.basic.EmailNotification method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
(aitoolbox.torchtrain.callbacks.basic.LogUpload method)
(aitoolbox.torchtrain.callbacks.basic.TerminateOnNaN method)
(aitoolbox.torchtrain.callbacks.basic.ThresholdEarlyStopping method)
(aitoolbox.torchtrain.callbacks.gradient.GradDistributionPlot method)
(aitoolbox.torchtrain.callbacks.model_save.ModelCheckpoint method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelPerformanceEvaluation method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelPerformancePrintReport method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryFileWriter method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryPlot method)
(aitoolbox.torchtrain.callbacks.performance_eval.TrainHistoryFormatter method)
(aitoolbox.torchtrain.callbacks.tensorboard.TensorboardFullTracking method)
(aitoolbox.torchtrain.callbacks.tensorboard.TensorboardTrainBatchLoss method)
(aitoolbox.torchtrain.callbacks.tensorboard.TensorboardTrainHistoryMetric method)
(aitoolbox.torchtrain.callbacks.wandb.WandBTracking method)
(aitoolbox.torchtrain.schedulers.basic.GeneralLRSchedulerCallback method)
(aitoolbox.torchtrain.schedulers.basic.LambdaLRScheduler method)
(aitoolbox.torchtrain.schedulers.basic.ReduceLROnPlateauMetricScheduler method)
(aitoolbox.torchtrain.schedulers.basic.ReduceLROnPlateauScheduler method)
on_multiprocess_start() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.ddp.InMultiProcessDataLoad method)
on_train_begin() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.DataSubsetTestRun method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
(aitoolbox.torchtrain.callbacks.basic.ListRegisteredCallbacks method)
(aitoolbox.torchtrain.callbacks.model_load.ModelLoadContinueTraining method)
on_train_end() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.EmailNotification method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
(aitoolbox.torchtrain.callbacks.basic.LogUpload method)
(aitoolbox.torchtrain.callbacks.model_save.ModelTrainEndSave method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelPerformanceEvaluation method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelPerformancePrintReport method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryFileWriter method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryPlot method)
(aitoolbox.torchtrain.callbacks.performance_eval.TrainHistoryFormatter method)
(aitoolbox.torchtrain.callbacks.tensorboard.TensorboardReporterBaseCB method)
on_train_loop_registration() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.callbacks.basic.DataSubsetTestRun method)
(aitoolbox.torchtrain.callbacks.basic.EmailNotification method)
(aitoolbox.torchtrain.callbacks.basic.FunctionOnTrainLoop method)
(aitoolbox.torchtrain.callbacks.basic.LogUpload method)
(aitoolbox.torchtrain.callbacks.gradient.GradDistributionPlot method)
(aitoolbox.torchtrain.callbacks.gradient.GradientCallbackBase method)
(aitoolbox.torchtrain.callbacks.model_load.ModelLoadContinueTraining method)
(aitoolbox.torchtrain.callbacks.model_save.ModelCheckpoint method)
(aitoolbox.torchtrain.callbacks.model_save.ModelTrainEndSave method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelPerformanceEvaluation method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryFileWriter method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryPlot method)
(aitoolbox.torchtrain.callbacks.tensorboard.TensorboardReporterBaseCB method)
(aitoolbox.torchtrain.callbacks.wandb.WandBTracking method)
OVERWRITE (aitoolbox.torchtrain.train_loop.components.message_passing.MessageHandling attribute)
P
parse_loss() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
PerplexityMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
plot_current_train_history() (aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryPlot method)
plot_gradient_distribution() (aitoolbox.experiment.result_reporting.report_generator.GradientPlotter static method)
plot_pdf() (aitoolbox.experiment.result_reporting.report_generator.GradientPlotter method)
(aitoolbox.experiment.result_reporting.report_generator.TrainingHistoryPlotter method)
plot_performance_curve() (aitoolbox.experiment.result_reporting.report_generator.TrainingHistoryPlotter static method)
plot_png() (aitoolbox.experiment.result_reporting.report_generator.GradientPlotter method)
(aitoolbox.experiment.result_reporting.report_generator.TrainingHistoryPlotter method)
plot_sentence_attention() (aitoolbox.nlp.experiment_evaluation.attention_heatmap.AttentionHeatMap static method)
PreCalculatedResultPackage (class in aitoolbox.experiment.result_package.abstract_result_packages)
PrecisionMetric (class in aitoolbox.experiment.core_metrics.classification)
PrecisionRecallCurveAUCMetric (class in aitoolbox.experiment.core_metrics.classification)
predict_on_test_set() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
predict_on_train_set() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
predict_on_validation_set() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
predict_with_model() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
prepare_folder_for_saving() (aitoolbox.nlp.experiment_evaluation.attention_heatmap.AttentionHeatMap static method)
prepare_result_package() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.experiment.result_package.abstract_result_packages.MultipleResultPackageWrapper method)
prepare_results_dict() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.experiment.result_package.abstract_result_packages.MultipleResultPackageWrapper method)
(aitoolbox.experiment.result_package.abstract_result_packages.PreCalculatedResultPackage method)
(aitoolbox.experiment.result_package.basic_packages.BinaryClassificationResultPackage method)
(aitoolbox.experiment.result_package.basic_packages.ClassificationResultPackage method)
(aitoolbox.experiment.result_package.basic_packages.GeneralResultPackage method)
(aitoolbox.experiment.result_package.basic_packages.RegressionResultPackage method)
(aitoolbox.experiment.result_package.hf_evaluate_packages.HFEvaluateResultPackage method)
(aitoolbox.experiment.result_package.torch_metrics_packages.TorchMetricsPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.GLUEResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.MachineTranslationResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.QuestionAnswerResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.QuestionAnswerSpanClassificationResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.TextSummarizationResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.XNLIResultPackage method)
prepare_results_saver() (aitoolbox.torchtrain.callbacks.gradient.GradDistributionPlot method)
(aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryBaseCB method)
(aitoolbox.torchtrain.callbacks.tensorboard.TensorboardReporterBaseCB method)
prepare_text() (aitoolbox.nlp.experiment_evaluation.NLP_metrics.ROUGEMetric method)
preproc_dataset_available() (aitoolbox.cloud.AWS.data_access.BaseDataLoader method)
print_callback_info() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler static method)
print_performance_report() (aitoolbox.torchtrain.callbacks.performance_eval.ModelPerformancePrintReport method)
print_registered_callback_names() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
PyTorchGoogleStorageModelLoader (class in aitoolbox.cloud.GoogleCloud.model_load)
PyTorchGoogleStorageModelSaver (class in aitoolbox.cloud.GoogleCloud.model_save)
PyTorchLocalModelLoader (class in aitoolbox.experiment.local_load.local_model_load)
PyTorchLocalModelSaver (class in aitoolbox.experiment.local_save.local_model_save)
PyTorchModelPredictor (class in aitoolbox.torchtrain.model_predict)
PyTorchS3ModelLoader (class in aitoolbox.cloud.AWS.model_load)
PyTorchS3ModelSaver (class in aitoolbox.cloud.AWS.model_save)
Q
qa_check_additional_results_dump_paths() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
qa_check_history_records() (aitoolbox.experiment.training_history.TrainingHistory method)
qa_check_hyperparameters_dict() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
qa_check_metrics_list() (aitoolbox.experiment.result_package.basic_packages.GeneralResultPackage method)
qa_concat_ctx_span_collate_fn() (in module aitoolbox.nlp.torch_collate_fns)
QuestionAnswerResultPackage (class in aitoolbox.nlp.experiment_evaluation.NLP_result_package)
QuestionAnswerSpanClassificationResultPackage (class in aitoolbox.nlp.experiment_evaluation.NLP_result_package)
R
read_messages() (aitoolbox.torchtrain.train_loop.components.message_passing.MessageService method)
RecallMetric (class in aitoolbox.experiment.core_metrics.classification)
ReduceLROnPlateauMetricScheduler (class in aitoolbox.torchtrain.schedulers.basic)
ReduceLROnPlateauScheduler (class in aitoolbox.torchtrain.schedulers.basic)
regex_clean_text() (aitoolbox.nlp.experiment_evaluation.NLP_metrics.ROUGEPerlMetric static method)
register_callbacks() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
register_train_loop_object() (aitoolbox.torchtrain.callbacks.abstract.AbstractCallback method)
(aitoolbox.torchtrain.schedulers.basic.GeneralLRSchedulerCallback method)
RegressionResultPackage (class in aitoolbox.experiment.result_package.basic_packages)
rm_suboptimal_model() (aitoolbox.experiment.local_save.local_model_save.LocalSubOptimalModelRemover static method)
ROCAUCMetric (class in aitoolbox.experiment.core_metrics.classification)
ROUGEMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
ROUGEPerlMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
S
S3ResultsSaver (class in aitoolbox.cloud.AWS.results_save)
save_experiment() (aitoolbox.experiment.experiment_saver.AbstractExperimentSaver method)
(aitoolbox.experiment.experiment_saver.BaseFullExperimentSaver method)
(aitoolbox.experiment.local_experiment_saver.BaseFullExperimentLocalSaver method)
save_experiment_python_file() (aitoolbox.experiment.result_reporting.hyperparam_reporter.HyperParamSourceReporter method)
save_experiment_results() (aitoolbox.cloud.AWS.results_save.AbstractResultsSaver method)
(aitoolbox.cloud.AWS.results_save.S3ResultsSaver method)
(aitoolbox.experiment.local_save.local_results_save.AbstractLocalResultsSaver method)
(aitoolbox.experiment.local_save.local_results_save.LocalResultsSaver method)
save_experiment_results_separate_files() (aitoolbox.experiment.local_save.local_results_save.AbstractLocalResultsSaver method)
(aitoolbox.experiment.local_save.local_results_save.LocalResultsSaver method)
save_experiment_source_files() (aitoolbox.experiment.result_reporting.hyperparam_reporter.HyperParamSourceReporter method)
save_file() (aitoolbox.cloud.AWS.data_access.BaseDataSaver method)
(aitoolbox.cloud.GoogleCloud.data_access.BaseGoogleStorageDataSaver method)
(aitoolbox.experiment.local_save.local_results_save.BaseLocalResultsSaver method)
save_folder() (aitoolbox.cloud.AWS.data_access.BaseDataSaver method)
save_hyperparams() (aitoolbox.torchtrain.callbacks.model_save.ModelCheckpoint method)
(aitoolbox.torchtrain.callbacks.model_save.ModelTrainEndSave method)
save_hyperparams_to_text_file() (aitoolbox.experiment.result_reporting.hyperparam_reporter.HyperParamSourceReporter method)
save_model() (aitoolbox.cloud.AWS.model_save.AbstractModelSaver method)
(aitoolbox.cloud.AWS.model_save.KerasS3ModelSaver method)
(aitoolbox.cloud.AWS.model_save.PyTorchS3ModelSaver method)
(aitoolbox.experiment.local_save.local_model_save.AbstractLocalModelSaver method)
(aitoolbox.experiment.local_save.local_model_save.KerasLocalModelSaver method)
(aitoolbox.experiment.local_save.local_model_save.PyTorchLocalModelSaver method)
save_to_cloud() (aitoolbox.torchtrain.callbacks.gradient.GradDistributionPlot method)
send_configured_alerts() (aitoolbox.torchtrain.callbacks.wandb.WandBTracking static method)
send_email() (aitoolbox.cloud.AWS.simple_email_service.SESSender method)
SESSender (class in aitoolbox.cloud.AWS.simple_email_service)
set_experiment_dir_path_for_additional_results() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.MachineTranslationResultPackage method)
(aitoolbox.nlp.experiment_evaluation.NLP_result_package.QuestionAnswerResultPackage method)
should_enable_callback() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
should_execute_optimizer_update() (aitoolbox.torchtrain.train_loop.train_loop.TrainLoop method)
split_on_execution_position() (aitoolbox.torchtrain.train_loop.components.callback_handler.CallbacksHandler method)
state_dict() (aitoolbox.torchtrain.multi_loss_optim.MultiOptimizer method)
(aitoolbox.torchtrain.schedulers.basic.AbstractScheduler method)
step() (aitoolbox.torchtrain.multi_loss_optim.MultiOptimizer method)
StepLRScheduler (class in aitoolbox.torchtrain.schedulers.basic)
store_evaluated_metrics_to_history() (aitoolbox.torchtrain.callbacks.performance_eval.ModelPerformanceEvaluation method)
str2bool() (in module aitoolbox.nlp.core.core)
subset_data_loader() (aitoolbox.torchtrain.callbacks.basic.DataSubsetTestRun static method)
T
TensorboardFullTracking (class in aitoolbox.torchtrain.callbacks.tensorboard)
TensorboardReporterBaseCB (class in aitoolbox.torchtrain.callbacks.tensorboard)
TensorboardTrainBatchLoss (class in aitoolbox.torchtrain.callbacks.tensorboard)
TensorboardTrainHistoryMetric (class in aitoolbox.torchtrain.callbacks.tensorboard)
TerminateOnNaN (class in aitoolbox.torchtrain.callbacks.basic)
TextSummarizationResultPackage (class in aitoolbox.nlp.experiment_evaluation.NLP_result_package)
threshold_value (aitoolbox.torchtrain.callbacks.wandb.AlertConfig attribute)
ThresholdEarlyStopping (class in aitoolbox.torchtrain.callbacks.basic)
to() (aitoolbox.torchtrain.multi_loss_optim.MultiLoss method)
torch_cat_transf() (in module aitoolbox.torchtrain.train_loop.components.pred_collate_fns)
TorchMetricsPackage (class in aitoolbox.experiment.result_package.torch_metrics_packages)
TrainHistoryFormatter (class in aitoolbox.torchtrain.callbacks.performance_eval)
training (aitoolbox.torchtrain.model.MultiGPUModelWrap attribute)
(aitoolbox.torchtrain.model.TTBasicModel attribute)
(aitoolbox.torchtrain.model.TTBasicMultiGPUModel attribute)
(aitoolbox.torchtrain.model.TTModel attribute)
(aitoolbox.torchtrain.parallel.TTDataParallel attribute)
(aitoolbox.torchtrain.parallel.TTDistributedDataParallel attribute)
TrainingHistory (class in aitoolbox.experiment.training_history)
TrainingHistoryPlotter (class in aitoolbox.experiment.result_reporting.report_generator)
TrainingHistoryWriter (class in aitoolbox.experiment.result_reporting.report_generator)
TrainLoop (class in aitoolbox.torchtrain.train_loop.train_loop)
TrainLoopCheckpoint (class in aitoolbox.torchtrain.train_loop.train_loop_tracking)
TrainLoopCheckpointEndSave (class in aitoolbox.torchtrain.train_loop.train_loop_tracking)
TrainLoopEndSave (class in aitoolbox.torchtrain.train_loop.train_loop_tracking)
trim() (aitoolbox.nlp.core.vocabulary.Vocabulary method)
try_infer_additional_logging_details() (aitoolbox.torchtrain.callbacks.wandb.WandBTracking method)
try_infer_experiment_details() (aitoolbox.torchtrain.callbacks.abstract.AbstractExperimentCallback method)
TTBasicModel (class in aitoolbox.torchtrain.model)
TTBasicMultiGPUModel (class in aitoolbox.torchtrain.model)
TTDataParallel (class in aitoolbox.torchtrain.parallel)
TTDistributedDataParallel (class in aitoolbox.torchtrain.parallel)
TTModel (class in aitoolbox.torchtrain.model)
TTParallelBase (class in aitoolbox.torchtrain.parallel)
U
unicode_to_ascii() (in module aitoolbox.nlp.core.core)
UNTIL_END_OF_EPOCH (aitoolbox.torchtrain.train_loop.components.message_passing.MessageHandling attribute)
UNTIL_READ (aitoolbox.torchtrain.train_loop.components.message_passing.MessageHandling attribute)
unzip_file() (in module aitoolbox.utils.file_system)
upload_log_file() (aitoolbox.torchtrain.callbacks.basic.LogUpload method)
upload_to_cloud() (aitoolbox.torchtrain.callbacks.tensorboard.TensorboardReporterBaseCB method)
V
validate_msg_handling_settings() (aitoolbox.torchtrain.train_loop.components.message_passing.MessageService static method)
Vocabulary (class in aitoolbox.nlp.core.vocabulary)
W
wandb_alert_level (aitoolbox.torchtrain.callbacks.wandb.AlertConfig attribute)
WandBTracking (class in aitoolbox.torchtrain.callbacks.wandb)
warn_about_result_data_problem() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
(aitoolbox.experiment.training_history.TrainingHistory method)
warn_if_results_dict_not_defined() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage method)
wrap_pre_prepared_history() (aitoolbox.experiment.training_history.TrainingHistory method)
write_csv_tsv() (aitoolbox.experiment.result_reporting.report_generator.TrainingHistoryWriter method)
write_current_train_history() (aitoolbox.torchtrain.callbacks.performance_eval.ModelTrainHistoryFileWriter method)
write_message() (aitoolbox.torchtrain.train_loop.components.message_passing.MessageService method)
write_txt() (aitoolbox.experiment.result_reporting.report_generator.TrainingHistoryWriter static method)
X
XNLIMetric (class in aitoolbox.nlp.experiment_evaluation.NLP_metrics)
XNLIResultPackage (class in aitoolbox.nlp.experiment_evaluation.NLP_result_package)
Z
zero_grad() (aitoolbox.torchtrain.multi_loss_optim.MultiOptimizer method)
zip_additional_results_dump() (aitoolbox.experiment.result_package.abstract_result_packages.AbstractResultPackage static method)
zip_folder() (in module aitoolbox.utils.file_system)
Read the Docs
v: latest
Versions
latest
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds