zae_engine.models package

Subpackages

• zae_engine.models.builds package
  • Submodules
  • zae_engine.models.builds.autoencoder module
    • AutoEncoder
      • AutoEncoder.encoder
      • AutoEncoder.bottleneck
      • AutoEncoder.decoder
      • AutoEncoder.feature_vectors
      • AutoEncoder.up_pools
      • AutoEncoder.fc
      • AutoEncoder.sig
      • AutoEncoder.hook_handles
      • AutoEncoder.feature_hook()
      • AutoEncoder.feature_output_hook()
      • AutoEncoder.forward()
      • AutoEncoder.get_hooks()
      • AutoEncoder.remove_hooks()
    • VAE
      • VAE.encoder
      • VAE.bottleneck
      • VAE.decoder
      • VAE.feature_vectors
      • VAE.up_pools
      • VAE.fc
      • VAE.sig
      • VAE.fc_mu
      • VAE.fc_logvar
      • VAE.fc_z
      • VAE.encoder_output_shape
      • VAE.encoder_output_features
      • VAE.condition_dim
      • VAE.forward()
      • VAE.reparameterize()
  • zae_engine.models.builds.cnn module
    • CNNBase
      • CNNBase.forward()
      • CNNBase.get_output_shape()
      • CNNBase.initializer()
      • CNNBase.make_body()
      • CNNBase.zero_initializer()
  • zae_engine.models.builds.nested_autoencoder module
    • NestedUNet
      • NestedUNet.forward()
  • zae_engine.models.builds.transformer module
    • BertBase
      • BertBase.forward()
      • BertBase.forward()
    • CoderBase
    • DecoderBase
      • DecoderBase.forward()
    • EncoderBase
      • EncoderBase.forward()
    • TransformerBase
      • TransformerBase.forward()
      • TransformerBase.forward()
  • Module contents
    • DummyModel
      • DummyModel.forward()
• zae_engine.models.converter package
  • Submodules
  • zae_engine.models.converter.dim_converter module
    • DimConverter
      • DimConverter.model
      • DimConverter.module_dict
      • DimConverter.new_module_dict
      • DimConverter.layer_dict
      • DimConverter.param_dict
      • DimConverter.find_convertable()
      • DimConverter.const_getter()
      • DimConverter.convert()
      • DimConverter.apply_new_dict()
      • DimConverter.const_getter()
      • DimConverter.convert()
      • DimConverter.correction_map
      • DimConverter.dim_correction()
      • DimConverter.find_convertable()
  • Module contents
• zae_engine.models.foundations package
  • Submodules
  • zae_engine.models.foundations.benchmark module
    • TimeSeriesBert
      • TimeSeriesBert.forward()
  • zae_engine.models.foundations.bert module
    • BertEmbedding
      • BertEmbedding.forward()
    • bert_base()
  • zae_engine.models.foundations.resnet module
    • cbam_injection()
    • cbamresnet101()
    • cbamresnet152()
    • cbamresnet18()
    • cbamresnet34()
    • cbamresnet50()
    • resnet101()
    • resnet152()
    • resnet18()
    • resnet34()
    • resnet50()
    • resnet_deco()
    • se_injection()
    • seresnet101()
    • seresnet152()
    • seresnet18()
    • seresnet34()
    • seresnet50()
  • zae_engine.models.foundations.unet module
    • unet_brain()
  • zae_engine.models.foundations.word_embedding module
    • FastTextEmbedding
      • FastTextEmbedding.embedding
      • FastTextEmbedding.forward()
      • FastTextEmbedding.forward()
    • Word2VecEmbedding
      • Word2VecEmbedding.embedding
      • Word2VecEmbedding.forward()
      • Word2VecEmbedding.forward()
  • Module contents

Submodules

zae_engine.models.utility module

zae_engine.models.utility.initializer(m)

Initialize the parameters in model. For Linear layer, weights are initialized following normal distribution. For BatchNorm layer, weights are initialized following uniform distribution. For LayerNorm layer, weights are not initialized. For others, weights are initialized following kaiming method based on normal distribution. (https://arxiv.org/abs/1502.01852v1)

Parameters:

m – The member of model.

Module contents