Skip to content

Commit

Permalink
[wip] gatenet recoding
Browse files Browse the repository at this point in the history
  • Loading branch information
NikolasMarkou committed Apr 9, 2022
1 parent e3cb837 commit e0d70e4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bfcnn/utilities.py
Original file line number Diff line number Diff line change
Expand Up @@ -668,7 +668,7 @@ def resnet_blocks(
# if x > 2.5: return 1
# if -2.5 <= x <= 2.5: return 0.2 * x + 0.5
# mask = keras.activations.hard_sigmoid(2.5 - y)
mask = keras.activations.sigmoid(2.5 - y)
mask = keras.activations.sigmoid(y)
x = keras.layers.Multiply()([x, mask])
# optional multiplier
if use_multiplier:
Expand Down

0 comments on commit e0d70e4

Please sign in to comment.