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

drop_last issue #7

Open
fuyw opened this issue Aug 29, 2024 · 0 comments
Open

drop_last issue #7

fuyw opened this issue Aug 29, 2024 · 0 comments

Comments

@fuyw
Copy link

fuyw commented Aug 29, 2024

The current code seems to have the drop_last issue as discussed in https://github.com/VEWOXIC/FITS.

The batch size is not set to 1 for the test dataset:

batch_size = args.batch_size # 10 # bsz=1 for evaluation

or the drop_last needs to be set to False:

drop_last = True

Sometimes the performance will drop after fixing this issue.

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