yamle.models package

yamle.models.model_factory(model_type=None)

This function is used to return a model instance based on the model type.

Parameters:

model_type (str) – The type of model to create.

Return type:

Type[Module]

Submodules

• yamle.models.convnet module
  • ConvNetModel
    • ConvNetModel.tasks
    • ConvNetModel.forward()
    • ConvNetModel.final_layer()
    • ConvNetModel.add_method_specific_layers()
    • ConvNetModel.add_specific_args()
    • ConvNetModel.training
  • ResidualConvNetModel
    • ResidualConvNetModel.training
• yamle.models.densenet module
  • EmptyBlock
    • EmptyBlock.forward()
    • EmptyBlock.training
  • DenseLayer
    • DenseLayer.bn_function()
    • DenseLayer.forward()
    • DenseLayer.replace_layers_for_quantization()
    • DenseLayer.training
  • Transition
    • Transition.forward()
    • Transition.training
  • DenseBlock
    • DenseBlock.forward()
    • DenseBlock.training
  • DenseNetModel
    • DenseNetModel.tasks
    • DenseNetModel.forward()
    • DenseNetModel.final_layer()
    • DenseNetModel.add_method_specific_layers()
    • DenseNetModel.training
    • DenseNetModel.replace_layers_for_quantization()
    • DenseNetModel.add_specific_args()
• yamle.models.fc module
  • FCModel
    • FCModel.tasks
    • FCModel.forward()
    • FCModel.final_layer()
    • FCModel.add_method_specific_layers()
    • FCModel.replace_layers_for_quantization()
    • FCModel.add_specific_args()
    • FCModel.training
  • ResidualFCModel
    • ResidualFCModel.training
• yamle.models.gp module
  • GPModel
    • GPModel.forward()
    • GPModel.final_layer()
    • GPModel.add_method_specific_layers()
    • GPModel.training
    • GPModel.add_specific_args()
    • GPModel.reset()
• yamle.models.mixer module
  • MixerLayer
    • MixerLayer.extra_repr()
  • MixerModel
    • MixerModel.tasks
    • MixerModel.add_method_specific_layers()
    • MixerModel.forward()
    • MixerModel.final_layer()
    • MixerModel.add_specific_args()
    • MixerModel.training
• yamle.models.model module
  • BaseModel
    • BaseModel.tasks
    • BaseModel.forward()
    • BaseModel.final_layer()
    • BaseModel.add_specific_args()
    • BaseModel.reset()
    • BaseModel.replace_layers_for_quantization()
    • BaseModel.add_method_specific_layers()
    • BaseModel.training
• yamle.models.operations module
  • output_activation()
  • OutputActivation
    • OutputActivation.forward()
    • OutputActivation.extra_repr()
    • OutputActivation.enable()
    • OutputActivation.disable()
    • OutputActivation.training
  • ReshapeOutput
    • ReshapeOutput.forward()
    • ReshapeOutput.extra_repr()
    • ReshapeOutput.training
  • ReshapeInput
    • ReshapeInput.forward()
    • ReshapeInput.training
  • Unsqueeze
    • Unsqueeze.forward()
    • Unsqueeze.training
  • Add
    • Add.forward()
    • Add.training
  • Multiply
    • Multiply.forward()
    • Multiply.training
  • ResidualLayer
    • ResidualLayer.forward()
    • ResidualLayer.training
  • ParallelModel
    • ParallelModel.forward()
    • ParallelModel.initialise_members_same()
    • ParallelModel.training
  • Normalization
    • Normalization.forward()
    • Normalization.training
  • Pooling
    • Pooling.forward()
    • Pooling.training
  • Activation
    • Activation.forward()
    • Activation.training
  • MatrixMultiplication
    • MatrixMultiplication.forward()
    • MatrixMultiplication.training
  • LinearNormActivation
    • LinearNormActivation.forward()
    • LinearNormActivation.replace_layers_for_quantization()
    • LinearNormActivation.training
  • Conv2dNormActivation
    • Conv2dNormActivation.forward()
    • Conv2dNormActivation.replace_layers_for_quantization()
    • Conv2dNormActivation.training
  • DoubleConv2d
    • DoubleConv2d.forward()
    • DoubleConv2d.training
  • DepthwiseSeparableConv2d
    • DepthwiseSeparableConv2d.reset_parameters()
    • DepthwiseSeparableConv2d.forward()
    • DepthwiseSeparableConv2d.training
  • CompletelySeparableConv2d
    • CompletelySeparableConv2d.reset_parameters()
    • CompletelySeparableConv2d.forward()
    • CompletelySeparableConv2d.training
  • SqueezeAndExcitation
    • SqueezeAndExcitation.forward()
    • SqueezeAndExcitation.training
  • LSTM
    • LSTM.forward()
    • LSTM.training
  • Lambda
    • Lambda.forward()
    • Lambda.training
  • Reduction
    • Reduction.forward()
    • Reduction.extra_repr()
    • Reduction.training
  • ScalarMultiplier
    • ScalarMultiplier.forward()
    • ScalarMultiplier.training
    • ScalarMultiplier.enable()
    • ScalarMultiplier.disable()
    • ScalarMultiplier.extra_repr()
  • ScalarAdder
    • ScalarAdder.training
    • ScalarAdder.forward()
  • LinearExtractor
    • LinearExtractor.training
    • LinearExtractor.forward()
    • LinearExtractor.extra_repr()
  • Conv2dExtractor
    • Conv2dExtractor.training
    • Conv2dExtractor.forward()
    • Conv2dExtractor.extra_repr()
