"weight" vs. "pos_weight" in nn.BCEWithLogitsLoss()

Thank you for the response!

Ya, I realized after some discussion on a related post that the default return type for pos_weight is a single mean value of the error calculation for every pixel. So what is propagated backwards is a single value. This made a lot more sense to me after.