kitcar_ml.utils.models package

Submodules

kitcar_ml.utils.models.advanced_sequential module

Classes:

AdvancedSequential()

This module extends the pytorch module with the feature to give multiple inputs to the forward functions.

class AdvancedSequential(*args: Module)

class AdvancedSequential(arg: OrderedDict[str, Module])

Bases: Sequential

This module extends the pytorch module with the feature to give multiple inputs to the forward functions.

Methods:

forward(*input)

Forward inputs.

Attributes:

forward(*input)

Forward inputs.

Allow passing multiple values in and iterate them through all models.

_modules: Dict[str, Module]

training: bool

_parameters: Dict[str, Optional[Parameter]]

_buffers: Dict[str, Optional[Tensor]]

_non_persistent_buffers_set: Set[str]

_backward_hooks: Dict[int, Callable]

_is_full_backward_hook: Optional[bool]

_forward_hooks: Dict[int, Callable]

_forward_pre_hooks: Dict[int, Callable]

_state_dict_hooks: Dict[int, Callable]

_load_state_dict_pre_hooks: Dict[int, Callable]

_load_state_dict_post_hooks: Dict[int, Callable]

kitcar_ml.utils.models.normalize module

Classes:

Normalize()

class Normalize

Bases: Module

Methods:

forward(inputs, *args)

Normalize input tensor.

Attributes:

forward(inputs: Tensor, *args)

Normalize input tensor.

training: bool

_parameters: Dict[str, Optional[Parameter]]

_buffers: Dict[str, Optional[Tensor]]

_non_persistent_buffers_set: Set[str]

_backward_hooks: Dict[int, Callable]

_is_full_backward_hook: Optional[bool]

_forward_hooks: Dict[int, Callable]

_forward_pre_hooks: Dict[int, Callable]

_state_dict_hooks: Dict[int, Callable]

_load_state_dict_pre_hooks: Dict[int, Callable]

_load_state_dict_post_hooks: Dict[int, Callable]

_modules: Dict[str, Optional[Module]]

Module contents