Skip to content

Skip connections

BackboneInternalSkipConnection

Bases: SkipConnection

This is a placeholder module used by the quantization engine only. This module will be replaced by converters. The module is to be used as a skip connection between blocks inside the backbone.

Source code in src/super_gradients/modules/skip_connections.py
26
27
28
29
30
31
class BackboneInternalSkipConnection(SkipConnection):
    """
    This is a placeholder module used by the quantization engine only.
    This module will be replaced by converters.
    The module is to be used as a skip connection between blocks inside the backbone.
    """

CrossModelSkipConnection

Bases: SkipConnection

This is a placeholder module used by the quantization engine only. This module will be replaced by converters. The module is to be used as a skip connection between backbone and the head.

Source code in src/super_gradients/modules/skip_connections.py
42
43
44
45
46
47
class CrossModelSkipConnection(SkipConnection):
    """
    This is a placeholder module used by the quantization engine only.
    This module will be replaced by converters.
    The module is to be used as a skip connection between backbone and the head.
    """

HeadInternalSkipConnection

Bases: SkipConnection

This is a placeholder module used by the quantization engine only. This module will be replaced by converters. The module is to be used as a skip connection between blocks inside the head.

Source code in src/super_gradients/modules/skip_connections.py
34
35
36
37
38
39
class HeadInternalSkipConnection(SkipConnection):
    """
    This is a placeholder module used by the quantization engine only.
    This module will be replaced by converters.
    The module is to be used as a skip connection between blocks inside the head.
    """

Residual

Bases: nn.Module

This is a placeholder module used by the quantization engine only. This module will be replaced by converters. The module is to be used as a residual skip connection within a single block.

Source code in src/super_gradients/modules/skip_connections.py
 4
 5
 6
 7
 8
 9
10
11
12
class Residual(nn.Module):
    """
    This is a placeholder module used by the quantization engine only.
    This module will be replaced by converters.
    The module is to be used as a residual skip connection within a single block.
    """

    def forward(self, x):
        return x

SkipConnection

Bases: nn.Module

This is a placeholder module used by the quantization engine only. This module will be replaced by converters. The module is to be used as a skip connection between blocks.

Source code in src/super_gradients/modules/skip_connections.py
15
16
17
18
19
20
21
22
23
class SkipConnection(nn.Module):
    """
    This is a placeholder module used by the quantization engine only.
    This module will be replaced by converters.
    The module is to be used as a skip connection between blocks.
    """

    def forward(self, x):
        return x