elektronn3.modules package

Submodules

• elektronn3.modules.axial_attention module
  • AxialAttention
    • AxialAttention.forward()
  • AxialImageTransformer
    • AxialImageTransformer.forward()
  • AxialPositionalEmbedding
    • AxialPositionalEmbedding.forward()
  • Deterministic
    • Deterministic.forward()
    • Deterministic.record_rng()
  • IrreversibleBlock
    • IrreversibleBlock.forward()
  • PermuteToFrom
    • PermuteToFrom.forward()
  • ReversibleBlock
    • ReversibleBlock.backward_pass()
    • ReversibleBlock.forward()
  • ReversibleSequence
    • ReversibleSequence.forward()
  • Rezero
    • Rezero.forward()
  • SelfAttention
    • SelfAttention.forward()
  • Sequential
    • Sequential.forward()
  • calculate_permutations()
  • exists()
  • map_el_ind()
  • sort_and_return_indices()
• elektronn3.modules.evonorm module
  • EvoNorm
    • EvoNorm.forward()
    • EvoNorm.reset_parameters()
  • group_std()
  • instance_std()
• elektronn3.modules.l1batchnorm module
  • L1BatchNorm
    • L1BatchNorm.forward()
  • L1GroupNorm
    • L1GroupNorm.forward()
  • l1_group_norm()
• elektronn3.modules.layers module
  • GatherExcite
    • GatherExcite.forward()
• elektronn3.modules.loss module
  • ACLoss
    • ACLoss.forward()
    • ACLoss.get_length()
    • ACLoss.get_region()
  • CombinedLoss
    • CombinedLoss.forward()
  • DiceLoss
    • DiceLoss.forward()
  • DistanceWeightedMSELoss
    • DistanceWeightedMSELoss.forward()
  • FixMatchSegLoss
    • FixMatchSegLoss.forward()
    • FixMatchSegLoss.get_random_augmenters()
  • FocalLoss
    • FocalLoss.forward()
  • GAPTripletMarginLoss
    • GAPTripletMarginLoss.forward()
  • LovaszLoss
    • LovaszLoss.forward()
  • MaskedMSELoss
    • MaskedMSELoss.forward()
  • MixedCombinedLoss
    • MixedCombinedLoss.forward()
  • NorpfDiceLoss
    • NorpfDiceLoss.forward()
  • SoftmaxBCELoss
    • SoftmaxBCELoss.forward()
  • dice_loss()
  • global_average_pooling()
  • norpf_dice_loss()
• elektronn3.modules.lovasz_losses module
  • StableBCELoss
    • StableBCELoss.forward()
  • binary_xloss()
  • flatten_binary_scores()
  • flatten_probas()
  • iou()
  • iou_binary()
  • lovasz_grad()
  • lovasz_hinge()
  • lovasz_hinge_flat()
  • lovasz_softmax()
  • lovasz_softmax_flat()
  • mean()
  • xloss()
• elektronn3.modules.wsconv module
  • FWS
    • FWS.forward()
    • FWS.standardize_weight()
  • WSConv1d
    • WSConv1d.weight
    • WSConv1d.bias
    • WSConv1d.forward()
    • WSConv1d.standardize_weight()
  • WSConv2d
    • WSConv2d.forward()
    • WSConv2d.standardize_weight()
  • WSConv3d
    • WSConv3d.forward()
    • WSConv3d.standardize_weight()
  • WSConvTranspose2d
    • WSConvTranspose2d.forward()
    • WSConvTranspose2d.standardize_weight()
  • WSConvTranspose3d
    • WSConvTranspose3d.forward()
    • WSConvTranspose3d.standardize_weight()