python - learning - caffe vs tensorflow

Implementar el modelo de regresión de caffe

Question

He entrenado una red de regresión con caffe . Utilizo la capa "EuclideanLoss" tanto en la fase de tren como en la de prueba. Los tracé y los resultados parecen prometedores.

Ahora quiero implementar el modelo y usarlo. Sé que si se usa SoftmaxLoss , la capa final debe ser Softmax en el archivo de implementación. ¿Qué debería ser esto en el caso de la Euclidean loss ?

Answer 1

Para implementar solo necesita descartar la capa de pérdida, en su caso, la capa "EuclideanLoss" . La salida de tu red es el "bottom" que alimentaste la capa de pérdida.

Para la capa "SoftmaxWithLoss" (y "SigmoidCrossEntropy" ) debe reemplazar la capa de pérdida, ya que la capa de pérdida incluye una capa adicional dentro de ella (por razones de cálculo).