search

eeyhsong / NICE-EEG

[ICLR 2024] M/EEG-based image decoding with contrastive learning. i. Propose a contrastive learning framework to align image and eeg. ii. Resolving brain activity for biological plausibility.
https://arxiv.org/abs/2308.13234
MIT License
85 stars 15 forks source link

where is the TSConv? #11

Closed tian-2024 closed 2 weeks ago

tian-2024 commented 2 months ago

sorry, I can't find the module, I can only see EEG_GAT. In my understanding, TSConv should contain sa and ga, and temp conv+avg pool+spa conv+ linear，but i can't find the part

eeyhsong commented 2 weeks ago

Hello @tian-2024,

Sorry for the late reply. You can see all encoders in https://github.com/eeyhsong/NICE-EEG/blob/main/eeg_encoder.py TSConv is here https://github.com/eeyhsong/NICE-EEG/blob/dccf0f36aba55b62b8307461a3c883034c4fb502/nice_stand.py#L61

Best.

tian-2024 commented 2 weeks ago

Hello @tian-2024,

Sorry for the late reply. You can see all encoders in https://github.com/eeyhsong/NICE-EEG/blob/main/eeg_encoder.py TSConv is here

https://github.com/eeyhsong/NICE-EEG/blob/dccf0f36aba55b62b8307461a3c883034c4fb502/nice_stand.py#L61

Best.

I understood~ thanks