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

objax.Vectorize batch_axis argument for multiple arguments default must be adaptively sized. #191

Open
JeremyNixon opened this issue Jan 12, 2021 · 0 comments

Comments

@JeremyNixon
Copy link

At present the default is set for a single argument. If there are two arguments, ex:
objax.Vectorize(augment, objax.VarCollection())
batch_axis=(0) as a default leads to an error.
The default batch_axis must be set to , batch_axis=(0, 0) for two arguments, and in general have one 0 for each argument.

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