fix: activation for last layer of UNet2D#16

hardik01shah wants to merge 2 commits intocosmic-cortex:mastercosmic-cortex/pytorch-UNet:masterfrom

hardik01shah:masterhardik01shah/pytorch-UNet:masterCopy head branch name to clipboard