Added typecast to rectify Windows build error

2018-03-16 19:57:51 +05:30
parent 321c6d8b83
commit 8bfe4c8e7e
1 changed files with 3 additions and 1 deletions
@@ -324,7 +324,9 @@ BOOST_AUTO_TEST_CASE(BatchNormTest)

  arma::mat input = dataset.submat(0, 0, dataset.n_rows-1, 2);

-  BatchNorm<> model(input.n_rows);
+  size_t numUnits = input.n_rows;
+
+  BatchNorm<> model(numUnits);
  model.Reset();

  // Non-Deteministic Forward Pass Test.