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

Regarding the Parameter Quantity and Accuracy of the EAT-S Model #18

Open
fyl-yyds opened this issue Aug 19, 2024 · 0 comments
Open

Regarding the Parameter Quantity and Accuracy of the EAT-S Model #18

fyl-yyds opened this issue Aug 19, 2024 · 0 comments

Comments

@fyl-yyds
Copy link

Hello
I think your paper is great, but I encountered some problems during training. I hope you can help me take a look.
When I used ESC-50 to train EAT-S on a single 4090,The default parameters used are README.md:
Fs=22.05KHz
seq_len = 114688 ~5sec
python trainer.py --max_lr 3e-4 --run_name r1 --emb_dim 128 --dataset esc50 --seq_len 114688 --mix_ratio 1 --epoch_mix 12 --mix_loss bce --batch_size 128 --n_epochs 3500 --ds_factors 4 4 4 4 --amp --save_path outputs
My model has a parameter size of around 5.18M and a 5-fold cross validation accuracy of 90.5, which is not consistent with the 5.3M and 92.15 values mentioned in the paper. Do you know what the problem is ?

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

1 participant