/// <summary>
/// Setup the layer.
/// </summary>
/// <param name="colBottom">Specifies the collection of bottom (input) Blobs.</param>
/// <param name="colTop">Specifies the collection of top (output) Blobs.</param>
public override void LayerSetUp(BlobCollection <T> colBottom, BlobCollection <T> colTop)
{
base.LayerSetUp(colBottom, colTop);
m_colSigmoidBottomVec.Clear();
m_colSigmoidBottomVec.Add(colBottom[0]);
m_colSigmoidTopVec.Clear();
m_colSigmoidTopVec.Add(m_blobSigmoidOutput);
m_sigmoidLayer.Setup(m_colSigmoidBottomVec, m_colSigmoidTopVec);
m_nIgnoreLabel = m_param.loss_param.ignore_label;
}
/// <summary>
/// Setup the layer.
/// </summary>
/// <param name="colBottom">Specifies the collection of bottom (input) Blobs.</param>
/// <param name="colTop">Specifies the collection of top (output) Blobs.</param>
public override void LayerSetUp(BlobCollection <T> colBottom, BlobCollection <T> colTop)
{
base.LayerSetUp(colBottom, colTop);
m_colSigmoidBottomVec = new BlobCollection <T>();
m_colSigmoidBottomVec.Add(colBottom[0]);
m_colSigmoidTopVec = new BlobCollection <T>();
m_colSigmoidTopVec.Add(m_blobSigmoidOutput);
m_sigmoidLayer.Setup(m_colSigmoidBottomVec, m_colSigmoidTopVec);
m_nIgnoreLabel = m_param.loss_param.ignore_label;
if (m_param.loss_param.normalization != LossParameter.NormalizationMode.NONE)
{
m_normalization = m_param.loss_param.normalization;
}
else
{
m_normalization = (m_param.loss_param.normalize) ? LossParameter.NormalizationMode.VALID : LossParameter.NormalizationMode.BATCH_SIZE;
}
}