Trying to port a written model from pytorch to flux. And I encounter some differences between the two libraries as expected. And wonder how to imitate the original one.
While creating deconvolution layers in Pytorch the output channel of previous layer might be smaller than the actual layer’s input channel. When I try to do the same thing in Flux, it is not permitted. Is there any way that I can imitate this behavior in Flux?
Thanks for the feedback. I was trying to use SkipConnection method to do that. Right now, the problem that I am facing is that the size of the u0 and d3 are different. u0 is of size (27,10, 90,2) and the d3 is of size (28, 18, 90, 2). How can I concat them to make it 180 channels ?