zae_engine package

Subpackages

• zae_engine.data package
  • Subpackages
    • zae_engine.data.collate package
      • Submodules
      • zae_engine.data.collate.core module
      • zae_engine.data.collate.modules module
      • Module contents
    • zae_engine.data.dataset package
      • Submodules
      • zae_engine.data.dataset.parquet module
      • Module contents
  • Module contents
• zae_engine.examples package
  • Submodules
  • zae_engine.examples.classification1D module
    • InferenceDataset
    • InferenceTrainer
      • InferenceTrainer.test_step()
      • InferenceTrainer.train_step()
    • core()
  • zae_engine.examples.ddpm_mnist module
    • CustomMNISTDataset
    • DDPM
      • DDPM.timestep_embedding
      • DDPM.t_embed_proj
      • DDPM.forward()
    • DDPMTrainer
      • DDPMTrainer.generate()
      • DDPMTrainer.noise_scheduling()
      • DDPMTrainer.test_step()
      • DDPMTrainer.train_step()
      • DDPMTrainer.visualize_samples()
    • ForwardDiffusion
    • NoiseScheduler
      • NoiseScheduler.beta
      • NoiseScheduler.alpha
      • NoiseScheduler.alpha_bar
      • NoiseScheduler.sqrt_alpha_bar
      • NoiseScheduler.sqrt_one_minus_alpha_bar
      • NoiseScheduler.posterior_variance
      • NoiseScheduler.timesteps
      • NoiseScheduler.cosine_beta_schedule()
      • NoiseScheduler.get_sigma()
      • NoiseScheduler.linear_beta_schedule()
    • TimestepEmbedding
      • TimestepEmbedding.forward()
  • zae_engine.examples.segmentation1D module
    • InferenceDataset
    • InferenceTrainer
      • InferenceTrainer.test_step()
      • InferenceTrainer.train_step()
    • core()
  • Module contents
• zae_engine.loss package
  • Submodules
  • zae_engine.loss.angular module
    • ArcFaceLoss
      • ArcFaceLoss.forward()
  • zae_engine.loss.iou module
    • GIoU()
    • IoU()
    • mIoU()
  • zae_engine.loss.norm module
    • mse()
  • Module contents
• zae_engine.metrics package
  • Submodules
  • zae_engine.metrics.bijective module
    • BijectiveMetrix
      • BijectiveMetrix.bijective_run_pair
      • BijectiveMetrix.injective_run_pair
      • BijectiveMetrix.surjective_run_pair
      • BijectiveMetrix.injective_mat
      • BijectiveMetrix.surjective_mat
      • BijectiveMetrix.bijective_mat
      • BijectiveMetrix.bijective_f1
      • BijectiveMetrix.injective_f1
      • BijectiveMetrix.surjective_f1
      • BijectiveMetrix.bijective_acc
      • BijectiveMetrix.injective_acc
      • BijectiveMetrix.surjective_acc
      • BijectiveMetrix.bijective_confusion()
      • BijectiveMetrix.map_and_confusion()
      • BijectiveMetrix.run_pairing()
      • BijectiveMetrix.summary()
  • zae_engine.metrics.confusion module
    • confusion_matrix()
    • print_confusion_matrix()
  • zae_engine.metrics.count module
    • Acc
      • Acc.accuracy()
      • Acc.top_k_accuracy()
    • accuracy()
    • f_beta()
    • f_beta_from_mat()
  • zae_engine.metrics.iou module
    • giou()
    • miou()
  • zae_engine.metrics.signals module
    • mse()
    • peak_signal_to_noise()
    • qilv()
    • rms()
    • signal_to_noise()
  • Module contents
• zae_engine.models package
  • Subpackages
    • zae_engine.models.builds package
      • Submodules
      • zae_engine.models.builds.autoencoder module
      • zae_engine.models.builds.cnn module
      • zae_engine.models.builds.nested_autoencoder module
      • zae_engine.models.builds.transformer module
      • Module contents
    • zae_engine.models.converter package
      • Submodules
      • zae_engine.models.converter.dim_converter module
      • Module contents
    • zae_engine.models.foundations package
      • Submodules
      • zae_engine.models.foundations.benchmark module
      • zae_engine.models.foundations.bert module
      • zae_engine.models.foundations.resnet module
      • zae_engine.models.foundations.unet module
      • zae_engine.models.foundations.word_embedding module
      • Module contents
  • Submodules
  • zae_engine.models.utility module
    • initializer()
  • Module contents
• zae_engine.nn_night package
  • Subpackages
    • zae_engine.nn_night.blocks package
      • Submodules
      • zae_engine.nn_night.blocks.conv_block module
      • zae_engine.nn_night.blocks.resblock module
      • zae_engine.nn_night.blocks.spatial_attention module
      • zae_engine.nn_night.blocks.unet_block module
      • Module contents
    • zae_engine.nn_night.layers package
      • Submodules
      • zae_engine.nn_night.layers.additional module
      • zae_engine.nn_night.layers.dynamic_pool module
      • zae_engine.nn_night.layers.involution module
      • zae_engine.nn_night.layers.position_encoding module
      • zae_engine.nn_night.layers.residual_connection module
      • zae_engine.nn_night.layers.selective_kernel_conv module
      • Module contents
  • Module contents
• zae_engine.operation package
  • Submodules
  • zae_engine.operation.run_length module
    • Run
      • Run.start_index
      • Run.end_index
      • Run.value
      • Run.end_index
      • Run.start_index
      • Run.value
    • RunLengthCodec
      • RunLengthCodec.encode()
      • RunLengthCodec.decode()
      • RunLengthCodec.sanitize()
      • RunLengthCodec.__call__()
      • RunLengthCodec.decode()
      • RunLengthCodec.encode()
      • RunLengthCodec.sanitize()
    • RunList
      • RunList.all_runs
      • RunList.sense
      • RunList.original_length
      • RunList.filtered()
      • RunList.raw()
  • Module contents
• zae_engine.schedulers package
  • Submodules
  • zae_engine.schedulers.core module
    • SchedulerBase
      • SchedulerBase.optimizer
      • SchedulerBase.total_iters
      • SchedulerBase.eta_min
      • SchedulerBase.last_epoch
      • SchedulerBase._step_count
      • SchedulerBase.get_lr()
    • SchedulerChain
      • SchedulerChain.schedulers
      • SchedulerChain.optimizer
      • SchedulerChain.next_iters
      • SchedulerChain.total_iters
      • SchedulerChain.i_scheduler
      • SchedulerChain.get_lr()
      • SchedulerChain.sanity_check()
      • SchedulerChain.step()
  • zae_engine.schedulers.scheduler module
    • CosineAnnealingScheduler
      • CosineAnnealingScheduler.get_lr()
    • WarmUpScheduler
      • WarmUpScheduler.get_lr()
  • Module contents
• zae_engine.trainer package
  • Subpackages
    • zae_engine.trainer.addons package
      • Submodules
      • zae_engine.trainer.addons.core module
      • zae_engine.trainer.addons.mix_precision module
      • zae_engine.trainer.addons.mpu module
      • zae_engine.trainer.addons.state_manager module
      • zae_engine.trainer.addons.web_logger module
      • Module contents
  • Module contents
• zae_engine.utils package
  • Subpackages
    • zae_engine.utils.decorators package
      • Module contents
  • Submodules
  • zae_engine.utils.io module
    • example_ecg()
    • example_mri()
    • image_from_url()
  • Module contents

Module contents