quaterion.train package

Subpackages

• quaterion.train.cache package
  • Submodules
    • quaterion.train.cache.cache_config module
      • CacheConfig
      • CacheType
      • KeyExtractorType
    • quaterion.train.cache.cache_encoder module
      • CacheMode
    • quaterion.train.cache.cache_model module
    • quaterion.train.cache.cache_train_collator module
    • quaterion.train.cache.in_memory_cache_encoder module
  • Module contents
    • CacheConfig
      • CacheConfig.batch_size
      • CacheConfig.cache_type
      • CacheConfig.key_extractors
      • CacheConfig.mapping
      • CacheConfig.num_workers
      • CacheConfig.save_dir
    • CacheType
      • CacheType.AUTO
      • CacheType.CPU
      • CacheType.GPU
      • CacheType.NONE
• quaterion.train.callbacks package
  • Submodules
    • quaterion.train.callbacks.cleanup_callback module
      • CleanupCallback
    • quaterion.train.callbacks.metrics_callback module
      • MetricsCallback
  • Module contents
• quaterion.train.xbm package
  • Submodules
    • quaterion.train.xbm.xbm_buffer module
      • XbmBuffer
    • quaterion.train.xbm.xbm_config module
      • XbmConfig
      • XbmDevice
  • Module contents

Submodules

• quaterion.train.cache_mixin module
  • CacheMixin
    • CacheMixin.check_encoders_persisted()
    • CacheMixin.load_cache()
    • CacheMixin.save_cache()
• quaterion.train.trainable_model module
  • TrainableModel
    • TrainableModel.configure_caches()
    • TrainableModel.configure_encoders()
    • TrainableModel.configure_head()
    • TrainableModel.configure_loss()
    • TrainableModel.configure_metrics()
    • TrainableModel.configure_xbm()
    • TrainableModel.process_results()
    • TrainableModel.save_servable()
    • TrainableModel.setup_dataloader()
    • TrainableModel.loss
    • TrainableModel.model

Module contents