Thanks for your reply. I am still getting a size mismatch error though - I’m assuming this is due to improperly calculating the input dimensions for my layers? Do you have any tips for how to do this?
For reference, the error message is: size mismatch, m1: [1028 x 2240], m2: [896 x 120]
And I am starting with a single channel 40x20 pixel image, trying to use the architecture described in my original post.