zae_engine.trainer package

Subpackages

• zae_engine.trainer.addons package
  • Submodules
  • zae_engine.trainer.addons.core module
    • AddOnBase
      • AddOnBase.apply()
      • AddOnBase.apply()
  • zae_engine.trainer.addons.mix_precision module
    • PrecisionMixerAddon
      • PrecisionMixerAddon.run_batch()
      • PrecisionMixerAddon.apply()
  • zae_engine.trainer.addons.mpu module
    • MultiGPUAddon
      • MultiGPUAddon.run()
      • MultiGPUAddon.train_process()
      • MultiGPUAddon.apply()
  • zae_engine.trainer.addons.state_manager module
    • StateManagerAddon
      • StateManagerAddon.save_state()
      • StateManagerAddon.load_state()
      • StateManagerAddon.save_model()
      • StateManagerAddon.save_optimizer()
      • StateManagerAddon.save_scheduler()
      • StateManagerAddon.apply()
  • zae_engine.trainer.addons.web_logger module
    • NeptuneLoggerAddon
      • NeptuneLoggerAddon.logging()
      • NeptuneLoggerAddon.init_neptune()
      • NeptuneLoggerAddon.apply()
    • WandBLoggerAddon
      • WandBLoggerAddon.logging()
      • WandBLoggerAddon.init_wandb()
      • WandBLoggerAddon.apply()
  • Module contents

Module contents