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

remove batch_size, make tensor 2D instead of 3D #49

Merged
merged 2 commits into from
Jun 4, 2024

Conversation

JernKunpittaya
Copy link
Collaborator

As in #48 ,
-Remove 'batch_size' in exporting onnx file, hence making our onnx file specific to a certain input shape aka an exact number of inputs, which is fine for now, since our witness approach requires us to bake new onnx file for each computation anyway. Also, it seems that ezkl.gen_witness still cannot really handle batch_size with value greater than 1.
-With Batch_size out, we now are able to do everything in 1D tensor, but we decided to make it in 2D aka using reshape(-1,1) instead to make it a lot less computing power for Regression function.

@JernKunpittaya JernKunpittaya merged commit 8efce17 into mpcstats Jun 4, 2024
1 check passed
@JernKunpittaya JernKunpittaya deleted the no_batch_size branch June 4, 2024 15:21
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.

1 participant