tmp

xmlyqing00 · Nov 29, 2018 · 54c5972 · 54c5972
1 parent 36f195b
commit 54c5972
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/evaluator/train_evaluator.cpp b/src/evaluator/train_evaluator.cpp
@@ -58,7 +58,8 @@ void test(  CompatibilityNet & comp_net,
         target = squeeze(target, /*dim*/1);
 
         Tensor output = comp_net.forward(data);
-
+        cout << target[0] << endl;
+        cout << output[0] << endl;
         test_loss += nll_loss(output, target, symbols_w, Reduction::Sum).template item<float>();
         auto pred = output.argmax(1);
         correct_n += pred.eq(target).sum().template item<int64_t>();