bug in mtml: shared_embedding

Xiaolong Wang · facebook-github-bot · commit f18b4b647319 · 2017-07-26T20:21:53.000-07:00
Summary: A bug reported in MTML group: https://fburl.com/lumicchc The reason is that in MTML, the `task_shared_embedding` was not correctly initalized in python Reviewed By: xianjiec Differential Revision: D5502875 fbshipit-source-id: 3538d917392568ecd37c39059dc86f866bce9543
diff --git a/caffe2/operators/distance_op.cc b/caffe2/operators/distance_op.cc
@@ -197,7 +197,7 @@ bool DotProductOp<float, CPUContext>::RunOnDevice() {
   auto* result = Output(DOT_OUT);
   CAFFE_ENFORCE_EQ(X.ndim(), Y.ndim());
   for (int i = 0; i < X.ndim(); ++i) {
-    CAFFE_ENFORCE_EQ(X.dim32(i), Y.dim32(i));
+    CAFFE_ENFORCE_EQ(X.dim32(i), Y.dim32(i), "dimension at ", i);
   }
   int N, D;
   if (X.size() > 0) {

Original file line number	Diff line number	Diff line change
`@@ -197,7 +197,7 @@ bool DotProductOp<float, CPUContext>::RunOnDevice() {`
`197`	`197`	`auto* result = Output(DOT_OUT);`
`198`	`198`	`CAFFE_ENFORCE_EQ(X.ndim(), Y.ndim());`
`199`	`199`	`for (int i = 0; i < X.ndim(); ++i) {`
`200`		`- CAFFE_ENFORCE_EQ(X.dim32(i), Y.dim32(i));`
	`200`	`+ CAFFE_ENFORCE_EQ(X.dim32(i), Y.dim32(i), "dimension at ", i);`
`201`	`201`	`}`
`202`	`202`	`int N, D;`
`203`	`203`	`if (X.size() > 0) {`