Caffe: ¿cómo obtener la fase de una capa de Python?
Creé un"Python"
capa"myLayer"
en caffe, y úsalo en la redtrain_val.prototxt
Inserto la capa así:
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!
}
Ahora, mi capa solo participa en elTRAIN
ing fase de la red,
¿Cómo puedo saber que en mi capasetup
¿¿función??
class myLayer(caffe.Layer):
def setup(self, bottom, top):
# I want to know here what is the phase?!!
...
PD,
Publiqué esta pregunta enGrupo de Google "Usuarios de Caffe" también. Actualizaré si algo aparece allí.