So, I have a lot of things going on under the hood, but everything is pretty normal, no fancy stuff going on. It should be a pretty normal looking codebase.
With that in mind,
I have this smaaaaall model
In [178]: net
Out[178]: 
Model(
  (conv): Sequential(
    (0): Conv2d(3, 10, kernel_size=(1, 1), stride=(1, 1))
    (1): Conv2d(10, 2, kernel_size=(1, 1), stride=(1, 1))
  )
)

And this ONE image

Figure_1.png800×800 101 KB

And I have trained for over 2000+ epochs.
This isn’t converging.
I want to segment the sky.
Actually, I am debugging. And this obviously means there is some bug somewhere.
What are your thoughts?