Hi, If you want to use the sigmoid activation function in the last layer you should use this to the model.py what is the purpose of using torch. sigmoid in every layer in the training and validation loop? I removed visdom and fit this model just as other CNN models like UNet. so, I also didn't understand of the code in the screenshot. Can you please explain this?
Hi, If you want to use the sigmoid activation function in the last layer you should use this to the model.py what is the purpose of using torch. sigmoid in every layer in the training and validation loop? I removed visdom and fit this model just as other CNN models like UNet. so, I also didn't understand of the code in the screenshot. Can you please explain this?