quaterion.train 包

子包

• quaterion.train.cache 包
  • 子模块
    • quaterion.train.cache.cache_config 模块
      • CacheConfig
      • CacheType
      • KeyExtractorType
    • quaterion.train.cache.cache_encoder 模块
      • CacheMode
    • quaterion.train.cache.cache_model 模块
    • quaterion.train.cache.cache_train_collator 模块
    • quaterion.train.cache.in_memory_cache_encoder 模块
  • 模块内容
    • 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 包
  • 子模块
    • quaterion.train.callbacks.cleanup_callback 模块
      • CleanupCallback
    • quaterion.train.callbacks.metrics_callback 模块
      • MetricsCallback
  • 模块内容
• quaterion.train.xbm 包
  • 子模块
    • quaterion.train.xbm.xbm_buffer 模块
      • XbmBuffer
    • quaterion.train.xbm.xbm_config 模块
      • XbmConfig
      • XbmDevice
  • 模块内容

子模块

• quaterion.train.cache_mixin 模块
  • CacheMixin
    • CacheMixin.check_encoders_persisted()
    • CacheMixin.load_cache()
    • CacheMixin.save_cache()
• quaterion.train.trainable_model 模块
  • 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

模块内容