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

DIffusion hyperparameters #15

Closed
gabrielvc opened this issue May 12, 2024 · 1 comment
Closed

DIffusion hyperparameters #15

gabrielvc opened this issue May 12, 2024 · 1 comment

Comments

@gabrielvc
Copy link

Hi,

As far as i'm concerned, there is an issue with the current diffusion hyperparameters defined in the config file.
Running the current parameters for T=200 yield an sqrt(alpha_bar) of approximately 0.4, which is of course far from the Gaussian, which would be equivalent to sqrt(alpha_bar) of approximately 0.01 or something really small.
I guess this is why I can't generate ECGs using the generate function, even if my trained loss is extremely small.
Indeed, the network is able to denoise moderately noisy ecgs but generation still don't work.

Furthermore, I could not find in your paper the hyperparameters you used in the diffusion training.
I would really like to be able to reproduce your work, but for the moment it is really challenging.

@juanlopezcode
Copy link
Collaborator

Hi,
The network should run with the provided diffusion hyperparameters, we mentioned them in the appendix of our paper, but they are also under the config file in this repo: https://github.com/AI4HealthUOL/SSSD-ECG/blob/main/src/sssd/config/config_SSSD_ECG.json

Follow the train.py and inference.py files under the following directory, these should be your unique files to run once dataset is located (don't update diffusion hyperparameters): SSSD-ECG/src
/sssd/

I hope this helps
Best
Juan

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