API Documentation

API Documentation

• libai.config
• libai.data
  • libai.data.data_utils module
  • libai.data.datasets module
  • libai.data.samplers module
  • libai.data.build module
• libai.engine
  • libai.engine.default module
  • libai.engine.hooks module
  • libai.engine.trainer module
• libai.evaluation
• libai.layers
• libai.models
  • VisionTransformer
  • SwinTransformer
  • ResMLP
  • BERT
  • RoBERTa
  • T5
  • GPT-2
• libai.optim
• libai.scheduler
• libai.tokenizer
• libai.utils
  • libai.utils.distributed module
  • libai.utils.events module
  • libai.utils.logger module
  • libai.utils.checkpoint module