Su Yan (Stella) issues

Repositories
Issues
Comments

Results 1 issues of


                                            Su Yan (Stella)

Flax nnx ConvTranspose Does Not Restore Input Shape When Used with Conv (Unexpected Behavior)

When using Flax’s `Conv` and `ConvTranspose` layers in pair, the `ConvTranspose` does not seem to correctly restore the original input shape, even when the parameters are set in a way...