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(); }