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:
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: