increased batch size

anjali411 · anjali411 · commit d75d0bccfbe1 · 2020-03-19T12:55:45.000-07:00
diff --git a/test/cpp/api/optim.cpp b/test/cpp/api/optim.cpp
@@ -26,7 +26,7 @@ bool test_optimizer_xor(Options options) {
       Linear(8, 1),
       Functional(torch::sigmoid));
 
-  const int64_t kBatchSize = 6;
+  const int64_t kBatchSize = 50;
   const int64_t kMaximumNumberOfEpochs = 3000;
 
   OptimizerClass optimizer(model->parameters(), options);
