public TestBase(string strName, int nDeviceID = DEFAULT_DEVICE_ID, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT, object tag = null)
{
m_strName = strName;
if (create_count == 1)
{
m_rgTests.Add(create(DataType.FLOAT, strName, nDeviceID, engine));
m_rgTests.Add(create(DataType.DOUBLE, strName, nDeviceID, engine));
}
else
{
for (int i = 0; i < create_count; i++)
{
ITest iTestF = create(i, DataType.FLOAT, strName, nDeviceID, engine);
if (iTestF != null)
{
m_rgTests.Add(iTestF);
}
ITest iTestD = create(i, DataType.DOUBLE, strName, nDeviceID, engine);
if (iTestD != null)
{
m_rgTests.Add(iTestD);
}
}
}
}
public ModelBuilderTest(string strName, int nDeviceID, EngineParameter.Engine engine)
: base(strName, null, nDeviceID)
{
m_engine = engine;
m_strName = strName;
m_strBaseDir = TestBase.GetTestPath("\\MyCaffe\\test_data", true, true);
}
public BeamSearchTest2(string strName, int nDeviceID, EngineParameter.Engine engine)
: base(strName, new List <int>() { 3, 2, 4, 1 }, nDeviceID)
{
m_engine = engine;
NetParameter net_param = new NetParameter();
LayerParameter input = new LayerParameter(LayerParameter.LayerType.INPUT);
input.input_param.shape.Add(new BlobShape(new List <int>()
{
1, 1, 1
}));
input.input_param.shape.Add(new BlobShape(new List <int>()
{
80, 1, 1
}));
input.input_param.shape.Add(new BlobShape(new List <int>()
{
80, 1, 1
}));
input.input_param.shape.Add(new BlobShape(new List <int>()
{
80, 1
}));
input.top.Add("dec");
input.top.Add("enc");
input.top.Add("encr");
input.top.Add("encc");
net_param.layer.Add(input);
string strModel = net_param.ToProto("root").ToString();
m_net = new Net <T>(m_cuda, m_log, net_param, new CancelEvent(), null);
InputLayerEx <T> layer = new InputLayerEx <T>(m_cuda, m_log, m_net.layers[0]);
layer.OnGetData += Layer_OnGetData;
m_net.layers[0] = layer;
m_rgTestSequences.Add("rdany but you can call me dany");
m_rgTestSequences.Add("rdany call me dany");
m_rgTestSequences.Add("rdany you can call me dany");
m_rgTestSequences.Add("my name is dany");
m_rgTestSequences.Add("call me dany");
m_rgrgTestSequenceIndexes = new List <List <int> >();
foreach (string strSequence in m_rgTestSequences)
{
string[] rgstrWords = strSequence.Split(' ');
List <int> rgIdx = new List <int>();
foreach (string strWord in rgstrWords)
{
int nIdx = layer.Vocabulary.WordToIndex(strWord);
rgIdx.Add(nIdx);
}
m_rgrgTestSequenceIndexes.Add(rgIdx);
}
}
public ReinforcementLossLayerTest(string strDs = "MNIST", EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("ReinforcementLoss Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
m_settings = new SettingsCaffe();
m_db = new MyCaffeImageDatabase();
m_db.InitializeWithDsName(m_settings, strDs, m_evtCancel.Name);
}
public AttentionLayerTest2(string strName, int nDeviceID, EngineParameter.Engine engine)
: base(strName, new List <int>() { 3, 2, 4, 1 }, nDeviceID)
{
m_engine = engine;
m_blobA = new Blob <T>(m_cuda, m_log);
m_blobB = new Blob <T>(m_cuda, m_log);
m_blobC = new Blob <T>(m_cuda, m_log);
m_blobState = new Blob <T>(m_cuda, m_log);
m_blobClip = new Blob <T>(m_cuda, m_log);
}
protected virtual ITest create(DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
{
if (dt == DataType.DOUBLE)
{
return(new Test <double>(strName, nDeviceID, engine));
}
else
{
return(new Test <float>(strName, nDeviceID, engine));
}
}
protected override ITest create(DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
{
if (dt == DataType.DOUBLE)
{
return(new BlobSimpleTest <double>(strName, nDeviceID));
}
else
{
return(new BlobSimpleTest <float>(strName, nDeviceID));
}
}
public ModelDataLayerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Text Model Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
m_settings = new SettingsCaffe();
m_settings.EnableLabelBalancing = false;
m_settings.EnableLabelBoosting = false;
m_settings.EnablePairInputSelection = false;
m_settings.EnableRandomInputSelection = false;
m_db = createImageDb(null);
m_db.InitializeWithDsName1(m_settings, "MNIST");
}
public TestBase(string strName, int nDeviceID = DEFAULT_DEVICE_ID, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT, object tag = null, bool bHalf = false)
{
m_bHalf = bHalf;
// If an auto test has set the GPUID, us it instead.
LocalDataStoreSlot lds = Thread.GetNamedDataSlot("GPUID");
if (lds != null)
{
object obj = Thread.GetData(lds);
if (obj != null)
{
string strGpuId = obj.ToString();
if (!string.IsNullOrEmpty(strGpuId))
{
int nVal;
if (int.TryParse(strGpuId, out nVal) && nDeviceID < 4)
{
nDeviceID = nVal;
}
}
}
}
m_strName = strName;
if (create_count == 1)
{
m_rgTests.Add(create(DataType.FLOAT, strName, nDeviceID, engine));
m_rgTests.Add(create(DataType.DOUBLE, strName, nDeviceID, engine));
}
else
{
for (int i = 0; i < create_count; i++)
{
ITest iTestF = create(i, DataType.FLOAT, strName, nDeviceID, engine);
if (iTestF != null)
{
m_rgTests.Add(iTestF);
}
ITest iTestD = create(i, DataType.DOUBLE, strName, nDeviceID, engine);
if (iTestD != null)
{
m_rgTests.Add(iTestD);
}
}
}
}
public Test(string strName, int nDeviceID = TestBase.DEFAULT_DEVICE_ID, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
{
m_dt = (typeof(T) == typeof(double)) ? DataType.DOUBLE : DataType.FLOAT;
m_cuda = GetCuda(nDeviceID);
string str = name;
if (str.Length > 0)
{
str += " -> ";
}
m_log = GetLog(str + strName + " - type " + (typeof(T)).ToString());
m_engine = engine;
}
/// <summary>
/// Parses the LSTM and Engine options.
/// </summary>
private void getLstmType()
{
if (m_strType == "CAFFE")
{
m_lstmEngine = EngineParameter.Engine.CAFFE;
}
else if (m_strType == "CUDNN")
{
m_lstmEngine = EngineParameter.Engine.CUDNN;
}
else
{
throw new Exception("Unknown Engine type '" + m_strType + "', expected <CAFFE|CUDNN> format.");
}
}
public ResNetSiameseModelBuilderTest(string strName, int nDeviceID, EngineParameter.Engine engine)
: base(strName, nDeviceID, engine)
{
if (strName.Contains(ResNetModelBuilder.MODEL.RESNET101.ToString()))
{
m_model = ResNetModelBuilder.MODEL.RESNET101;
}
else if (strName.Contains(ResNetModelBuilder.MODEL.RESNET152.ToString()))
{
m_model = ResNetModelBuilder.MODEL.RESNET152;
}
else
{
m_model = ResNetModelBuilder.MODEL.RESNET56;
}
}
public AnnotatedDataLayerTest(string strDs = null, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Annotated Data Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
m_settings = new SettingsCaffe();
m_settings.EnableLabelBalancing = false;
m_settings.EnableLabelBoosting = false;
m_settings.EnablePairInputSelection = false;
m_settings.EnableRandomInputSelection = false;
m_settings.ImageDbLoadDataCriteria = true; // Required, for Annotation Data is stored in the Data Criteria.
m_db = new MyCaffeImageDatabase();
if (strDs != null && strDs.Length > 0)
{
m_db.InitializeWithDsName(m_settings, strDs);
}
}
protected override ITest create(DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
{
string strPath = TestBase.CudaPath;
if (dt == DataType.DOUBLE)
{
CudaDnn <double> .SetDefaultCudaPath(strPath);
return(new DataLayerTest <double>(strName, nDeviceID, this));
}
else
{
CudaDnn <float> .SetDefaultCudaPath(strPath);
return(new DataLayerTest <float>(strName, nDeviceID, this));
}
}
protected virtual ITest create(DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
{
ITest iTest;
if (dt == DataType.DOUBLE)
{
iTest = new Test <double>(strName, nDeviceID, engine, m_bHalf);
}
else
{
iTest = new Test <float>(strName, nDeviceID, engine, m_bHalf);
}
iTest.SetParent(this);
return(iTest);
}
public BinaryHashLayerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Binary Hash Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public CropLayerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("BatchNorm Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public GRNLayerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("GRN Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public TanhLayerTest2(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Tanh Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (dt == common.DataType.DOUBLE)
{
return(new MyCaffePreProcessorTest <double>(strName, nDeviceID, engine));
}
else
{
return(new MyCaffePreProcessorTest <float>(strName, nDeviceID, engine));
}
}
public MyCaffePreProcessorTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("MyCaffePreProcessor Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public PersistCaffeTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Filter Persist Caffe", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public GaussianFillerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Gaussian Filler Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public PositiveUnitballFillerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("PositiveUnitball Filler Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public ContrastiveLossLayerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Contrastive Loss Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public MyCaffeControlTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Caffe Control Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public InfogainLossLayerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Infogain Loss Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
public InnerProductLayerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("InnerProduct Layer Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
protected override ITest create(common.DataType dt, string strName, int nDeviceID, EngineParameter.Engine engine)
{
if (dt == common.DataType.DOUBLE)
{
return(new ContrastiveLossLayerTest <double>(strName, nDeviceID, engine));
}
else
{
return(new ContrastiveLossLayerTest <float>(strName, nDeviceID, engine));
}
}
public UniformFillerTest(EngineParameter.Engine engine = EngineParameter.Engine.DEFAULT)
: base("Uniform Filler Test", TestBase.DEFAULT_DEVICE_ID, engine)
{
}
