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

Allow specific input shape #99

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Allow specific input shape #99

wants to merge 1 commit into from

Conversation

doomedramen
Copy link

Currently it assumes the input layer is 1x1xINPUT_LENGTH.
This does not support 2D or 3D inputs, my changes allow the user to specify the shape.

@doomedramen
Copy link
Author

This could also be achieved by getting the X,Y,Z sizes from the input data:

layer_defs.push({type: 'input', out_sx: this.data[0].sx:, out_sy: this.data[0].sy, out_depth:this.data[0].depth});

this is currently hard coded to:

layer_defs.push({type: 'input', out_sx: 1, out_sy: 1, out_depth: input_depth});

@TrevorBlythe
Copy link

It works the same either way bruh.

@371148606
Copy link

371148606 commented Apr 20, 2022 via email

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.

None yet

3 participants