• yamle.models.resnet module
  • BasicBlock
    • BasicBlock.expansion
    • BasicBlock.forward()
    • BasicBlock.replace_layers_for_quantization()
    • BasicBlock.training
  • Bottleneck
    • Bottleneck.base_width_multiplier
    • Bottleneck.groups
    • Bottleneck.expansion
    • Bottleneck.base_width
    • Bottleneck.forward()
    • Bottleneck.replace_layers_for_quantization()
    • Bottleneck.training
  • Blocks
    • Blocks.forward()
    • Blocks.training
  • ResNetModel
    • ResNetModel.tasks
    • ResNetModel.training
    • ResNetModel.forward()
    • ResNetModel.final_layer()
    • ResNetModel.add_method_specific_layers()
    • ResNetModel.replace_layers_for_quantization()
    • ResNetModel.add_specific_args()
• yamle.models.rnn module
  • RNNModel
    • RNNModel.tasks
    • RNNModel.forward()
    • RNNModel.final_layer()
    • RNNModel.add_method_specific_layers()
    • RNNModel.add_specific_args()
    • RNNModel.training
  • RNNAutoEncoderModel
    • RNNAutoEncoderModel.tasks
    • RNNAutoEncoderModel.forward()
    • RNNAutoEncoderModel.final_layer()
    • RNNAutoEncoderModel.add_specific_args()
    • RNNAutoEncoderModel.training
  • InputRepeater
    • InputRepeater.forward()
    • InputRepeater.training
• yamle.models.transformer module
  • PreNorm
  • FeedForward
    • FeedForward.extra_repr()
  • Attention
    • Attention.forward()
    • Attention.extra_repr()
    • Attention.training
  • TransformerEncoderLayer
    • TransformerEncoderLayer.extra_repr()
  • PositionalEncoding
    • PositionalEncoding.forward()
    • PositionalEncoding.reset_parameters()
    • PositionalEncoding.training
  • TransformerModel
    • TransformerModel.tasks
    • TransformerModel.reset_parameters()
    • TransformerModel.forward()
    • TransformerModel.final_layer()
    • TransformerModel.generate()
    • TransformerModel.add_method_specific_layers()
    • TransformerModel.add_specific_args()
    • TransformerModel.training
• yamle.models.unet module
  • DownBlock
    • DownBlock.forward()
    • DownBlock.training
  • UpBlock
    • UpBlock.forward()
    • UpBlock.training
  • UNetModel
    • UNetModel.tasks
    • UNetModel.add_method_specific_layers()
    • UNetModel.forward()
    • UNetModel.final_layer()
    • UNetModel.add_specific_args()
    • UNetModel.training
• yamle.models.vgg module
  • EmptyBlock
    • EmptyBlock.forward()
    • EmptyBlock.training
  • VGGBlock
    • VGGBlock.forward()
    • VGGBlock.replace_layers_for_quantization()
    • VGGBlock.training
  • Blocks
    • Blocks.forward()
    • Blocks.training
  • VGGModel
    • VGGModel.tasks
    • VGGModel.forward()
    • VGGModel.final_layer()
    • VGGModel.add_method_specific_layers()
    • VGGModel.replace_layers_for_quantization()
    • VGGModel.add_specific_args()
    • VGGModel.training
• yamle.models.visual_transformer module
  • SpatialPositionalEmbedding
    • SpatialPositionalEmbedding.forward()
    • SpatialPositionalEmbedding.get_cls_token_indices()
    • SpatialPositionalEmbedding.training
  • VisualTransformerModel
    • VisualTransformerModel.tasks
    • VisualTransformerModel.add_method_specific_layers()
    • VisualTransformerModel.forward()
    • VisualTransformerModel.final_layer()
    • VisualTransformerModel.add_specific_args()
    • VisualTransformerModel.training