2016-12-14 14 views
2

Я использую pycaffe для создания своей сети и хочу использовать relu-слой relu вместо обычного слоя, как я могу поместить это в аргумент функции?Реализовать rely-layer relu с pycaffe

from caffe import layers als L, params as P 
n.relu1 = L.Relu(n.conv1, in_place=True) 

это мой текущий код-сниппет

ответ

3

Вы можете использовать negative_slope параметр:

n.relu1 = L.ReLU(n.conv1, in_place=True, relu_param={'negative_slope': 0.1}) 
+0

спасибо, я не знаю, что я должен использовать скобки вокруг параметров – Kev1n91

+0

@ Kev1n91 параметр - словарь, следовательно, фигурные скобки. – Shai