Hello,
It is better to use ReLU activation function between hidden layers, and use sigmoid at last to generate probability map.
model = torch.nn.Sequential(fc1, relu, fc2, relu, fc3, sigmoid)

PS: you could format your code snippet with three `