public void Load(String ndPath)
{
Symbol lenet = Lenet.CreateLenet();
Dictionary <String, NDArray> map = NDArray.LoadToMap(ndPath);
Dictionary <String, NDArray> args = new Dictionary <string, NDArray>();
Dictionary <String, NDArray> auxs = new Dictionary <string, NDArray>();
foreach (var pair in map)
{
int idx = pair.Key.IndexOf(':');
if (idx > 0)
{
String tp = pair.Key.Substring(0, idx);
String name = pair.Key.Substring(idx + 1);
if (tp == "arg")
{
args[name] = pair.Value;
}
else if (tp == "aux")
{
auxs[name] = pair.Value;
}
}
else
{
args[pair.Key] = pair.Value;
}
}
Context ctx = Context.Gpu();
args["data"] = _data;
args["data_label"] = _label;
lenet.InferArgsMap(ctx, args, args, new XavierInitializer(2));
_symbol = lenet;
_args = args;
_exe = lenet.SimpleBind(_ctx, args, new XavierInitializer(2));
}
private static void Train()
{
Lenet lenet = new Lenet();
lenet.Run();
}
