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

cloud

Subpackages

  • AWS
    • Submodules
      • data_access
        • BaseDataSaver
        • BaseDataLoader
        • AbstractDatasetFetcher
      • model_load
        • BaseModelLoader
        • PyTorchS3ModelLoader
      • model_save
        • AbstractModelSaver
        • BaseModelSaver
        • PyTorchS3ModelSaver
        • KerasS3ModelSaver
      • results_save
        • AbstractResultsSaver
        • BaseResultsSaver
        • S3ResultsSaver
      • simple_email_service
        • SESSender
  • GoogleCloud
    • Submodules
      • data_access
        • BaseGoogleStorageDataSaver
        • BaseGoogleStorageDataLoader
      • model_load
        • BaseModelGoogleStorageLoader
        • PyTorchGoogleStorageModelLoader
      • model_save
        • BaseModelGoogleStorageSaver
        • PyTorchGoogleStorageModelSaver
        • KerasGoogleStorageModelSaver
      • results_save
        • BaseResultsGoogleStorageSaver
        • GoogleStorageResultsSaver
Previous Next

© Copyright 2023, Marko Vidoni. Revision 6b9bcfda.

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