dmlc/MXNet.cpp

A newbie question

Closed this issue · 1 comments

thirdwing commented

In https://github.com/dmlc/MXNet.cpp/blob/master/example/lenet.cpp#L98-L100 we have

args_map["data"] =
      NDArray(Shape(batch_size, 1, W, H), ctx_dev, false);
args_map["data"] = data_array.Slice(0, batch_size).Copy(ctx_dev);

Is there any reason we need two assignments for args_map["data"]?

zhangchen-qinyinghua commented

No, the first assignment is not needed. We should remove that.