Revision 46b6016d0e6554ac8e9a394438b0f548d9b59acc authored by liqfu on 14 March 2018, 16:30:30 UTC, committed by liqfu on 14 March 2018, 16:30:30 UTC
1 parent f1f3bb4
Eval.cpp
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE.md file in the project root for full license information.
//
// Eval.cpp : Defines the exported functions for the DLL application.
//
#define _CRT_SECURE_NO_WARNINGS // "secure" CRT not available on all platforms --add this at the top of all CPP files that give "function or variable may be unsafe" warnings
#include "stdafx.h"
#include "Basics.h"
#define EVAL_LOCAL
#include "Eval.h"
namespace Microsoft { namespace MSR { namespace CNTK {
template <class ElemType>
std::string GetEvalName(ElemType)
{
std::string empty;
return empty;
}
template <>
std::string GetEvalName(float)
{
std::string name = "GetEvalF";
return name;
}
template <>
std::string GetEvalName(double)
{
std::string name = "GetEvalD";
return name;
}
template <class ElemType>
void Eval<ElemType>::Init(const std::string& /*config*/)
{
LogicError("Init shouldn't be called, use constructor");
// not implemented, calls the underlying class instead
}
// Destroy - cleanup and remove this class
// NOTE: this destroys the object, and it can't be used past this point
template <class ElemType>
void Eval<ElemType>::Destroy()
{
m_eval->Destroy();
}
// Eval Constructor
template <class ElemType>
void Eval<ElemType>::GetEvalClass(const std::string& config)
{
typedef void (*GetEvalProc)(IEvaluateModel<ElemType>** peval);
// initialize just in case
m_eval = NULL;
std::wstring module = L"CNTKEval";
// get the name for the dll we want to use, default to CNTKEval.dll
std::string::size_type found = config.find("evaluator=");
if (found != std::string::npos)
{
std::string::size_type end = config.find_first_of("\n \t", found);
if (end != std::string::npos)
{
module = msra::strfun::utf16(config.substr(found, end - found));
}
}
// create a variable of each type just to call the proper templated version
ElemType elemType = ElemType();
GetEvalProc getEvalProc = (GetEvalProc) m_plugin->Load(module, GetEvalName(elemType));
getEvalProc(&m_eval);
}
// Eval Constructor
// options - [in] string of options (i.e. "-windowsize:11 -addenergy") data reader specific
template <class ElemType>
Eval<ElemType>::Eval(const std::string& config) : m_plugin(make_shared<Plugin>())
{
GetEvalClass(config);
m_eval->Init(config);
}
// destructor - cleanup temp files, etc.
template <class ElemType>
Eval<ElemType>::~Eval()
{
// free up resources
if (m_eval != NULL)
{
m_eval->Destroy();
m_eval = NULL;
}
}
// CreateNetwork - create the network from the specified description
// networkDescription - network description
template <class ElemType>
void Eval<ElemType>::CreateNetwork(const std::string& networkDescription)
{
m_eval->CreateNetwork(networkDescription);
}
// GetNodeDimensions - Get the node dimensions of the specified nodes
// dimensions - map from name of node to dimension of the node
// nodeGroup - type of node we are requesting (input/output/specified)
template <class ElemType>
void Eval<ElemType>::GetNodeDimensions(std::map<std::wstring, size_t>& dimensions, NodeGroup nodeGroup)
{
m_eval->GetNodeDimensions(dimensions, nodeGroup);
}
// StartEvaluateMinibatchLoop - Prepare network for Evaluate() calls.
// outputNodeName - name of node that will be evaluated
template <class ElemType>
void Eval<ElemType>::StartEvaluateMinibatchLoop(const std::wstring& outputNodeName)
{
m_eval->StartEvaluateMinibatchLoop(outputNodeName);
}
// Evaluate - Evalute using the model with the given inputs and outputs
// inputs - map from node name to input vector
// outputs - map from node name to output vector, outputs vectors need to be preallocated by caller, sizing will happen during evaluation
template <class ElemType>
void Eval<ElemType>::Evaluate(std::map<std::wstring, std::vector<ElemType>*>& inputs, std::map<std::wstring, std::vector<ElemType>*>& outputs)
{
m_eval->Evaluate(inputs, outputs);
}
// Evaluate - Evaluate using the network without input and provide the outputs
// outputs - map from node name to output vector, outputs vectors need to be preallocated by caller, sizing will happen during evaluation
template <class ElemType>
void Eval<ElemType>::Evaluate(std::map<std::wstring, std::vector<ElemType>*>& outputs)
{
m_eval->Evaluate(outputs);
}
// ResetState - Reset the cell state when we get the start of an utterance
template <class ElemType>
void Eval<ElemType>::ResetState()
{
m_eval->ResetState();
}
//The explicit instantiation
template class Eval<double>;
template class Eval<float>;
} } }
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...