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

modified for dynamic batch size in sample #31

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

zhaomingwork
Copy link

when config to small batchsize, sample funcion in solver.py will out of scope, such as:
acoustic_token_gt = codes_np_gt[i]
IndexError: index 7 is out of bounds for axis 0 with size 7
so change to dynamic batchsize to avoid such problem.

@yt605155624
Copy link
Contributor

@zhaomingwork maybe min(10, content_gt.size(0)) is better? cause content_gt.size(0) can be very large

@zhaomingwork
Copy link
Author

@zhaomingwork maybe min(10, content_gt.size(0)) is better? cause content_gt.size(0) can be very large

ok. done.

yt605155624 added a commit to yt605155624/SoundStorm that referenced this pull request Aug 9, 2023
@yt605155624
Copy link
Contributor

fix this in #30

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

Successfully merging this pull request may close these issues.

2 participants