Gradient not flowing throughout the entire process but loss is reducing

Hello,
are your input (images?) rescaled between 0 and 1 ?
I saw that your model output is 1. Do you mean to use sigmoid activation ? if so, then it is not included in BCELoss.

If that is the problem, some parts of this discussion may help: