February 15, 2023 at 11:37 am #56968
The only thing that isn’t quite right is the HAIBAL architecture compared to the model design.
In your architecture, the number of convolution filters is constant (16). I call this the simple variant, which I also started with.
However, the model design starts with 32 conv filters and increases to 1024 filters at the bottom of the UNet.