mitigate memory usage during torch.save (#20)

OpenNMT · Dec 22, 2016 · ac65838 · ac65838
1 parent b30ec03
commit ac65838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/preprocess.lua b/preprocess.lua
@@ -380,7 +380,7 @@ local function main()
   end
 
   print('Saving data to \'' .. opt.save_data .. '-train.t7\'...')
-  torch.save(opt.save_data .. '-train.t7', data)
+  torch.save(opt.save_data .. '-train.t7', data, 'binary', false)
 
 end