dsipts.models.d3vae package

Submodules

• dsipts.models.d3vae.diffusion_process module
  • get_beta_schedule()
  • default()
  • extract()
  • noise_like()
  • GaussianDiffusion
    • GaussianDiffusion.__init__()
    • GaussianDiffusion.q_sample()
    • GaussianDiffusion.q_sample_target()
    • GaussianDiffusion.p_losses()
    • GaussianDiffusion.log_prob()
• dsipts.models.d3vae.embedding module
  • PositionalEmbedding
    • PositionalEmbedding.__init__()
    • PositionalEmbedding.forward()
  • TokenEmbedding
    • TokenEmbedding.__init__()
    • TokenEmbedding.forward()
  • TemporalEmbedding
    • TemporalEmbedding.__init__()
    • TemporalEmbedding.forward()
  • DataEmbedding
    • DataEmbedding.__init__()
    • DataEmbedding.forward()
• dsipts.models.d3vae.encoder module
  • Cell
    • Cell.__init__()
    • Cell.forward()
  • soft_clamp5()
  • sample_normal_jit()
  • Normal
    • Normal.__init__()
    • Normal.sample()
    • Normal.sample_given_eps()
    • Normal.log_p()
    • Normal.kl()
  • NormalDecoder
    • NormalDecoder.__init__()
    • NormalDecoder.log_prob()
    • NormalDecoder.sample()
  • log_density_gaussian()
  • Encoder
    • Encoder.__init__()
    • Encoder.init_pre_process()
    • Encoder.init_encoder_tower()
    • Encoder.init_decoder_tower()
    • Encoder.init_sampler()
    • Encoder.init_post_process()
    • Encoder.forward()
    • Encoder.decoder_output()
• dsipts.models.d3vae.model module
  • diffusion_generate
    • diffusion_generate.__init__()
    • diffusion_generate.forward()
  • denoise_net
    • denoise_net.__init__()
    • denoise_net.extract()
    • denoise_net.forward()
  • pred_net
    • pred_net.forward()
  • Discriminator
    • Discriminator.__init__()
    • Discriminator.forward()
• dsipts.models.d3vae.neural_operations module
  • SyncBatchNormSwish
    • SyncBatchNormSwish.__init__()
    • SyncBatchNormSwish.forward()
  • get_skip_connection()
  • norm()
  • logit()
  • act()
  • SwishFN
    • SwishFN.forward()
    • SwishFN.backward()
  • Swish
    • Swish.__init__()
    • Swish.forward()
  • normalize_weight_jit()
  • Conv2D
    • Conv2D.__init__()
    • Conv2D.forward()
    • Conv2D.normalize_weight()
  • Identity
    • Identity.__init__()
    • Identity.forward()
  • SyncBatchNorm
    • SyncBatchNorm.__init__()
    • SyncBatchNorm.forward()
  • get_batchnorm()
  • ELUConv
    • ELUConv.__init__()
    • ELUConv.forward()
  • BNELUConv
    • BNELUConv.__init__()
    • BNELUConv.forward()
  • BNSwishConv
    • BNSwishConv.__init__()
    • BNSwishConv.forward()
  • FactorizedReduce
    • FactorizedReduce.__init__()
    • FactorizedReduce.forward()
  • UpSample
    • UpSample.__init__()
    • UpSample.forward()
  • EncCombinerCell
    • EncCombinerCell.__init__()
    • EncCombinerCell.forward()
  • DecCombinerCell
    • DecCombinerCell.__init__()
    • DecCombinerCell.forward()
  • ConvBNSwish
    • ConvBNSwish.__init__()
    • ConvBNSwish.forward()
  • SE
    • SE.__init__()
    • SE.forward()
  • InvertedResidual
    • InvertedResidual.__init__()
    • InvertedResidual.forward()
• dsipts.models.d3vae.resnet module
  • weights_init()
  • MyConvo2d
    • MyConvo2d.__init__()
    • MyConvo2d.forward()
  • Square
    • Square.__init__()
    • Square.forward()
  • Swish
    • Swish.__init__()
    • Swish.forward()
  • MeanPoolConv
    • MeanPoolConv.__init__()
    • MeanPoolConv.forward()
  • ConvMeanPool
    • ConvMeanPool.__init__()
    • ConvMeanPool.forward()
  • ResidualBlock
    • ResidualBlock.__init__()
    • ResidualBlock.forward()
  • Res12_Quadratic
    • Res12_Quadratic.__init__()
    • Res12_Quadratic.forward()
• dsipts.models.d3vae.utils module
  • average_tensor()
  • get_stride_for_cell_type()
  • get_input_size()
  • groups_per_scale()
  • get_arch_cells()