Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor code #11

Open
dungscout96 opened this issue Sep 18, 2024 · 1 comment
Open

Refactor code #11

dungscout96 opened this issue Sep 18, 2024 · 1 comment
Assignees

Comments

@dungscout96
Copy link
Collaborator

These task parameters such as

win_length = 50,
should be put in the task_params dictionary, replacing the current "mask_probability" key as it was for the MaskedContrastiveLearning task. This should be done for all tasks in Banville paper

@makhalid1999
Copy link
Collaborator

I have refactored the code, and added a standard implementation for all, and taking guidance from the update on the main branch, I have implemented the Trainer class, which would run for all tasks. My implementation is on the banville branch:

https://github.com/sccn/eeg-ssl/blob/banville/libs/ssl_utils.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants