We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
It slower than before too many times when I use this sampler
(self.indices[i] for i in torch.multinomial(self.weights, self.num_samples, replacement=True)) it seems that this expression cost too much time!
Any one have any solution?
The text was updated successfully, but these errors were encountered:
@nothingeasy In my view, for function is surplus. Modify as follow:
return iter(torch.multinomial(self.weights, self.num_samples, replacement=True).tolist())
I test this modify currently(Not comfirm correct)
Sorry, something went wrong.
No branches or pull requests
It slower than before too many times when I use this sampler
(self.indices[i] for i in torch.multinomial(self.weights, self.num_samples, replacement=True))
it seems that this expression cost too much time!
Any one have any solution?
The text was updated successfully, but these errors were encountered: