Caffe: как получить фазу слоя Python?
Я создал"Python"
слой"myLayer"
в кафе, и использовать его в сетиtrain_val.prototxt
Я вставляю слой так:
layer {
name: "my_py_layer"
type: "Python"
bottom: "in"
top: "out"
python_param {
module: "my_module_name"
layer: "myLayer"
}
include { phase: TRAIN } # THIS IS THE TRICKY PART!
}
Теперь мой слой участвует только вTRAIN
фаза сети,
как я могу знать, что в моем слоеsetup
функция ??
class myLayer(caffe.Layer):
def setup(self, bottom, top):
# I want to know here what is the phase?!!
...
PS,
Я разместил этот вопрос наГруппа пользователей "Caffe Users" также. Я обновлюсь, если что-нибудь появится там.