     if (num_target_tokens < 0 || target_ids.size() > num_target_tokens) {
       num_target_tokens = target_ids.size();
     }
-    if (source_sentences.size() == batch_size) break;
+    if (source_sentences.size() == 1000) break;
   }
   std::cerr << "Done." << std::endl;
   std::cerr << source_sentences.size()

  std::cout << g.graphviz() << std::endl;

  std::cerr << "Training..." << std::endl;
  Adam opt;
  int num_epochs = 20;
  for(size_t epoch = 0; epoch < num_epochs; ++epoch) {

              << "Loss = " << g["cost"].val()[0]
              << std::endl;
  }
#endif

  return 0;
}
