models

dreem.models

Model architectures and layers.

Modules:

Name	Description
attention_head	Module containing different components of multi-head attention heads.
embedding	Module containing different position and temporal embeddings.
global_tracking_transformer	Module containing GTR model used for training.
gtr_runner	Module containing training, validation and inference logic.
mlp	Multi-Layer Perceptron (MLP) module.
model_utils	Module containing model helper functions.
transformer	DETR Transformer class.
visual_encoder	Module for different visual feature extractors.