Skip to content

Commit 29370e4

Browse files
author
Adrià Arrufat
committed
parse logistic (aka sigmoid) activations
1 parent 07548d4 commit 29370e4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tool/darknet2pytorch.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -271,6 +271,8 @@ def create_network(self, blocks):
271271
model.add_module('relu{0}'.format(conv_id), nn.ReLU(inplace=True))
272272
elif activation == 'mish':
273273
model.add_module('mish{0}'.format(conv_id), Mish())
274+
elif activation == 'logistic':
275+
model.add_module('sigmoid{0}'.format(conv_id), nn.Sigmoid())
274276
else:
275277
print("convalution havn't activate {}".format(activation))
276278

0 commit comments

Comments
 (0)