HingeLossParameter.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Text;
using MyCaffe.basecode;
 
namespace MyCaffe.param
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class HingeLossParameter : LayerParameterBase
    {
        public enum Norm
        {
            L1 = 1,
            L2 = 2
        }
 
        Norm m_norm = Norm.L1;
 
        public HingeLossParameter()
        {
        }
 
        [Description("Specify the Norm to use: L1 or L2")]
        public Norm norm
        {
            get { return m_norm; }
            set { m_norm = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            HingeLossParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            HingeLossParameter p = (HingeLossParameter)src;
            m_norm = p.m_norm;
        }
 
        public override LayerParameterBase Clone()
        {
            HingeLossParameter p = new HingeLossParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add("norm", norm.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static HingeLossParameter FromProto(RawProto rp)
        {
            string strVal;
            HingeLossParameter p = new HingeLossParameter();
 
            if ((strVal = rp.FindValue("norm")) != null)
            {
                switch (strVal)
                {
                    case "L1":
                        p.norm = Norm.L1;
                        break;
 
                    case "L2":
                        p.norm = Norm.L2;
                        break;
 
                    default:
                        throw new Exception("Unknown 'norm' value: " + strVal);
                }
            }
 
            return p;
        }
    }
}