-
Notifications
You must be signed in to change notification settings - Fork 66
Open
Description
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?
Metadata
Metadata
Assignees
Labels
No labels