keras-unet-collection icon indicating copy to clipboard operation
keras-unet-collection copied to clipboard

Add new activation functions

Open mehrdad-dev opened this issue 2 years ago • 2 comments

How can I add an activation function like tanh for the "output_activation" parameter? @yingkaisha

Thanks!

mehrdad-dev avatar Jun 08 '22 10:06 mehrdad-dev

Hi @mehrdad-dev, Did you get to know how to add tanh as an output activation function?

akaiml avatar Jul 09 '22 17:07 akaiml

you should change the code directly! like this, and then pass your activation function

if activation:
        if activation == 'Sigmoid':
            X = Activation('sigmoid', name='{}_activation'.format(name))(X)            
        else:
            X = activation(X)

mehrdad-dev avatar Jul 10 '22 04:07 mehrdad-dev