Fix memory setting

jozef-mokry · Feb 20, 2016 · a43fcf8 · a43fcf8
1 parent 6d9369d
commit a43fcf8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lm/interpolate/interpolate_main.cc b/lm/interpolate/interpolate_main.cc
@@ -37,13 +37,14 @@ int main(int argc, char *argv[]) {
   po::variables_map vm;
   po::store(po::parse_command_line(argc, argv, options), vm);
   if (argc == 1 || vm["help"].as<bool>()) {
-		std::cerr << "Interpolate multiple models\n\n" << options << std::endl;
+		std::cerr << "Interpolate multiple models\n" << options << std::endl;
     return 1;
   }
   po::notify(vm);
   instances_config.sort = pipe_config.sort;
-  instances_config.lazy_memory = instances_config.sort.total_memory;
   instances_config.model_read_chain_mem = instances_config.sort.buffer_size;
+  instances_config.extension_write_chain_mem = instances_config.sort.total_memory;
+  instances_config.lazy_memory = instances_config.sort.total_memory;
 
   if (pipe_config.lambdas.empty() && tuning_file.empty()) {
     std::cerr << "Provide a tuning file with -t xor weights with -w." << std::endl;