protosleepnet.baselines.train_st_3ch_mixer#

Train SleepTransformer (Phan) 3ch with channel mixer on SHHS.

Per-channel EpochTransformer(in_chan=1) + modality embeddings + accuracy-weighted ZeroEmbeddingDropout + TransformerEncoder mixer (no residual) + attention pooling + SequenceTransformer + classifier.

Usage:

python -m protosleepnet.baselines.train_st_3ch_mixer –gpu_id 0

Attributes#

DROPOUT

D_CLF

D_MODEL

EPOCH_KWARGS

HF_REPO_ID

MODEL_NAME

N_CLASSES

SEQ_KWARGS

TRAIN_CONFIG

Functions#

build_model([cdropout])

main()

Module Contents#

protosleepnet.baselines.train_st_3ch_mixer.build_model(cdropout=0.5)#
protosleepnet.baselines.train_st_3ch_mixer.main()#
protosleepnet.baselines.train_st_3ch_mixer.DROPOUT = 0.1#
protosleepnet.baselines.train_st_3ch_mixer.D_CLF = 1024#
protosleepnet.baselines.train_st_3ch_mixer.D_MODEL = 128#
protosleepnet.baselines.train_st_3ch_mixer.EPOCH_KWARGS#
protosleepnet.baselines.train_st_3ch_mixer.HF_REPO_ID = '4rooms/physioex'#
protosleepnet.baselines.train_st_3ch_mixer.MODEL_NAME = 'sleeptransformer-phan-3ch-mixer'#
protosleepnet.baselines.train_st_3ch_mixer.N_CLASSES = 5#
protosleepnet.baselines.train_st_3ch_mixer.SEQ_KWARGS#
protosleepnet.baselines.train_st_3ch_mixer.TRAIN_CONFIG#