Skip to content

Allow specific input shape#99

Open
doomedramen wants to merge 1 commit intokarpathy:masterfrom
doomedramen:patch-1
Open

Allow specific input shape#99
doomedramen wants to merge 1 commit intokarpathy:masterfrom
doomedramen:patch-1

Conversation

@doomedramen
Copy link
Copy Markdown

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
Copy Markdown
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
Copy Markdown

It works the same either way bruh.

@371148606
Copy link
Copy Markdown

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.

3 participants