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
  • aitoolbox
  • Edit on GitHub

aitoolbox

Subpackages

  • cloud
    • Subpackages
      • AWS
        • Submodules
      • GoogleCloud
        • Submodules
  • experiment
    • Subpackages
      • core_metrics
        • Submodules
      • local_load
        • Submodules
      • local_save
        • Submodules
      • result_package
        • Submodules
      • result_reporting
        • Submodules
    • Submodules
      • experiment_saver
        • AbstractExperimentSaver
        • BaseFullExperimentSaver
        • BaseFullExperimentS3Saver
        • FullPyTorchExperimentS3Saver
        • FullKerasExperimentS3Saver
        • BaseFullExperimentGoogleStorageSaver
        • FullPyTorchExperimentGoogleStorageSaver
        • FullKerasExperimentGoogleStorageSaver
      • local_experiment_saver
        • BaseFullExperimentLocalSaver
        • FullPyTorchExperimentLocalSaver
        • FullKerasExperimentLocalSaver
      • training_history
        • TrainingHistory
  • nlp
    • Subpackages
      • core
        • Submodules
      • experiment_evaluation
        • Submodules
    • Submodules
      • torch_collate_fns
        • qa_concat_ctx_span_collate_fn()
  • torchtrain
    • Subpackages
      • callbacks
        • Submodules
      • data
        • Submodules
      • schedulers
        • Submodules
      • train_loop
        • Subpackages
        • Submodules
    • Submodules
      • model
        • TTModel
        • TTBasicModel
        • TTBasicMultiGPUModel
        • MultiGPUModelWrap
        • ModelWrap
      • model_predict
        • PyTorchModelPredictor
      • multi_loss_optim
        • MultiLoss
        • MultiOptimizer
      • parallel
        • TTParallelBase
        • TTDataParallel
        • TTDistributedDataParallel
  • utils
    • Submodules
      • dict_util
        • combine_prediction_metadata_batches()
        • flatten_dict()
        • combine_dict_elements()
        • flatten_combine_dict()
      • file_system
        • create_folder_hierarchy()
        • zip_folder()
        • unzip_file()
      • util
        • function_exists()
        • copy_function()
        • is_empty_function()
        • flatten_list_of_lists()

© Copyright 2023, Marko Vidoni. Revision 6b9bcfda.

Built with Sphinx using a theme provided by Read the Docs.