MishParameter.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.ComponentModel;
using MyCaffe.basecode;
 
namespace MyCaffe.param.beta
{
    [Serializable]
    [TypeConverter(typeof(ExpandableObjectConverter))]
    public class MishParameter : EngineParameter
    {
        double m_dfThreshold = 20;
        int m_nMethod = 0;
 
        public MishParameter()
            : base()
        {
        }
 
        public string useCaffeReason()
        {
            return "Currenly only CAFFE supported.";
        }
 
        public bool useCudnn()
        {
            return false;
        }
 
        [Description("Specifies the max threshold value used in exp functions.")] 
        public double threshold
        {
            get { return m_dfThreshold; }
            set { m_dfThreshold = value; }
        }
 
        public int method
        {
            get { return m_nMethod; }
            set { m_nMethod = value; }
        }
 
        public override object Load(System.IO.BinaryReader br, bool bNewInstance = true)
        {
            RawProto proto = RawProto.Parse(br.ReadString());
            MishParameter p = FromProto(proto);
 
            if (!bNewInstance)
                Copy(p);
 
            return p;
        }
 
        public override void Copy(LayerParameterBase src)
        {
            base.Copy(src);
 
            if (src is MishParameter)
            {
                MishParameter p = (MishParameter)src;
                m_dfThreshold = p.m_dfThreshold;
                m_nMethod = p.m_nMethod;
            }
        }
 
        public override LayerParameterBase Clone()
        {
            MishParameter p = new MishParameter();
            p.Copy(this);
            return p;
        }
 
        public override RawProto ToProto(string strName)
        {
            RawProto rpBase = base.ToProto("engine");
            RawProtoCollection rgChildren = new RawProtoCollection();
 
            rgChildren.Add(rpBase.Children);
 
            if (threshold != 0)
                rgChildren.Add("threshold", threshold.ToString());
 
            if (method != 0)
                rgChildren.Add("method", threshold.ToString());
 
            return new RawProto(strName, "", rgChildren);
        }
 
        public static new MishParameter FromProto(RawProto rp)
        {
            string strVal;
            MishParameter p = new MishParameter();
 
            ((EngineParameter)p).Copy(EngineParameter.FromProto(rp));
 
            if ((strVal = rp.FindValue("threshold")) != null)
                p.threshold = ParseDouble(strVal);
 
            if ((strVal = rp.FindValue("method")) != null)
                p.method = int.Parse(strVal);
 
            return p;
        }
    }
}